github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-25 04:53:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

format code

Browse Source
This commit is contained in:
tanwei 2025-07-06 11:30:52 +08:00
parent 40fc8a664d
commit 613a8f1cbf
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_gaussdb.py
View File

@ -1,11 +1,11 @@
"""GaussDB connector.""" """GaussDB connector."""
import logging import logging
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Any, Iterable, List, Optional, Tuple, Type, cast from typing import Any, Iterable, List, Optional, Tuple, Type, cast
from urllib.parse import quote from urllib.parse import quote
from urllib.parse import quote_plus as urlquote from urllib.parse import quote_plus as urlquote
import psycopg2
from sqlalchemy import text from sqlalchemy import text
from dbgpt.core.awel.flow import ( from dbgpt.core.awel.flow import (
@ -19,14 +19,11 @@ from dbgpt.util.i18n_utils import _
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@auto_register_resource( @auto_register_resource(
label=_("GaussDB datasource"), label=_("GaussDB datasource"),
category=ResourceCategory.DATABASE, category=ResourceCategory.DATABASE,
tags={"order": TAGS_ORDER_HIGH}, tags={"order": TAGS_ORDER_HIGH},
description=_( description=_("Huawei's distributed database with PostgreSQL/MySQL compatibility."),
"Huawei's distributed database with PostgreSQL/MySQL compatibility."
),
) )
@dataclass @dataclass
class GaussDBParameters(RDBMSDatasourceParameters): class GaussDBParameters(RDBMSDatasourceParameters):
@ -79,6 +76,7 @@ class GaussDBConnector(RDBMSConnector):
f"{cls.driver}://{quote(user)}:{urlquote(pwd)}@{host}:{str(port)}/{db_name}" f"{cls.driver}://{quote(user)}:{urlquote(pwd)}@{host}:{str(port)}/{db_name}"
) )
from sqlalchemy.dialects.postgresql.base import PGDialect from sqlalchemy.dialects.postgresql.base import PGDialect
PGDialect._get_server_version_info = lambda *args: (7, 0) PGDialect._get_server_version_info = lambda *args: (7, 0)
return cast(GaussDBConnector, cls.from_uri(db_url, engine_args, **kwargs)) return cast(GaussDBConnector, cls.from_uri(db_url, engine_args, **kwargs))