shortGPT/database/content_database.py

from uuid import uuid4
from shortGPT.database.db_document import TINY_MONGO_DATABASE, TinyMongoDocument

from shortGPT.database.content_data_manager import ContentDataManager
class ContentDatabase:
    def __init__(self, ):
        self.content_collection = TINY_MONGO_DATABASE["content_db"]["content_documents"]

    def instanciateContentDataManager(self, id: str, content_type: str, new=False):
        db_doc = TinyMongoDocument("content_db", "content_documents", id)
        return ContentDataManager(db_doc, content_type, new)

    def getContentDataManager(self, id, content_type: str):
        try:
            db_doc = TinyMongoDocument("content_db", "content_documents", id)
            return ContentDataManager(db_doc, content_type, False)
        except:
            return None

    def createContentDataManager(self, content_type: str) -> ContentDataManager:
        try:
            new_short_id = uuid4().hex[:24]
            db_doc = TinyMongoDocument("content_db", "content_documents", new_short_id, True)
            return ContentDataManager(db_doc, content_type, True)
        except:
            return None