codosdk 1.0.46__tar.gz → 1.0.47__tar.gz

{codosdk-1.0.46 → codosdk-1.0.47}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codosdk
-Version: 1.0.46
+Version: 1.0.47
 Summary: CODO项目的Python SDK
 Home-page: https://github.com/ss1917/codo_sdk/
 Author: shenshuo

{codosdk-1.0.46 → codosdk-1.0.47}/codosdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codosdk
-Version: 1.0.46
+Version: 1.0.47
 Summary: CODO项目的Python SDK
 Home-page: https://github.com/ss1917/codo_sdk/
 Author: shenshuo

{codosdk-1.0.46 → codosdk-1.0.47}/codosdk.egg-info/SOURCES.txt

@@ -48,4 +48,5 @@ websdk2/cloud/ucloud_api.py
 websdk2/utils/__init__.py
 websdk2/utils/cc_crypto.py
 websdk2/utils/date_format.py
-websdk2/utils/pydantic_utils.py
+websdk2/utils/pydantic_utils.py
+websdk2/utils/pydantic_utils_v2.py

{codosdk-1.0.46 → codosdk-1.0.47}/setup.py

@@ -9,7 +9,7 @@ desc   : CODO SDK
 import sys
 from setuptools import setup, find_packages
 
-VERSION = '1.0.46'
+VERSION = '1.0.47'
 
 if sys.version_info < (2, 7) or (3, 0) <= sys.version_info < (3, 6):
     print('This program requires at least Python 2.7 or 3.6 to run.')

codosdk-1.0.47/websdk2/utils/pydantic_utils_v2.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Version : 0.0.9
+Contact : 191715030@qq.com
+Author  : shenshuo
+Date    : 2024/12/30 20:28
+Desc    : https://github.com/tiangolo/pydantic-sqlalchemy
+"""
+
+####
+from typing import Container, Optional, Type
+from pydantic import BaseModel, create_model, ValidationError, ConfigDict
+
+
+#  删除的时候一般只有id
+class PydanticDel(BaseModel):
+    id: int
+
+
+class PydanticDelList(BaseModel):
+    id_list: list[int]
+
+
+def sqlalchemy_to_pydantic(db_model: Type, *, exclude: Container[str] = ()) -> Type[BaseModel]:
+    table = db_model.metadata.tables[db_model.__tablename__]
+    fields = {}
+    config = ConfigDict(from_attributes=True)
+    for column in table.columns:
+        name = column.name
+        if name in exclude:
+            continue
+        python_type: Optional[type] = None
+        if hasattr(column.type, "impl"):
+            if hasattr(column.type.impl, "python_type"):
+                python_type = column.type.impl.python_type
+        elif hasattr(column.type, "python_type"):
+            python_type = column.type.python_type
+        assert python_type, f"Could not infer python_type for {column}"
+
+        default = ... if column.default is None and not column.nullable else None
+
+        fields[name] = (python_type, default)
+
+    return create_model(db_model.__name__, __config__=config, **fields)