added db connection

Author: nameless-coder

Diff for: .env

@@ -0,0 +1 @@
+DATABASE_URL="mongodb://localhost:27017/logger"

Diff for: docker-compose.yml

@@ -6,6 +6,10 @@ services:
     container_name: mongo
     image: mongo
     restart: always
+    ports:
+      - 27017:27017
+    env_file:
+      - .env
     volumes:
       - ./data:/data/db
 

Diff for: src/database/__init__.py

@@ -0,0 +1,61 @@
+from typing import Optional, Any, List
+
+from beanie import init_beanie, PydanticObjectId
+from models.production_run import ProductionRun
+
+from motor.motor_asyncio import AsyncIOMotorClient
+from pydantic import BaseSettings, BaseModel
+
+
+class Settings(BaseSettings):
+    DATABASE_URL: Optional[str] = None
+    SECRET_KEY: Optional[str] = "default"
+
+    async def initialize_database(self):
+        client = AsyncIOMotorClient(self.DATABASE_URL)
+        await init_beanie(database=client.get_default_database(),
+                          document_models=[ProductionRun])
+
+    class Config:
+        env_file = ".env"
+
+
+class Database:
+    def __init__(self, model):
+        self.model = model
+
+    async def save(self, document):
+        await document.create()
+        return
+
+    async def get(self, id: PydanticObjectId) -> bool:
+        doc = await self.model.get(id)
+        if doc:
+            return doc
+        return False
+
+    async def get_all(self) -> List[Any]:
+        docs = await self.model.find_all().to_list()
+        return docs
+
+    async def update(self, id: PydanticObjectId, body: BaseModel) -> Any:
+        doc_id = id
+        des_body = body.dict()
+
+        des_body = {k: v for k, v in des_body.items() if v is not None}
+        update_query = {"$set": {
+            field: value for field, value in des_body.items()
+        }}
+
+        doc = await self.get(doc_id)
+        if not doc:
+            return False
+        await doc.update(update_query)
+        return doc
+
+    async def delete(self, id: PydanticObjectId) -> bool:
+        doc = await self.get(id)
+        if not doc:
+            return False
+        await doc.delete()
+        return True

Diff for: src/database/connection.py

@@ -1,11 +1,21 @@
 from fastapi import FastAPI
+from database.connection import Settings
 
 from routes.production_run import production_run_router
 
 import uvicorn
 
 api = FastAPI(docs_url=None, redoc_url="/docs")
 
+settings = Settings()
+
+api.include_router(production_run_router)
+
+
+@api.on_event("startup")
+async def init_db():
+    await settings.initialize_database()
+
 
 @api.get('/')
 async def home() -> dict:
@@ -14,8 +24,5 @@ async def home() -> dict:
     }
 
 
-api.include_router(production_run_router)
-
-
 if __name__ == '__main__':
     uvicorn.run("main:api", host="0.0.0.0", port=8000, reload=True)

Diff for: src/main.py

@@ -1,8 +1,9 @@
 from datetime import datetime
+from beanie import Document
 from pydantic import BaseModel
 
 
-class ProductionRun(BaseModel):
+class ProductionRun(Document):
     id: int
     name: str
     dt: datetime = None