arpakitlib 1.7.243__py3-none-any.whl → 1.7.244__py3-none-any.whl

arpakitlib/_arpakit_project_template/src/core/settings.py

@@ -8,6 +8,7 @@ from pydantic_core.core_schema import ValidationInfo
 
 from arpakitlib.ar_json_util import safely_transfer_obj_to_json_str
 from arpakitlib.ar_settings_util import SimpleSettings
+from arpakitlib.ar_sqlalchemy_util import generate_sqlalchemy_url
 from src.core.const import ProjectPaths
 
 
@@ -27,14 +28,40 @@ class Settings(SimpleSettings):
     @field_validator("sql_db_url", mode="after")
     def validate_sql_db_url(cls, v: Any, validation_info: ValidationInfo) -> str | None:
         if v is not None:
-            return None
-        user = validation_info.data.get('sql_db_user')
-        password = validation_info.data.get('sql_db_password')
-        port = validation_info.data.get('sql_db_port')
-        database = validation_info.data.get('sql_db_database')
-        if user is not None and password is not None and port is not None and database is not None:
-            return f"postgresql://{user}:{password}@127.0.0.1:{port}/{database}"
-        return None
+            return v
+
+        user = validation_info.data.get("sql_db_user")
+        password = validation_info.data.get("sql_db_password")
+        port = validation_info.data.get("sql_db_port")
+        database = validation_info.data.get("sql_db_database")
+
+        return generate_sqlalchemy_url(
+            base="postgresql",
+            user=user,
+            password=password,
+            port=port,
+            database=database
+        )
+
+    async_sql_db_url: str | None = None
+
+    @field_validator("async_sql_db_url", mode="after")
+    def validate_async_sql_db_url(cls, v: Any, validation_info: ValidationInfo) -> str | None:
+        if v is not None:
+            return v
+
+        user = validation_info.data.get("sql_db_user")
+        password = validation_info.data.get("sql_db_password")
+        port = validation_info.data.get("sql_db_port")
+        database = validation_info.data.get("sql_db_database")
+
+        return generate_sqlalchemy_url(
+            base="postgresql+asyncpg",
+            user=user,
+            password=password,
+            port=port,
+            database=database
+        )
 
     sql_db_echo: bool = False
 

arpakitlib/_arpakit_project_template/src/db/util.py

@@ -16,6 +16,7 @@ def get_base_dbm() -> type[BaseDBM]:
 def create_sqlalchemy_db() -> SQLAlchemyDB:
     return SQLAlchemyDB(
         db_url=get_cached_settings().sql_db_url,
+        async_db_url=get_cached_settings().async_sql_db_url,
         db_echo=get_cached_settings().sql_db_echo,
         base_dbm=get_base_dbm()
     )

arpakitlib/ar_sqlalchemy_util.py

@@ -3,6 +3,7 @@ import asyncio
 import logging
 from datetime import timedelta, datetime
 from typing import Any
+from urllib.parse import quote_plus
 from uuid import uuid4
 
 from sqlalchemy import create_engine, QueuePool, text, func, inspect, AsyncAdaptedQueuePool
@@ -23,37 +24,98 @@ def get_string_info_from_declarative_base(class_: type[DeclarativeBase]):
     return res
 
 
+def generate_sqlalchemy_url(
+        *,
+        base: str = "postgresql",
+        user: str | None = None,
+        password: str | None = None,
+        host: str = "127.0.0.1",
+        port: int | None = None,
+        database: str | None = None,
+        **query_params
+) -> str:
+    """
+    Генерация URL для SQLAlchemy.
+
+    :param base: Базовая строка для подключения, например "postgresql+asyncpg" или "sqlite".
+    :param user: Имя пользователя (необязательно).
+    :param password: Пароль (необязательно).
+    :param host: Хост (по умолчанию "127.0.0.1").
+    :param port: Порт (необязательно).
+    :param database: Имя базы данных или путь к файлу для SQLite (необязательно).
+    :param query_params: Дополнительные параметры строки подключения.
+    :return: Строка подключения для SQLAlchemy.
+    """
+    # Формируем часть с авторизацией
+    auth_part = ""
+    if user and password:
+        auth_part = f"{quote_plus(user)}:{quote_plus(password)}@"
+    elif user:
+        auth_part = f"{quote_plus(user)}@"
+
+    # Формируем часть с хостом и портом
+    host_part = ""
+    if base.startswith("sqlite"):
+        # Для SQLite хост и порт не нужны
+        host_part = ""
+    else:
+        host_part = f"{host}"
+        if port:
+            host_part += f":{port}"
+
+    # Формируем часть с базой данных
+    database_part = f"/{database}" if database else ""
+    if base.startswith("sqlite") and database:
+        # Для SQLite путь указывается как абсолютный
+        database_part = f"/{database}"
+
+    # Дополнительные параметры
+    query_part = ""
+    if query_params:
+        query_items = [f"{key}={quote_plus(str(value))}" for key, value in query_params.items()]
+        query_part = f"?{'&'.join(query_items)}"
+
+    return f"{base}://{auth_part}{host_part}{database_part}{query_part}"
+
+
 class SQLAlchemyDB:
     def __init__(
             self,
             *,
-            db_url: str = "postgresql://arpakitlib:arpakitlib@localhost:50517/arpakitlib",
-            async_db_url: str = "postgresql+asyncpg://arpakitlib:arpakitlib@localhost:50517/arpakitlib",
+            db_url: str | None = "postgresql://arpakitlib:arpakitlib@localhost:50517/arpakitlib",
+            async_db_url: str | None = "postgresql+asyncpg://arpakitlib:arpakitlib@localhost:50517/arpakitlib",
             db_echo: bool = False,
             base_dbm: type[BaseDBM] | None = None,
             db_models: list[Any] | None = None,
     ):
