arpakitlib 1.8.266__py3-none-any.whl → 1.8.268__py3-none-any.whl

arpakitlib/ar_clone_pydantic_model_fields.py

@@ -10,7 +10,7 @@ _ARPAKIT_LIB_MODULE_VERSION = "3.0"
 def clone_pydantic_model_fields(
         *,
         model_cls: Type[BaseModel],
-        base_model: Type[BaseModel],
+        base_model: Type[BaseModel] = BaseModel,
         fields_to_remove: Iterable[str] | None = None,
         new_class_name: str | None = None,
         class_name_prefix: str | None = "Cloned"

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.266.dist-info → arpakitlib-1.8.268.dist-info}/METADATA

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

{arpakitlib-1.8.266.dist-info → arpakitlib-1.8.268.dist-info}/RECORD

@@ -388,7 +388,7 @@ arpakitlib/ar_base_worker_util.py,sha256=e-Vj9w1D-59KN3Zz7TQlOB2lW4fZphJlsPpFkmX
 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
-arpakitlib/ar_clone_pydantic_model_fields.py,sha256=Tz609Z-HH23bDxIBPbHEeKwMLoLtrS9qHntEUe7ClDc,1441
+arpakitlib/ar_clone_pydantic_model_fields.py,sha256=5i77NGEjnY2ppk_Ot179egQGNDvg7zre8NBl5Es1nkg,1453
 arpakitlib/ar_datetime_util.py,sha256=Xe1NiT9oPQzNSG7RVRkhukhbg4i-hhS5ImmV7sPUc8o,971
 arpakitlib/ar_dict_util.py,sha256=oet-9AJEjQZfG_EI82BuYW0jdW2NQxKjPXol_nfTXjw,447
 arpakitlib/ar_dream_ai_api_client_util.py,sha256=Z1oii_XSsgunYx17SdcHl54S4KPQti7-MJs0xXZ8vL0,2830
@@ -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.266.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.266.dist-info/METADATA,sha256=7Bv4WfJmbcKL7mPKzo_ubrItgW1icyVCL-N__viFjoY,3919
-arpakitlib-1.8.266.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.266.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.266.dist-info/RECORD,,
+arpakitlib-1.8.268.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.268.dist-info/METADATA,sha256=loF6G97KNy-SAoM8vY2UPZam4JHIjGfdBb0hwjKkUTI,3919
+arpakitlib-1.8.268.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.268.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.268.dist-info/RECORD,,