th3r00t/pyShelf
1
0
Fork 0
mirror of https://github.com/th3r00t/pyShelf.git synced 2026-04-28 01:59:35 -04:00
Code Issues Packages Projects Releases Wiki Activity

Changing json returns

Browse Source
This commit is contained in:
th3r00t
2023-03-16 23:17:45 -04:00
parent d4cef4289e
commit 7baa5f895c
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/frontend/lib/FastAPIServer.py vendored
View File

@@ -22,6 +22,8 @@ templates = Jinja2Templates(directory="src/frontend/templates")
origins = [ origins = [
"http://localhost", "http://localhost",
"http://localhost:8081", "http://localhost:8081",
"http://localhost:8080",
"*"
] ]
app.add_middleware( app.add_middleware(
CORSMiddleware, CORSMiddleware,
@@ -60,25 +62,28 @@ def books_tojson(obj) -> dumps:
"""Convert an object to a dictionary.""" """Convert an object to a dictionary."""
_books: list = [] _books: list = []
for book in obj: for book in obj:
convert_none = lambda x: x if x is not None else "None"
_books.append({ _books.append({
"book_id": book[0].book_id, "book_id": book[0].book_id,
"title": book[0].title, "title": book[0].title,
"author": book[0].author, "author": book[0].author,
"categories": book[0].categories, "categories": convert_none(book[0].categories),
"cover": base64decode(book[0].cover), "cover": base64decode(book[0].cover),
"pages": book[0].pages, "pages": convert_none(book[0].pages),
"progress": book[0].progress, "progress": convert_none(book[0].progress),
"file_name": book[0].file_name, "file_name": book[0].file_name,
"description": book[0].description, "description": convert_none(book[0].description),
"date": convertDateTime(book[0].date), "date": convertDateTime(book[0].date),
"rights": book[0].rights, "rights": convert_none(book[0].rights),
"tags": book[0].tags, "tags": convert_none(book[0].tags),
"identifier": book[0].identifier, "identifier": convert_none(book[0].identifier),
"publisher": book[0].publisher, "publisher": convert_none(book[0].publisher),
}) })
# compressed = gzip.compress(dumps(_books).encode("utf-8")) # compressed = gzip.compress(dumps(_books).encode("utf-8"))
# compressed = gzip.compress(dumps(_books).encode()) # compressed = gzip.compress(dumps(_books).encode())
return dumps(_books) # breakpoint()
# return dumps(_books)
return _books
def book_tojson(book) -> dumps: def book_tojson(book) -> dumps:
@@ -166,6 +171,7 @@ class FastAPIServer():
headers = {"Accept-Encoding": "gzip"} headers = {"Accept-Encoding": "gzip"}
"""Home page responder.""" """Home page responder."""
return JSONResponse(content=books_tojson(books)) return JSONResponse(content=books_tojson(books))
# return JSONResponse(content=books)
@app.get("/book/{book_id}", response_class=JSONResponse) @app.get("/book/{book_id}", response_class=JSONResponse)
async def book(request: Request, book_id: int): async def book(request: Request, book_id: int):