-        self._logger = logging.getLogger(self.__class__.__name__)
-        self.engine = create_engine(
-            url=db_url,
-            echo=db_echo,
-            pool_size=10,
-            max_overflow=10,
-            poolclass=QueuePool,
-            pool_timeout=timedelta(seconds=30).total_seconds(),
-        )
+        self._logger = logging.getLogger()
+
+        self.db_url = db_url
+        if self.db_url is not None:
+            self.engine = create_engine(
+                url=db_url,
+                echo=db_echo,
+                pool_size=10,
+                max_overflow=10,
+                poolclass=QueuePool,
+                pool_timeout=timedelta(seconds=30).total_seconds(),
+            )
         self.sessionmaker = sessionmaker(bind=self.engine)
         self.func_new_session_counter = 0
-        self.async_engine = create_async_engine(
-            url=async_db_url,
-            echo=db_echo,
-            pool_size=10,
-            max_overflow=10,
-            poolclass=AsyncAdaptedQueuePool,
-            pool_timeout=timedelta(seconds=30).total_seconds()
-        )
+
+        self.async_db_url = async_db_url
+        if self.async_db_url is not None:
+            self.async_engine = create_async_engine(
+                url=async_db_url,
+                echo=db_echo,
+                pool_size=10,
+                max_overflow=10,
+                poolclass=AsyncAdaptedQueuePool,
+                pool_timeout=timedelta(seconds=30).total_seconds()
+            )
         self.async_sessionmaker = async_sessionmaker(bind=self.async_engine)
         self.func_new_async_session_counter = 0
+
         self.base_dbm = base_dbm
 
     def is_table_exists(self, table_name: str) -> bool:

