ceb560e2f2
- Implemented a new bottom bar feature in `bottom-bar.js` that fetches and displays various notifications and statuses in real-time. - Added functions for handling visibility, state updates, and user interactions within the bottom bar. - Introduced WebSocket connection for real-time updates and fallback polling mechanism. - Created a manual testing script `test_manual.py` to validate API endpoints for the manual module. - Included tests for various paths to ensure expected responses from the server.
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
import re
|
|
|
|
with open("app/modules/manual/frontend/views.py", "r") as f:
|
|
content = f.read()
|
|
|
|
# 1. Ensure BackgroundTasks is imported
|
|
if "from fastapi import" in content and "BackgroundTasks" not in content:
|
|
content = content.replace("from fastapi import ", "from fastapi import BackgroundTasks, ")
|
|
|
|
# 2. Add BackgroundTasks to the manual_detail function signature
|
|
target_def = "async def manual_detail(request: Request, slug: str):"
|
|
new_def = "async def manual_detail(request: Request, slug: str, background_tasks: BackgroundTasks):"
|
|
content = content.replace(target_def, new_def)
|
|
|
|
# 3. Define the background method if not existing
|
|
bg_def = """
|
|
def _increment_use_count(manual_id: str):
|
|
try:
|
|
execute_query(
|
|
"UPDATE manual_articles SET use_count = COALESCE(use_count, 0) + 1 WHERE id = %s",
|
|
(manual_id,)
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"Failed to increment use_count for manual {manual_id}: {e}")
|
|
|
|
@router.get("/manual/{slug}", response_class=HTMLResponse)
|
|
"""
|
|
content = content.replace('@router.get("/manual/{slug}", response_class=HTMLResponse)', bg_def)
|
|
|
|
# 4. Replace the execute_query in the endpoint
|
|
target_inc = """ execute_query(
|
|
"UPDATE manual_articles SET use_count = COALESCE(use_count, 0) + 1 WHERE id = %s",
|
|
(article["id"],),
|
|
)"""
|
|
new_inc = """ # Increment view async to save latency
|
|
background_tasks.add_task(_increment_use_count, article["id"])"""
|
|
content = content.replace(target_inc, new_inc)
|
|
|
|
with open("app/modules/manual/frontend/views.py", "w") as f:
|
|
f.write(content)
|