cal/major-domo-database
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update db_engine.py

Browse Source 
Handle count() == 0 bug
This commit is contained in:
Cal Corum 2023-09-15 00:36:22 -05:00
parent 5fae0a30df
commit 63641dafb3
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
app/db_engine.py
View File

@ -43,9 +43,6 @@ WEEK_NUMS = {
def model_csv_headers(this_obj, exclude=None) -> List:
if this_obj is None:
return ['None']
data = model_to_dict(this_obj, recurse=False, exclude=exclude)
return [x for x in data.keys()]
@ -56,9 +53,12 @@ def model_to_csv(this_obj, exclude=None) -> List:
def query_to_csv(all_items: ModelSelect, exclude=None):
data_list = [model_csv_headers(all_items[0], exclude=exclude)]
for x in all_items:
data_list.append(model_to_csv(x, exclude=exclude))
if all_items.count() == 0:
data_list = [['No data found']]
else:
data_list = [model_csv_headers(all_items[0], exclude=exclude)]
for x in all_items:
data_list.append(model_to_csv(x, exclude=exclude))
return DataFrame(data_list).to_csv(header=False, index=False)