Merge pull request #8 from hexfrost/beta-0.1.7

Beta 0.1.7

Author: kaziamov

pyproject.toml

@@ -22,7 +22,7 @@ dependencies = [
     "sqlalchemy-utils>=0.41.2",
 ]
 name = "hexfrost-toolbox"
-version = "0.1.6b"
+version = "0.1.7b"
 description = "Open source library with useful utils for fast development"
 readme = "README.md"
 

tests/fixtures/database.py

@@ -9,7 +9,7 @@ def db_settings():
         POSTGRES_USER="postgres",
         POSTGRES_PASSWORD = "postgres",
         POSTGRES_HOST = "0.0.0.0",
-        POSTGRES_PORT = "5432",
+        POSTGRES_PORT = 5432,
         POSTGRES_DB = "postgres"
     )
     return data

toolbox/auth/__init__.py

@@ -40,6 +40,14 @@ def set_settings(cls, settings: BearerTokenMiddlewareSettings):
         cls._exclude_paths = settings.exclude_paths
         cls._token_validator = settings.token_validator
 
+    @classmethod
+    def update_settings(cls, token_validator=Callable, exclude_paths=None):
+        cls._token_validator = token_validator
+        if not exclude_paths:
+            cls._exclude_paths = []
+        else:
+            cls._exclude_paths = exclude_paths
+
     async def _is_token_valid(self, token: str) -> bool:
         if await self._token_validator(token):
             return True

toolbox/auth/middlewares/__init__.py

@@ -17,7 +17,7 @@ class DatabaseConnectionSettings:
     POSTGRES_USER: str
     POSTGRES_PASSWORD: str
     POSTGRES_HOST: str
-    POSTGRES_PORT: str
+    POSTGRES_PORT: int
     POSTGRES_DB: str
     SCHEMA_MAPPING: dict | None = None
 

toolbox/auth/middlewares/fastapi_.py

@@ -52,17 +52,13 @@ async def temporary_database(settings: "DatabaseConnectionSettings", base_model,
             except Exception as e:
                 logger.error({"msg": e})
 
-        engine = db_manager.get_engine()
-        async with engine.begin() as connection:
-            try:
-                await connection.run_sync(base_model.metadata.create_all)
-
-            except Exception as e:
-                logger.error({"msg": f"Error creating schema: {e}"})
-
-                raise
+    from sqlalchemy import create_engine
 
-        yield
+    engine = create_engine(settings.get_dsn())
+    base_model.metadata.drop_all(bind=engine)
+    base_model.metadata.create_all(bind=engine)
+    yield
+    base_model.metadata.drop_all(bind=engine)
 
     try:
         conn = await asyncpg.connect(dsn=dsn)