lesscode-database 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

lesscode_database/connect_pool.py

@@ -674,6 +674,36 @@ class Pool:
                                   **params)
         return pool
 
+    @staticmethod
+    def sync_create_dm_pool(conn_info: ConnectionInfo):
+        try:
+            dmPython = importlib.import_module("dmPython")
+        except ImportError:
+            raise Exception(f"dmPython is not exist,run:pip install dmPython")
+        try:
+            pooled_db = importlib.import_module("dbutils.pooled_db")
+        except ImportError:
+            raise Exception(f"DBUtils is not exist,run:pip install DBUtils==3.0.2")
+        params = conn_info.params
+        if not isinstance(params, dict):
+            params = dict()
+        blocking = params.pop("blocking", True)
+        mincached = params.pop("mincached", conn_info.min_size)
+        maxusage = params.pop("maxusage", conn_info.min_size)
+        maxshared = params.pop("maxshared", conn_info.max_size)
+        maxcached = params.pop("maxcached", conn_info.max_size)
+        ping = params.pop("ping", 1)
+        autocommit = params.pop("autocommit", True)
+        pool = pooled_db.PooledDB(creator=dmPython, host=conn_info.host, port=conn_info.port,
+                                  user=conn_info.user,
+                                  password=conn_info.password, schema=conn_info.db_name,
+                                  mincached=mincached, blocking=blocking, maxusage=maxusage,
+                                  maxshared=maxshared, maxcached=maxcached,
+                                  ping=ping, maxconnections=conn_info.max_size,
+                                  autoCommit=autocommit,
+                                  **params)
+        return pool
+
     @staticmethod
     def create_sqlalchemy_pool(conn_info: ConnectionInfo):
         """
@@ -734,10 +764,10 @@ class Pool:
         :param conn_info: 连接信息
         :return:
         """
+        db_type = "mysql"
         if conn_info.dsn:
             url = conn_info.dsn
         else:
-            db_type = "mysql"
             if conn_info.params:
                 if conn_info.params.get("db_type"):
                     db_type = conn_info.params.pop("db_type")
@@ -771,23 +801,35 @@ class Pool:
                 url = "oracle+cx_oracle://{}:{}@{}:{}/{}?charset=utf8mb4".format(
                     conn_info.user, conn_info.password, conn_info.host, conn_info.port,
                     conn_info.db_name)
+            elif db_type == "dm":
+                url = 'dm+dmPython://{}:{}@{}:{}'.format(
+                    conn_info.user, conn_info.password, conn_info.host, conn_info.port)
             else:
                 raise ImportError("UNSUPPORTED DB TYPE")
         try:
             sqlalchemy = importlib.import_module("sqlalchemy")
         except ImportError:
             raise ImportError(f"sqlalchemy is not exist,run:pip install sqlalchemy")
-        engine = sqlalchemy.create_engine(url, echo=conn_info.params.get("echo",
-                                                                         True) if conn_info.params else True,
+        params = conn_info.params
+        if not params:
+            params = dict()
+        echo = params.pop("echo", True)
+        pool_recycle = params.pop("pool_recycle", 3600)
+        max_overflow = params.pop("max_overflow", 0)
+        pool_timeout = params.pop("pool_timeout", 10)
+        pool_pre_ping = params.pop("pool_pre_ping", False)
+        if conn_info.db_name and db_type=="dm":
+            if "connect_args" not in params:
+                params["connect_args"] = {"schema": conn_info.db_name}
+            else:
+                params["connect_args"]["schema"] = conn_info.db_name
+
+        engine = sqlalchemy.create_engine(url, echo=echo,
                                           pool_size=conn_info.min_size,
-                                          pool_recycle=conn_info.params.get("pool_recycle",
-                                                                            3600) if conn_info.params else 3600,
-                                          max_overflow=conn_info.params.get("max_overflow",
-                                                                            0) if conn_info.params else 0,
-                                          pool_timeout=conn_info.params.get("pool_timeout",
-                                                                            10) if conn_info.params else 10,
-                                          pool_pre_ping=conn_info.params.get("pool_pre_ping",
-                                                                             True) if conn_info.params else True)
+                                          pool_recycle=pool_recycle,
+                                          max_overflow=max_overflow,
+                                          pool_timeout=pool_timeout,
+                                          pool_pre_ping=pool_pre_ping)
         return engine
 
     @staticmethod
@@ -939,6 +981,12 @@ def get_pool(conn_info: ConnectionInfo):
         else:
             return run_sync(Pool.sync_create_clickhouse_pool(conn_info)), conn_info
 
+    elif conn_info.dialect == "dm":
+        if conn_info.async_enable:
+            return run_sync(Pool.sync_create_dm_pool(conn_info)), conn_info
+        else:
+            return run_sync(Pool.sync_create_dm_pool(conn_info)), conn_info
+
     elif conn_info.dialect == "sqlalchemy":
         if conn_info.async_enable:
             return run_sync(Pool.create_sqlalchemy_pool(conn_info)), conn_info

