rb-commons 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

rb_commons/configs/config.py

@@ -0,0 +1,30 @@
+from pydantic_settings import BaseSettings
+
+class CommonConfigs(BaseSettings):
+    service_name: str
+    service_port: int
+    service_id: str
+    service_hostname: str = '127.0.0.1'
+    service_host: str
+
+    consul_host: str = '127.0.0.1'
+    consul_port: int = 8888
+
+#     db
+    POSTGRES_HOST: str
+    POSTGRES_USER: str
+    POSTGRES_PORT: int
+    POSTGRES_PASSWORD: str
+    POSTGRES_DB: str
+    DB_ALEMBIC_URL: str
+
+    @property
+    def database_url(self):
+        return f"postgresql+asyncpg://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}@{self.POSTGRES_HOST}:{self.POSTGRES_PORT}/{self.POSTGRES_DB}"
+
+    class Config:
+        env_file = ".env"
+        env_file_encoding = "utf-8"
+
+
+configs = CommonConfigs()

rb_commons/orm/exceptions.py

@@ -0,0 +1,11 @@
+class DatabaseException(Exception):
+    def __init__(self, text: str, user_id: int | None = None , original_exception=None):
+        self.user_id = user_id
+        self.text = text
+        self.original_exception = original_exception
+
+class InternalException(Exception):
+    def __init__(self, text: str, user_id: int | None = None, telegram_id: int | None = None, original_exception=None):
+        self.user_id = user_id
+        self.text = text
+        self.original_exception = original_exception

rb_commons/orm/managers.py

@@ -0,0 +1,178 @@
+from typing import TypeVar, Type, Generic, Optional, List, Dict
+from sqlalchemy import select, delete, update
+from sqlalchemy.exc import IntegrityError, SQLAlchemyError, NoResultFound
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import declarative_base
+from rb_commons.orm.exceptions import DatabaseException, InternalException
+
+ModelType = TypeVar('ModelType', bound=declarative_base())
+
+class BaseManager(Generic[ModelType]):
+    model: Type[ModelType]
+
+    def __init__(self, session: AsyncSession) -> None:
+        self.session = session
+
+    async def get(self, **kwargs) -> Optional[ModelType]:
+        """
+           get object based on conditions
+       """
+        query = select(self.model).filter_by(**kwargs)
+        result = await self.session.execute(query)
+        return result.scalar_one_or_none()
+
+    async def filter(self, **kwargs) -> List[ModelType]:
+        """
+           Filter objects based on conditions
+       """
+        query = select(self.model).filter_by(**kwargs)
+        result = await self.session.execute(query)
+        return list(result.scalars().all())
+
+    async def create(self, **kwargs) -> ModelType:
+        """
+               Create a new object
+       """
+        obj = self.model(**kwargs)
+
+        try:
+            self.session.add(obj)
+            await self.session.flush()
+            await self.session.commit()
+            await self.session.refresh(obj)
+            return obj
+        except IntegrityError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Constraint violation: {str(e)}") from e
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Database error occurred: {str(e)}") from e
+        except Exception as e:
+            await self.session.rollback()
+            raise InternalException(f"Unexpected error during creation: {str(e)}") from e
+
+
+    async def delete(self, id: Optional[int] = None, **filters) -> bool | None:
+        """
+        Delete object(s) with flexible filtering options
+
+        :param id: Specific ID to delete
+        :param filters: Additional filter conditions
+        :return: Number of deleted records or None
+        """
+        try:
+            if id is not None:
+                filters['id'] = id
+
+            delete_stmt = delete(self.model).filter_by(**filters)
+            result = await self.session.execute(delete_stmt)
+            await self.session.commit()
+            return result.rowcount() > 0
+        except NoResultFound:
+            return False
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Delete operation failed: {str(e)}") from e
+
+    async def bulk_delete(self, **filters) -> int:
+        """
+        Bulk delete with flexible filtering
+
+        :param filters: Conditions for bulk deletion
+        :return: Number of deleted records
+        """
+        try:
+            delete_stmt = delete(self.model).filter_by(**filters)
+            result = await self.session.execute(delete_stmt)
+            await self.session.commit()
+            return result.rowcount()
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Bulk delete failed: {str(e)}") from e
+
+    async def update_by_filters(self, filters: Dict, **update_fields) -> Optional[ModelType]:
+        """
+        Update object(s) with flexible filtering options
+
+        :param filters: Conditions for selecting records to update
+        :param update_fields: Fields and values to update
+        :return: Number of updated records
+        """
+        if not update_fields:
+            raise InternalException("No fields provided for update")
+
+        try:
+            update_stmt = update(self.model).filter_by(**filters).values(**update_fields)
+            await self.session.execute(update_stmt)
+            await self.session.commit()
+            updated_instance = await self.get(**filters)
+            return updated_instance
+        except IntegrityError as e:
+            await self.session.rollback()
+            raise InternalException(f"Constraint violation: {str(e)}") from e
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Update operation failed: {str(e)}") from e
+
+    async def update(self, instance: ModelType, **update_fields) -> Optional[ModelType]:
+        """
+        Update an existing database instance with new fields
+
+        :param instance: The database model instance to update
+        :param update_fields: Keyword arguments of fields to update
+        :return: The updated instance
+
+        :raises ValueError: If no update fields are provided
+        :raises RuntimeError: For database-related errors
+        """
+        # Validate update fields
+        if not update_fields:
+            raise InternalException("No fields provided for update")
+
+        try:
+            # Apply updates directly to the instance
+            for key, value in update_fields.items():
+                setattr(instance, key, value)
+
+            self.session.add(instance)
+            await self.session.commit()
+            await self.session.refresh(instance)
+
+            return instance
+
+        except IntegrityError as e:
+            await self.session.rollback()
+            raise InternalException(f"Constraint violation: {str(e)}") from e
+
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Update operation failed: {str(e)}") from e
+
+    async def save(self, instance: ModelType) -> Optional[ModelType]:
+        """
+        Save instance
+
+        :param instance: The database model instance to save
+        :return: The saved instance
+
+        :raises ValueError: If no update fields are provided
+        :raises RuntimeError: For database-related errors
+        """
+        try:
+            self.session.add(instance)
+            await self.session.commit()
+            await self.session.refresh(instance)
+            return instance
+
+        except IntegrityError as e:
+            await self.session.rollback()
+            raise InternalException(f"Constraint violation: {str(e)}") from e
+
+        except SQLAlchemyError as e:
+            await self.session.rollback()
+            raise DatabaseException(f"Update operation failed: {str(e)}") from e
+
+
+    async def is_exists(self, **kwargs) -> bool:
+        return await self.get(**kwargs) is not None
+

