add api.log&errorlog->db

backend/app/core/apiexception.py

@@ -0,0 +1,17 @@
+from app.db.schemas import ErrorCreate
+from app.db.session import SessionLocal
+from app.db.crud import create_log
+
+class APIException(Exception):
+
+    def __init__(self,location:str,title:str,content:str):
+      if(len(content) > 5000):
+         content =content[0:5000]
+      self.error = ErrorCreate(location=location,title=title,content=content)
+
+def writedblog(exc: APIException):
+    db = SessionLocal()
+    try:
+        create_log(db,exc.error)
+    finally:
+        db.close()

backend/app/db/crud.py

@@ -282,4 +282,12 @@ def get_eventactions(db: Session,eventid: str):
     eveactions = db.query(models.Action).join(models.EventAction,models.EventAction.actionid == models.Action.id ).join(models.Event,models.Event.id == models.EventAction.eventid).filter(models.Event.eventid == eventid).all()
     if not eveactions:
         raise HTTPException(status_code=404, detail="Data not found")
-    return eveactions
+    return eveactions
+
+
+def create_log(db: Session, error:schemas.ErrorCreate):
+    db_log = models.ErrorLog(location=error.location,title=error.title,content=error.content)  
+    db.add(db_log)
+    db.commit()  
+    db.refresh(db_log) 
+    return db_log

backend/app/db/models.py

@@ -90,4 +90,12 @@ class EventAction(Base):
     __tablename__ = "eventaction"
 
     eventid = Column(Integer,ForeignKey("event.id"))
-    actionid = Column(Integer,ForeignKey("action.id"))
+    actionid = Column(Integer,ForeignKey("action.id"))
+
+
+class ErrorLog(Base):
+       __tablename__ = "errorlog"
+       id = Column(Integer, primary_key=True, index=True)
+       location = Column(String(200))
+       title = Column(String(50))
+       content = Column(String(5000))

backend/app/db/schemas.py

@@ -139,4 +139,9 @@ class Event(Base):
     mobile: bool
 
     class Config:
-        orm_mode = True
+        orm_mode = True
+
+class ErrorCreate(BaseModel):
+    location:str
+    title:str
+    content:str

backend/app/main.py

@@ -11,6 +11,11 @@ from app.core.auth import get_current_active_user
 from app.core.celery_app import celery_app
 from app import tasks
 from fastapi.middleware.cors import CORSMiddleware
+import logging
+from app.core.apiexception import APIException, writedblog
+from app.db.crud import create_log
+from fastapi.responses import JSONResponse
+import asyncio
 
 Base.metadata.create_all(bind=engine) 
 
@@ -37,6 +42,21 @@ app.add_middleware(
 #     request.state.db.close()
 #     return response
 
+@app.on_event("startup")
+async def startup_event():
+    logger = logging.getLogger("uvicorn.access")
+    handler = logging.handlers.RotatingFileHandler("Log/api.log",mode="a",maxBytes = 100*1024, backupCount = 3)
+    handler.setFormatter(logging.Formatter("%(asctime)s - %(levelname)s - %(message)s"))
+    logger.addHandler(handler)
+
+@app.exception_handler(APIException)
+async def api_exception_handler(request: Request, exc: APIException):
+    loop = asyncio.get_event_loop()
+    loop.run_in_executor(None,writedblog,exc)
+    return JSONResponse(
+            status_code=400,
+            content={"detail": f"{exc.error.content}"},
+    )
 
 @app.get("/api/v1")
 async def root():