yuuki/litey
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ページサイズをctxにまとめる

Browse source
This commit is contained in:
yuuki 2025-04-06 08:33:43 +00:00
parent 97457bfdec
commit 35da875d10
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
app.py
View file

@ -84,6 +84,8 @@ async def lifespan(app: FastAPI):
#ctx["mongo_client"].litey.notes.create_index("id", unique=True) #ctx["mongo_client"].litey.notes.create_index("id", unique=True)
ctx["mongo_client"].litey.ngs.create_index("word", unique=True) ctx["mongo_client"].litey.ngs.create_index("word", unique=True)
ctx["page_size"] = 100
pprint(ctx) pprint(ctx)
redis_uri = environ.get("REDIS_URI", "redis://127.0.0.1:6379/") redis_uri = environ.get("REDIS_URI", "redis://127.0.0.1:6379/")
@ -135,11 +137,11 @@ def get_ip(req: Request) -> str:
return req.headers.get("CF-Connecting-IP") or req.headers.get("X-Forwarded-For") or req.client.host return req.headers.get("CF-Connecting-IP") or req.headers.get("X-Forwarded-For") or req.client.host
def get_max_page() -> int: def get_max_page() -> int:
return ceil(ctx["mongo_client"].litey.notes.count_documents({}) / 50) return ceil(ctx["mongo_client"].litey.notes.count_documents({}) / ctx["page_size"])
def get_litey_notes(id: str = None, page: int = 0) -> List[dict]: def get_litey_notes(id: str = None, page: int = 0) -> List[dict]:
if not id: if not id:
cursor = ctx["mongo_client"].litey.notes.find({}, { "_id": False }).sort("date", DESCENDING).skip(page * 50).limit(50) cursor = ctx["mongo_client"].litey.notes.find({}, { "_id": False }).sort("date", DESCENDING).skip(page * ctx["page_size"]).limit(ctx["page_size"])
return list(cursor) return list(cursor)
return ctx["mongo_client"].litey.notes.find_one({ "id": id }, { "_id": False }) return ctx["mongo_client"].litey.notes.find_one({ "id": id }, { "_id": False })