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

fix: column alias bug fix (#2499)

Browse Source
This commit is contained in:
magic.chen
2025-03-21 15:50:43 +08:00
committed by GitHub
parent 73052eda75
commit ca0d1ecea3
4 changed files with 38 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/dbgpt-core/src/dbgpt/storage/metadata/db_storage.py
View File

@@ -3,7 +3,7 @@
from contextlib import contextmanager
from typing import Dict, Iterator, List, Optional, Type, Union
from sqlalchemy import URL
from sqlalchemy import URL, inspect
from sqlalchemy.orm import DeclarativeMeta, Session
from dbgpt.core import Serializer
@@ -20,11 +20,12 @@ from .db_manager import BaseModel, BaseQuery, DatabaseManager
def _copy_public_properties(src: BaseModel, dest: BaseModel):
"""Copy public properties from src to dest."""
for column in src.__table__.columns: # type: ignore
if column.name != "id":
value = getattr(src, column.name)
for column in inspect(src).mapper.column_attrs:
if column.key != "id":
alais = column.key
value = getattr(src, alais)
if value is not None:
setattr(dest, column.name, value)
setattr(dest, column.key, value)
class SQLAlchemyStorage(StorageInterface[T, BaseModel]):