arpakitlib 1.8.267__py3-none-any.whl → 1.8.269__py3-none-any.whl

arpakitlib/ar_base_worker_util.py

@@ -75,8 +75,10 @@ class BaseWorker(ABC):
     def sync_on_error(self, exception: Exception, **kwargs):
         pass
 
-    def sync_safe_run(self):
+    def sync_safe_run(self, *, timeout_before_run: timedelta | None = None):
         self._logger.info("start")
+        if timeout_before_run is not None:
+            sync_safe_sleep(timeout_before_run)
         try:
             self.sync_on_startup()
         except Exception as exception:
@@ -117,8 +119,10 @@ class BaseWorker(ABC):
     async def async_on_error(self, exception: Exception, **kwargs):
         pass
 
-    async def async_safe_run(self):
+    async def async_safe_run(self, *, timeout_before_run: timedelta | None = None):
         self._logger.info("start async_safe_run")
+        if timeout_before_run is not None:
+            await async_safe_sleep(timeout_before_run)
         try:
             await self.async_on_startup()
         except Exception as exception:

arpakitlib/ar_sqlalchemy_drop_check_constraints.py

@@ -0,0 +1,82 @@
+from __future__ import annotations
+
+from typing import Optional
+
+import sqlalchemy
+from sqlalchemy.engine import Engine
+from sqlalchemy.orm import DeclarativeBase
+
+
+def _quote_ident_pg(ident: str) -> str:
+    # простое кавычирование для PG (экранируем двойные кавычки)
+    return '"' + ident.replace('"', '""') + '"'
+
+
+def _quote_ident_mysql(ident: str) -> str:
+    # простое кавычирование для MySQL/MariaDB (экранируем обратные апострофы)
+    return '`' + ident.replace('`', '``') + '`'
+
+
+def _qualified_name(
+        schema: Optional[str], table: str, dialect_name: str
+) -> str:
+    if dialect_name == "postgresql":
+        q = _quote_ident_pg
+    elif dialect_name in {"mysql", "mariadb"}:
+        q = _quote_ident_mysql
+    else:
+        # по умолчанию без кавычек — но мы не должны сюда попасть,
+        # поскольку ниже ограничиваем поддержанные диалекты.
+        q = lambda x: x
+
+    if schema:
+        return f"{q(schema)}.{q(table)}"
+    return q(table)
+
+
+def drop_sqlalchemy_check_constraints(*, base_: type[DeclarativeBase], engine: Engine) -> None:
+    """
+    Удаляет ВСЕ существующие в БД CHECK-ограничения для таблиц из base_.metadata,
+    выполняя прямые SQL-запросы (без SQLAlchemy DDL-объектов).
+
+    Поддерживаемые диалекты:
+      - PostgreSQL: ALTER TABLE <sch>.<tbl> DROP CONSTRAINT <name>
+      - MySQL 8.0.16+/MariaDB 10.2+: ALTER TABLE <sch>.<tbl> DROP CHECK <name>
+
+    Для SQLite кидает NotImplementedError.
+    """
+
+    if engine.dialect.name == "sqlite":
+        raise NotImplementedError(
+            "Удаление CHECK-констреинтов прямыми запросами для SQLite не поддержано: "
+            "в большинстве версий требуется пересоздание таблицы."
+        )
+    if engine.dialect.name not in {"postgresql", "mysql", "mariadb"}:
+        raise NotImplementedError(f"Пока не реализовано для диалекта: {engine.dialect.name}")
+
+    with engine.begin() as conn:
+        conn_inspection_ = sqlalchemy.inspect(conn)
+
+        for table in base_.metadata.tables.values():
+            fqtn = _qualified_name(table.schema, table.name, engine.dialect.name)
+
+            # берём ВСЕ реальные CHECK-и из базы
+            checks = conn_inspection_.get_check_constraints(table.name, schema=table.schema)
+
+            for check_ in checks:
+                name = check_.get("name")
+                if not name:
+                    # На всякий случай: без имени нам нечего дропать безопасно
+                    # (в PG/ MySQL синтаксис требует имя).
+                    continue
+
+                if engine.dialect.name == "postgresql":
+                    ident = _quote_ident_pg(name)
+                    sql = f"ALTER TABLE {fqtn} DROP CONSTRAINT {ident}"
+                else:  # mysql / mariadb
+                    ident = _quote_ident_mysql(name)
+                    # В MySQL/MariaDB именно DROP CHECK <name>
+                    sql = f"ALTER TABLE {fqtn} DROP CHECK {ident}"
+
+                # выполняем прямой SQL-запрос
+                conn.exec_driver_sql(sql)

arpakitlib/ar_sqlalchemy_util.py

@@ -305,6 +305,10 @@ class SQLAlchemyDb:
         from arpakitlib.ar_ensure_sqlalchemy_check_constraints import ensure_sqlalchemy_check_constraints
         ensure_sqlalchemy_check_constraints(base_=self.base_dbm, engine=self.engine)
 