{arpakitlib-1.7.243.dist-info → arpakitlib-1.7.244.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arpakitlib
-Version: 1.7.243
+Version: 1.7.244
 Summary: arpakitlib
 License: Apache-2.0
 Keywords: arpakitlib,arpakit,arpakit-company,arpakitcompany,arpakit_company
@@ -39,6 +39,7 @@ Requires-Dist: paramiko (>=3.5.0,<4.0.0)
 Requires-Dist: pika (>=1.3.2,<2.0.0)
 Requires-Dist: poetry (>=2.0.1,<3.0.0)
 Requires-Dist: psycopg2-binary (>=2.9.10,<3.0.0)
+Requires-Dist: pulp (>=2.9.0,<3.0.0)
 Requires-Dist: pydantic (>=2.10.5,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.7.1,<3.0.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
@@ -46,6 +47,7 @@ Requires-Dist: pymongo (>=4.10.1,<5.0.0)
 Requires-Dist: pytz (>=2024.2,<2025.0)
 Requires-Dist: pyzabbix (>=1.3.1,<2.0.0)
 Requires-Dist: requests[socks] (>=2.32.3,<3.0.0)
+Requires-Dist: scipy (>=1.15.1,<2.0.0)
 Requires-Dist: sqladmin (>=0.20.1,<0.21.0)
 Requires-Dist: sqlalchemy (>=2.0.37,<3.0.0)
 Requires-Dist: twine (>=6.1.0,<7.0.0)

{arpakitlib-1.7.243.dist-info → arpakitlib-1.7.244.dist-info}/RECORD

@@ -90,7 +90,7 @@ arpakitlib/_arpakit_project_template/src/core/_check_logging.py,sha256=APQp8jQa3
 arpakitlib/_arpakit_project_template/src/core/_check_settings.py,sha256=BQUcr-yj3cxz5GQo0jXe99wMoqHKrrrKD3-UovsJAt0,284
 arpakitlib/_arpakit_project_template/src/core/_generate_settings_env_example.py,sha256=SiEJe8AYQPOWicsaCwf9RdXp6UAmfkBdBT18AAInGb4,483
 arpakitlib/_arpakit_project_template/src/core/const.py,sha256=e2Y0NIQHfzm3bmnbQnGy3Z5YKt6OYnIsRoqVY8oIV2E,1008
-arpakitlib/_arpakit_project_template/src/core/settings.py,sha256=0-35RjLUMEWkHnuH454AKafwWpxH9cUAdRRc97nVLzQ,3094
+arpakitlib/_arpakit_project_template/src/core/settings.py,sha256=R4dsQHOB9wVBn-A76sbJFQbsfCHyyR3AMKV8HZHIHmM,3819
 arpakitlib/_arpakit_project_template/src/core/util.py,sha256=mcikqcjljZa2qhYeoR1tR9JUSprrVSod8XcIK_PqS6o,1547
 arpakitlib/_arpakit_project_template/src/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arpakitlib/_arpakit_project_template/src/db/_check_conn_sqlalchemy_db.py,sha256=GPK7-w7x8ESqbZ0v1nI8m_ysWfnmjgRwyyh6kM8vYBQ,249
@@ -99,7 +99,7 @@ arpakitlib/_arpakit_project_template/src/db/_init_sqlalchemy_db.py,sha256=tHJ1NB
 arpakitlib/_arpakit_project_template/src/db/_reinit_sqlalchemy_db.py,sha256=_4O_xr6FXt3qcWOmO82CKG3XX6_RqbqaOYSEygxpBfA,339
 arpakitlib/_arpakit_project_template/src/db/const.py,sha256=dcvj5C9E2F2KCsGZPBBncQf_EvVJAC1qQgnyD8P4ZEw,6
 arpakitlib/_arpakit_project_template/src/db/sqlalchemy_model.py,sha256=tIdSerx8m6FA0cx4PDNJF8cWY1utpS_g0sZ0jBzUAbI,286
-arpakitlib/_arpakit_project_template/src/db/util.py,sha256=Ww0CYifxFcNkRvMedmDu806_Ic_Hbb1C5KQX8UsiLdc,787
+arpakitlib/_arpakit_project_template/src/db/util.py,sha256=wYMVXtw_XURv4DFSx4n6CXnKYE-xskzpYSRk9lBIELY,848
 arpakitlib/_arpakit_project_template/src/operation_execution/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arpakitlib/_arpakit_project_template/src/operation_execution/_remove_operations.py,sha256=KsvAk2zQCp0PX6-QiRW0LKHoJfmH0wVJcBZviFqAoBI,471
 arpakitlib/_arpakit_project_template/src/operation_execution/_start_operation_executor_worker.py,sha256=K861UI9LFNxfCoPpQ5AJX-oUwjB6bfFNKScKsyU9Xd8,575
@@ -205,14 +205,14 @@ arpakitlib/ar_settings_util.py,sha256=rnoTqbRuhiq7294D4crD5kbnU8-gNWJbwGU_Ls2gJo
 arpakitlib/ar_sleep_util.py,sha256=OaLtRaJQWMkGjfj_mW1RB2P4RaSWsAIH8LUoXqsH0zM,1061
 arpakitlib/ar_sqladmin_util.py,sha256=6Nv9VQssk9PB0piyuss__soYKdjVhdbIeXIv4AgUxmQ,2660
 arpakitlib/ar_sqlalchemy_model_util.py,sha256=ezui1VEVuABYgrywpP_A2wYvDBlIxwQTZ0HQgBJfZE0,6293
-arpakitlib/ar_sqlalchemy_util.py,sha256=Eol7SxNdHoi9XAfaUCNTe5jDt-csyo3GaY4fTBJlzGc,5763
+arpakitlib/ar_sqlalchemy_util.py,sha256=9dXCdKWmUPyMt19sF2YVBB-4x_tl5g2Bsfhn0c_SQ6M,8184
 arpakitlib/ar_ssh_runner_util.py,sha256=e9deuUdBW7Eh0Exx2nTBhk57SaOZYaJaSjNk8q6dbJk,6804
 arpakitlib/ar_str_util.py,sha256=yU5gOwNXUQaH5b_tM5t6fXUn9oUcv5EQbVnq2wXXIpQ,3378
 arpakitlib/ar_type_util.py,sha256=9C3ErtUVs0tAUqtK-foFzjJOykfBOntfCz2IogDOgfA,4134
 arpakitlib/ar_yookassa_api_client_util.py,sha256=VozuZeCJjmLd1zj2BdC9WfiAQ3XYOrIMsdpNK-AUlm0,5347
 arpakitlib/ar_zabbix_api_client_util.py,sha256=Q-VR4MvoZ9aHwZeYZr9G3LwN-ANx1T5KFmF6pvPM-9M,6402
-arpakitlib-1.7.243.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.7.243.dist-info/METADATA,sha256=unXXDHN-CYxrtUSEVR4kITElMCrzgW7DX0-Vz7-ryMw,3310
-arpakitlib-1.7.243.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-arpakitlib-1.7.243.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.7.243.dist-info/RECORD,,
+arpakitlib-1.7.244.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.7.244.dist-info/METADATA,sha256=7Q2Y0Wb_OIUHWc5aXYt1L2EHIfYCEeelb-JrayEGrA0,3386
+arpakitlib-1.7.244.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+arpakitlib-1.7.244.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.7.244.dist-info/RECORD,,