Skip to content

Commit 1b8e052

Browse files
committed
fix model type handling
1 parent 8058dfb commit 1b8e052

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/fastapi_quickcrud/misc/schema_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)