+    def drop_check_constraints(self):
+        from arpakitlib.ar_sqlalchemy_drop_check_constraints import drop_sqlalchemy_check_constraints
+        drop_sqlalchemy_check_constraints(base_=self.base_dbm, engine=self.engine)
+
     def init(self, ensure_check_constraints: bool = True):
         self.base_dbm.metadata.create_all(bind=self.engine, checkfirst=True)
         if ensure_check_constraints:

{arpakitlib-1.8.267.dist-info → arpakitlib-1.8.269.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arpakitlib
-Version: 1.8.267
+Version: 1.8.269
 Summary: arpakitlib
 License: Apache-2.0
 Keywords: arpakitlib,arpakit,arpakit-company,arpakitcompany,arpakit_company

{arpakitlib-1.8.267.dist-info → arpakitlib-1.8.269.dist-info}/RECORD

@@ -384,7 +384,7 @@ arpakitlib/ar_arpakit_schedule_uust_api_client_util.py,sha256=MRPaF31CRhYA45ldPn
 arpakitlib/ar_arpakit_schedule_uust_site_util.py,sha256=8wLct9Gd4MWkXzB6nSmETAwTPLw8lfpWgx0LoWSAOvg,1643
 arpakitlib/ar_arpakitlib_cli_util.py,sha256=RJGcfEZ_q74FJ4tqdXvt7xQpShTszOvKu1mbp3D8qzw,2599
 arpakitlib/ar_base64_util.py,sha256=udSSpeXMZx0JgQknl4hQgZ8kr1Ps_aQOloIXu4T9dMQ,1286
-arpakitlib/ar_base_worker_util.py,sha256=e-Vj9w1D-59KN3Zz7TQlOB2lW4fZphJlsPpFkmX-gtY,5982
+arpakitlib/ar_base_worker_util.py,sha256=QXrk_bnxgxtbmz8YhfShxuLMMJEz2HK2KZpYeFcRm1I,6267
 arpakitlib/ar_blank_util.py,sha256=qFUdY8usL_pRYamz8Rw1fW3fzNIgrLmpdYP8q-_PQvw,2281
 arpakitlib/ar_cache_file_util.py,sha256=Fo2pH-Zqm966KWFBHG_pbiySGZvhIFCYqy7k1weRfJ0,3476
 arpakitlib/ar_class_util.py,sha256=i76pQW_7k_S2m_DlQh6xNjtggv9Col3WSx9W_bwk98E,722
@@ -426,12 +426,13 @@ arpakitlib/ar_schedule_uust_api_client_util.py,sha256=rXI2_3OTaIBgR-GixM1Ti-Ue1f
 arpakitlib/ar_settings_util.py,sha256=Y5wi_cmsjDjfJpM0VJHjbo0NoVPKfypKaD1USowwDtQ,1327
 arpakitlib/ar_sleep_util.py,sha256=ggaj7ML6QK_ADsHMcyu6GUmUpQ_9B9n-SKYH17h-9lM,1045
 arpakitlib/ar_sqladmin_util.py,sha256=SEoaowAPF3lhxPsNjwmOymNJ55Ty9rmzvsDm7gD5Ceo,861
-arpakitlib/ar_sqlalchemy_util.py,sha256=FDva9onjtCPrYZYIHHb93NMwD1WlmaORjiWgPRJQKoU,16180
+arpakitlib/ar_sqlalchemy_drop_check_constraints.py,sha256=XEqnMrIwSYasW_UOJ8xU-JhsVrcYeyehalFuSvmJMak,3518
+arpakitlib/ar_sqlalchemy_util.py,sha256=EMMaoVerrDaFXCCwH2BbbpaVr_oKp4nDlbnoJRJg_84,16404
 arpakitlib/ar_str_util.py,sha256=2lGpnXDf2h1cBZpVf5i1tX_HCv5iBd6IGnrCw4QWWlY,4350
 arpakitlib/ar_type_util.py,sha256=Cs_tef-Fc5xeyAF54KgISCsP11NHyzIsglm4S3Xx7iM,4049
 arpakitlib/ar_yookassa_api_client_util.py,sha256=VozuZeCJjmLd1zj2BdC9WfiAQ3XYOrIMsdpNK-AUlm0,5347
-arpakitlib-1.8.267.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.267.dist-info/METADATA,sha256=5U0ksCQqqwNpyS6KXV0K1iN-BCW2u_pV-oOQrhmSBOc,3919
-arpakitlib-1.8.267.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.267.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.267.dist-info/RECORD,,
+arpakitlib-1.8.269.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.269.dist-info/METADATA,sha256=bA31Hw4OoYrLzkLSB8hVlkHmS77vA_TS00PKwkJe28g,3919
+arpakitlib-1.8.269.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.269.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.269.dist-info/RECORD,,