@@ -145,7 +145,7 @@ def __init__(self, db_model: Type, sql_type, exclude_column=None, constraints=No
145145 self ._exclude_column = exclude_column
146146 self .alias_mapper : Dict [str , str ] = {} # Table not support alias
147147 if self .exclude_primary_key :
148- self .__db_model : Table = db_model . __table__
148+ self .__db_model : Table = db_model
149149 self .__db_model_table : Table = db_model .__table__
150150 self .__columns = db_model .__table__ .c
151151 self .db_name : str = db_model .__tablename__
@@ -154,6 +154,7 @@ def __init__(self, db_model: Type, sql_type, exclude_column=None, constraints=No
154154 self .__db_model_table : Table = db_model .__table__
155155 self .db_name : str = db_model .__tablename__
156156 self .__columns = db_model .__table__ .c
157+ model = self .__db_model
157158 self .primary_key_str , self ._primary_key_dataclass_model , self ._primary_key_field_definition \
158159 = self ._extract_primary ()
159160 self .unique_fields : List [str ] = self ._extract_unique ()
@@ -496,7 +497,6 @@ def _extra_relation_level(self, model=None, processed_table=None) -> Dict[str, T
496497 relation_level += self ._extra_relation_level (self .foreign_mapper [relation_table ]["db_model" ],
497498 processed_table = processed_table
498499 )
499- print (relation_level )
500500 return relation_level
501501
502502 def _extra_foreign_table_from_declarative_base (self , model ) -> Dict [str , Table ]:
0 commit comments