Skip to content

Commit c577b68

Browse files
committed
Fix bug
1 parent 4fdc544 commit c577b68

3 files changed

Lines changed: 7 additions & 7 deletions

File tree

src/fastapi_quickcrud/crud_router.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,8 @@ def get_transaction_session():
9191
:param router_kwargs: Optional arguments that ``APIRouter().include_router`` takes.
9292
:return:
9393
"""
94-
NO_PRIMARY_KEY = False
9594

96-
db_model = convert_table_to_model(db_model)
95+
db_model, NO_PRIMARY_KEY = convert_table_to_model(db_model)
9796

9897
constraints = db_model.__table__.constraints
9998

src/fastapi_quickcrud/misc/utils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def sqlalchemy_to_pydantic(
104104
constraints = None,
105105
exclude_primary_key=False) -> CRUDModel:
106106

107-
db_model = convert_table_to_model(db_model)
107+
db_model, _ = convert_table_to_model(db_model)
108108
if exclude_columns is None:
109109
exclude_columns = []
110110
request_response_mode_set = {}
@@ -325,8 +325,9 @@ def group_by_foreign_key(item):
325325
return response_list
326326

327327
def convert_table_to_model(db_model):
328+
NO_PRIMARY_KEY = False
328329
if not isinstance(db_model, Table):
329-
return db_model
330+
return db_model, NO_PRIMARY_KEY
330331
db_name = str(db_model.fullname)
331332
table_dict = {'__table__': db_model,
332333
'__tablename__': db_name}
@@ -341,4 +342,4 @@ def convert_table_to_model(db_model):
341342
col, = i.expression.base_columns
342343
table_dict[str(i.key)] = col
343344

344-
return type(f'{db_name}DeclarativeBaseClass', (declarative_base(),), table_dict)
345+
return type(f'{db_name}DeclarativeBaseClass', (declarative_base(),), table_dict), NO_PRIMARY_KEY

tutorial/basic_usage/quick_usage_with_async_SQLALchemy_Base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,10 @@ class ExampleTable(Base):
5050
numeric_value = Column(Numeric)
5151
text_value = Column(Text, info={'alias_name': 'text_alias'})
5252
text_alias = synonym('text_value')
53-
time_value = Column(Time(False), default = datetime.now(timezone.utc).strftime('%H:%M:%S'))
53+
time_value = Column(Time(False), default=datetime.now(timezone.utc).strftime('%H:%M:%S'))
5454
timestamp_value = Column(DateTime(False), default=datetime.now())
5555
timestamptz_value = Column(DateTime(True), default=datetime.now(timezone.utc))
56-
timetz_value = Column(Time(True), default = datetime.now(timezone.utc).strftime('%H:%M:%S%z'))
56+
timetz_value = Column(Time(True), default=datetime.now(timezone.utc).strftime('%H:%M:%S%z'))
5757
uuid_value = Column(UUID(as_uuid=True))
5858
varchar_value = Column(String)
5959
array_value = Column(ARRAY(Integer()))

0 commit comments

Comments
 (0)