{rb_commons-0.1.0.dist-info → rb_commons-0.1.1.dist-info}/METADATA

@@ -1,35 +1,21 @@
 Metadata-Version: 2.1
 Name: rb-commons
-Version: 0.1.0
-Summary: Orm commons for sqlalchemy based projects
+Version: 0.1.1
+Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/Abdulvoris101/rb-commons
 Author: Abdulvoris
 Author-email: erkinovabdulvoris101@gmail.com
 Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: annotated-types~=0.7.0
-Requires-Dist: anyio~=4.7.0
-Requires-Dist: click~=8.1.8
-Requires-Dist: colorama~=0.4.6
-Requires-Dist: fastapi~=0.115.6
-Requires-Dist: greenlet~=3.1.1
-Requires-Dist: h11~=0.14.0
-Requires-Dist: httptools~=0.6.4
-Requires-Dist: idna~=3.10
-Requires-Dist: pydantic~=2.10.4
-Requires-Dist: pydantic-settings~=2.7.1
-Requires-Dist: pydantic_core~=2.27.2
-Requires-Dist: python-dotenv~=1.0.1
-Requires-Dist: PyYAML~=6.0.2
-Requires-Dist: sniffio~=1.3.1
-Requires-Dist: SQLAlchemy~=2.0.36
-Requires-Dist: starlette~=0.41.3
-Requires-Dist: typing_extensions~=4.12.2
-Requires-Dist: uvicorn~=0.34.0
-Requires-Dist: watchfiles~=1.0.3
-Requires-Dist: websockets~=14.1
+Requires-Dist: annotated-types==0.7.0
+Requires-Dist: greenlet==3.1.1
+Requires-Dist: pydantic==2.10.4
+Requires-Dist: PyJWT==2.10.1
+Requires-Dist: python-dotenv==1.0.1
+Requires-Dist: SQLAlchemy==2.0.36
 
 # RB-Commons
 

rb_commons-0.1.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+rb_commons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/configs/config.py,sha256=8Z2_om4yHAQI7PYX5qzVNR78DMAYsgbKfoa5skGBWlk,746
+rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
+rb_commons/orm/managers.py,sha256=VopUw29sCXrPTriBALR9qwkG5yCu6BDHRgR7cCygNiE,6726
+rb_commons-0.1.1.dist-info/METADATA,sha256=84zZkWX-fJFyuUuy2Q7h--H_RBRDytEwwMDXjggwRt4,6276
+rb_commons-0.1.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+rb_commons-0.1.1.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.1.1.dist-info/RECORD,,

rb_commons-0.1.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+rb_commons

rb_commons-0.1.0.dist-info/RECORD

@@ -1,4 +0,0 @@
-rb_commons-0.1.0.dist-info/METADATA,sha256=Sk7kcuRLGSWvp0FHU6UkpKYi_2At2g8Tgd-MxGqJ-hI,6703
-rb_commons-0.1.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-rb_commons-0.1.0.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-rb_commons-0.1.0.dist-info/RECORD,,

rb_commons-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-