sharedkernel 1.1.0__tar.gz → 1.1.1__tar.gz

{sharedkernel-1.1.0 → sharedkernel-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sharedkernel
-Version: 1.1.0
+Version: 1.1.1
 Summary: sharekernel is an shared package between all python projects
 Author: Smilinno
 Requires-Dist: numpy

{sharedkernel-1.1.0 → sharedkernel-1.1.1}/setup.py

@@ -8,6 +8,6 @@ setup(
     # Needed for dependencies
     install_requires=['numpy','requests','pymongo','fastapi==0.89.1','PyJWT','pymilvus','chromadb'],
     # *strongly* suggested for sharing
-    version='1.1.0',
+    version='1.1.1',
     description='sharekernel is an shared package between all python projects',
 )

sharedkernel-1.1.1/sharedkernel/database/mongo_generic_repository.py

@@ -0,0 +1,50 @@
+from pymongo import MongoClient
+from bson import ObjectId
+from typing import Generic, TypeVar, List, Type
+from pydantic import BaseModel
+from sharedkernel.string_extentions import camel_to_snake
+
+T = TypeVar("T", bound=BaseModel)
+
+
+class MongoGenericRepository(Generic[T]):
+    def __init__(self, database: MongoClient, model: Type[T]):
+        self.database = database
+        self.__collection_name = camel_to_snake(model.__name__)
+        self.__collection = self.database[self.__collection_name]
+        self.model = model
+
+    def _map_to_model(self, document: dict) -> T:
+        document["id"] = str(document.pop("_id"))
+        return self.model.parse_obj(document)
+
+    def find_one(self, id: str) -> T:
+        query = {"_id": ObjectId(id), "is_deleted": False}
+        result = self.__collection.find_one(query)
+        return self._map_to_model(result) if result else None
+
+    def insert_one(self, data: T) -> str:
+        delattr(data, "id")
+        result = self.__collection.insert_one(data.dict())
+        return str(result.inserted_id)
+
+    def insert_many(self, data: List[T]) -> List[str]:
+        data_list = [delattr(d.dict(), "id") for d in data]
+        result = self.__collection.insert_many(data_list)
+        return [str(id_) for id_ in result.inserted_ids]
+
+    def update_one(self, id: str, data: T) -> int:
+        delattr(data, "id")
+        query = {"_id": ObjectId(id)}
+        result = self.__collection.update_one(query, {"$set": data.dict()})
+        return result.modified_count
+
+    def delete_one(self, id: str) -> int:
+        query = {"_id": ObjectId(id)}
+        result = self.__collection.delete_one(query)
+        return result.deleted_count
+
+    def get_all(self, page_number=1, page_size=10) -> List[T]:
+        skip_count = (page_number - 1) * page_size
+        result = self.__collection.find().skip(skip_count).limit(page_size)
+        return [self._map_to_model(doc) for doc in result]

{sharedkernel-1.1.0 → sharedkernel-1.1.1}/sharedkernel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sharedkernel
-Version: 1.1.0
+Version: 1.1.1
 Summary: sharekernel is an shared package between all python projects
 Author: Smilinno
 Requires-Dist: numpy

{sharedkernel-1.1.0 → sharedkernel-1.1.1}/sharedkernel.egg-info/SOURCES.txt

@@ -10,8 +10,7 @@ sharedkernel.egg-info/dependency_links.txt
 sharedkernel.egg-info/requires.txt
 sharedkernel.egg-info/top_level.txt
 sharedkernel/database/__init__.py
-sharedkernel/database/adapter.py
-sharedkernel/database/mongo_repository_base.py
+sharedkernel/database/mongo_generic_repository.py
 sharedkernel/database/vector_database_repository/__init__.py
 sharedkernel/database/vector_database_repository/chroma_startegy.py
 sharedkernel/database/vector_database_repository/milvus_strategy.py

sharedkernel-1.1.0/sharedkernel/database/adapter.py

@@ -1,6 +0,0 @@
-from pymongo import MongoClient
-
-class MongoDBClient:
-    def __init__(self,connection_string):
-        self.client = MongoClient(connection_string)
-        self.domain = self.client.domain

sharedkernel-1.1.0/sharedkernel/database/mongo_repository_base.py

@@ -1,82 +0,0 @@
-from typing import (Optional,
-                    List
-                    )
-from bson.objectid import ObjectId
-from typing import Generic, TypeVar
-from pydantic import BaseModel
-from pydantic import parse_obj_as
-
-from sharedkernel.string_extentions import camel_to_snake
-from .adapter import MongoDBClient
-
-
-from sharedkernel.exception.exception import BusinessException
-from sharedkernel.enum.error_code import ErrorCode
-
-mongo_client= MongoDBClient()
-
-ResultT = TypeVar("ResultT")
-
-T = TypeVar("T", bound=BaseModel)
-
-class MongoRepositoryBase(Generic[T]):
-    
-    def __init__(self,model):
-      self.model = model
-      self.model_name= camel_to_snake(self.model.__name__)
-      
-      
-    def find(self,id: str) -> T:
-        if (not ObjectId.is_valid(id)):
-            raise BusinessException(ErrorCode.Item_NotFound)
-        
-        query = {"_id": ObjectId(id),"is_deleted":False}
-        
-        item = mongo_client.domain[self.model_name].find_one(query)
-        
-        if (not item):
-            raise BusinessException(ErrorCode.Item_NotFound)
-        
-        item['_id']=str(item['_id'])
-        item['id']= item['_id']
-        item.pop('_id')
-
-        return self.model.parse_obj(item) if item else None
-    
-        
-    def get_all(self,query:dict = None) -> Optional[List]:
-        if query:
-            query["is_deleted"]= False
-        else:
-            query= {"is_deleted":False}
-        items = list(mongo_client.domain[self.model_name].find(query))
-
-        for item in items:
-            item['_id'] = str(item['_id'])
-            item['id']= item['_id']
-            item.pop('_id')
-        
-        return parse_obj_as(List[self.model],items)
-    
-
-    def insert(self, insert_data: T) -> str:
-        delattr(insert_data, "id")
-        inserted_data= mongo_client.domain[self.model_name].insert_one(insert_data.dict())
-        
-        return str(inserted_data.inserted_id)
-    
-    def bulk_insert(self, insert_data: List[T]):
-        insert_data=[delattr(data, "id") for data in insert_data]
-        mongo_client.domain[self.model_name].insert_many(insert_data)
-
-
-
-    def update(self,id:str,update_data: T):
-        query = {"_id": ObjectId(id)}
-        delattr(update_data, "id")
-        mongo_client.domain[self.model_name].update_one(query,  {"$set": update_data.dict()}, upsert=False)
-        
-
-    def query(self):
-        return mongo_client.domain[self.model_name]
-