Brushed up the backend, added writing task 1 academic prompt gen and grading ENCOA-274

ielts_be/repositories/impl/document_stores/__init__.py

@@ -0,0 +1,7 @@
+from .firestore import Firestore
+from .mongo import MongoDB
+
+__all__ = [
+    "Firestore",
+    "MongoDB"
+]

ielts_be/repositories/impl/document_stores/firestore.py

@@ -0,0 +1,50 @@
+import logging
+from typing import Optional, List, Dict
+
+from google.cloud.firestore_v1.async_client import AsyncClient
+from google.cloud.firestore_v1.async_collection import AsyncCollectionReference
+from google.cloud.firestore_v1.async_document import AsyncDocumentReference
+from ielts_be.repositories import IDocumentStore
+
+
+class Firestore(IDocumentStore):
+    def __init__(self, client: AsyncClient):
+        self._client = client
+        self._logger = logging.getLogger(__name__)
+
+    async def save_to_db(self, collection: str, item, doc_id: Optional[str] = None) -> Optional[str]:
+        collection_ref: AsyncCollectionReference = self._client.collection(collection)
+
+        if doc_id:
+            document_ref: AsyncDocumentReference = collection_ref.document(doc_id)
+            await document_ref.set(item)
+            doc_snapshot = await document_ref.get()
+            if doc_snapshot.exists:
+                self._logger.info(f"Document added with ID: {document_ref.id}")
+                return document_ref.id
+        else:
+            update_time, document_ref = await collection_ref.add(item)
+            if document_ref:
+                self._logger.info(f"Document added with ID: {document_ref.id}")
+                return document_ref.id
+
+        return None
+
+    async def find(self, collection: str, query: Optional[Dict] = None) -> List[Dict]:
+        collection_ref: AsyncCollectionReference = self._client.collection(collection)
+        docs = []
+        async for doc in collection_ref.stream():
+            docs.append(doc.to_dict())
+        return docs
+
+    async def get_doc_by_id(self, collection: str, doc_id: str) -> Optional[Dict]:
+        collection_ref: AsyncCollectionReference = self._client.collection(collection)
+        doc_ref: AsyncDocumentReference = collection_ref.document(doc_id)
+        doc = await doc_ref.get()
+
+        if doc.exists:
+            return doc.to_dict()
+        return None
+
+    async def update(self, collection: str, filter_query: Dict, update: Dict) -> Optional[str]:
+        raise NotImplemented()

ielts_be/repositories/impl/document_stores/mongo.py

@@ -0,0 +1,41 @@
+import logging
+import uuid
+from typing import Optional, List, Dict
+
+from motor.motor_asyncio import AsyncIOMotorDatabase
+
+from ielts_be.repositories import IDocumentStore
+
+
+class MongoDB(IDocumentStore):
+
+    def __init__(self, mongo_db: AsyncIOMotorDatabase):
+        self._mongo_db = mongo_db
+        self._logger = logging.getLogger(__name__)
+
+    async def save_to_db(self, collection: str, item, doc_id: Optional[str] = None) -> Optional[str]:
+        collection_ref = self._mongo_db[collection]
+
+        if doc_id is None:
+            doc_id = str(uuid.uuid4())
+
+        item['id'] = doc_id
+
+        result = await collection_ref.insert_one(item)
+        if result.inserted_id:
+            # returning id instead of _id
+            self._logger.info(f"Document added with ID: {doc_id}")
+            return doc_id
+
+        return None
+
+    async def find(self, collection: str, query: Optional[Dict] = None) -> List[Dict]:
+        query = query if query else {}
+        cursor = self._mongo_db[collection].find(query)
+        return [document async for document in cursor]
+
+    async def update(self, collection: str, filter_query: Dict, update: Dict) -> Optional[str]:
+        return (await self._mongo_db[collection].update_one(filter_query, update, upsert=True)).upserted_id
+
+    async def get_doc_by_id(self, collection: str, doc_id: str) -> Optional[Dict]:
+        return await self._mongo_db[collection].find_one({"id": doc_id})