lesscode_database/version.py

@@ -1 +1 @@
-__version__ = "0.0.9"
+__version__ = "0.0.10"

{lesscode_database-0.0.9.dist-info → lesscode_database-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lesscode-database
-Version: 0.0.9
+Version: 0.0.10
 Summary: lesscode_database是数据库连接工具包
 Author: navysummer
 Author-email: navysummer@yeah.net
@@ -57,13 +57,15 @@ db_options.conn_list = [
     ConnectionInfo(dialect="redis", name="redis", host="127.0.0.1", port=6379, user=None,
                    password=None, db_name=1, enable=False, async_enable=True),
     # sqlalchemy 异步支持mysql，postgresql，tidb，ocean_base，doris;
-    # 同步支持mysql，postgresql，tidb，ocean_base，doris，sqlite3，mssql，oracle
+    # 同步支持mysql，postgresql，tidb，ocean_base，doris，sqlite3，mssql，oracle,dm
     ConnectionInfo(dialect="sqlalchemy", name="sqlalchemy", host="127.0.0.1", port=3306, user="root",
                    password="root", db_name="test", enable=False, async_enable=False, params={"db_type": "mysql"}),
     ConnectionInfo(dialect="neo4j", name="neo4j", host="127.0.0.1", port=7687, user="neo4j",
                    password="neo4j", db_name=None, enable=False, async_enable=True),
     ConnectionInfo(dialect="clickhouse", name="clickhouse", dsn="clickhouse://localhost", host="127.0.0.1", port=9000,
                    user="default", password="", db_name='', enable=True, async_enable=False),
+    ConnectionInfo(dialect="dm", name="dm", dsn=None, host="127.0.0.1", port=5236,
+                   user="default", password="", db_name='', enable=True, async_enable=False),
 
 ]
 
@@ -358,4 +360,12 @@ async def async_esapi_test():
 loop = asyncio.get_event_loop()
 loop.run_until_complete(async_esapi_test())
 
+
+# 达梦数据库 同步测试
+with DsHelper("dm").pool.dedicated_connection() as conn:
+    with conn.cursor() as cursor:
+        cursor.execute('SELECT 1')
+        print(cursor.fetchall())
+
+
 ```

{lesscode_database-0.0.9.dist-info → lesscode_database-0.0.10.dist-info}/RECORD

@@ -1,14 +1,14 @@
 lesscode_database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lesscode_database/connect_pool.py,sha256=_RdmtKmacohFP2FwEet_4HY05Vr5zGlwj8lkFnfmK64,44572
+lesscode_database/connect_pool.py,sha256=6iFUU8w4sNs2HgGiH5M0PZI7wuEUnlNgzx-hT65USyo,46467
 lesscode_database/connection_info.py,sha256=HMlA-hA0LE7jGUS8crZAgrBMB-o6B9mxrWYEywgqXHg,1307
 lesscode_database/db_options.py,sha256=Z8Iy3pqDYW5YVVfvoFSsAYecK9LfiwRasYc3fzqIBt8,1928
 lesscode_database/db_request.py,sha256=siZGXIdPuTlY4NwVLnU8098bVM_6jgQzSzeO4BzkDrA,4832
 lesscode_database/ds_helper.py,sha256=W3VrTe_l0mGX0uEbN7UI6NlQbruLpr2WXgO7AexzQ-k,4845
 lesscode_database/dynamic_import_package.py,sha256=J8hgGRHe6KrprOgOq-xbKHVAYHSjUBcNyTaSPvBmvIk,164
 lesscode_database/mongo_base_model.py,sha256=cKHNmi5GJBGRxCtqsggEcrzIZK7__Z_4cVaIHxmqZsc,6406
-lesscode_database/version.py,sha256=46Yjk3fz9o8aTN8E95McnzpJcjGzVJmHmQqUZ5mXzfc,22
-lesscode_database-0.0.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lesscode_database-0.0.9.dist-info/METADATA,sha256=v1FcYg9dFAyZ4-U_t34STUYQWhF4QpEabH0aQRfk1jg,11692
-lesscode_database-0.0.9.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-lesscode_database-0.0.9.dist-info/top_level.txt,sha256=h6cg13be6kkDfNaX9nWeDcfducTb5bKG5iYRO2JPmAM,18
-lesscode_database-0.0.9.dist-info/RECORD,,
+lesscode_database/version.py,sha256=-nNlMKS9nph3FR78_ZG9RGKrbxseeNp2K6nMr0pVGaU,23
+lesscode_database-0.0.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lesscode_database-0.0.10.dist-info/METADATA,sha256=w3pRRvHPOi0bcIoGKVKlmNKtVIWpYFyrIXflWvEL5Ns,12066
+lesscode_database-0.0.10.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+lesscode_database-0.0.10.dist-info/top_level.txt,sha256=h6cg13be6kkDfNaX9nWeDcfducTb5bKG5iYRO2JPmAM,18
+lesscode_database-0.0.10.dist-info/RECORD,,