From 613a8f1cbfe80aea6ff4e219c7706a13f77b58aa Mon Sep 17 00:00:00 2001 From: tanwei Date: Sun, 6 Jul 2025 11:30:52 +0800 Subject: [PATCH] format code --- .../src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py index 9085ddb58..c56b3464f 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py @@ -1,11 +1,11 @@ """GaussDB connector.""" + import logging from dataclasses import dataclass, field from typing import Any, Iterable, List, Optional, Tuple, Type, cast from urllib.parse import quote from urllib.parse import quote_plus as urlquote -import psycopg2 from sqlalchemy import text from dbgpt.core.awel.flow import ( @@ -19,14 +19,11 @@ from dbgpt.util.i18n_utils import _ logger = logging.getLogger(__name__) - @auto_register_resource( label=_("GaussDB datasource"), category=ResourceCategory.DATABASE, tags={"order": TAGS_ORDER_HIGH}, - description=_( - "Huawei's distributed database with PostgreSQL/MySQL compatibility." - ), + description=_("Huawei's distributed database with PostgreSQL/MySQL compatibility."), ) @dataclass class GaussDBParameters(RDBMSDatasourceParameters): @@ -79,6 +76,7 @@ class GaussDBConnector(RDBMSConnector): f"{cls.driver}://{quote(user)}:{urlquote(pwd)}@{host}:{str(port)}/{db_name}" ) from sqlalchemy.dialects.postgresql.base import PGDialect + PGDialect._get_server_version_info = lambda *args: (7, 0) return cast(GaussDBConnector, cls.from_uri(db_url, engine_args, **kwargs))