| 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 |
| |
| |
