FrederikBaerentsen/BrickTracker
1
1
Fork 2
Code Issues 7 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Added test if retired sets not existing

Browse Source
This commit is contained in:
FrederikBaerentsen 2024-12-27 11:06:43 +01:00
parent 44cc91c26a
commit d650627710
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
app.py
View File

@ -547,24 +547,27 @@ def wishlist():
cursor.close()
conn.close()
retired_sets_dict = {}
with open('retired_sets.csv', mode='r', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
header = next(reader)
for row in reader:
key = row[2]
retired_sets_dict[key] = row
conn = sqlite3.connect('app.db')
cursor = conn.cursor()
cursor.execute('SELECT * from wishlist;')
results = cursor.fetchall()
wishlist = [list(i) for i in results]
for w in wishlist:
set_num = w[0].split('-')[0]
w.append(retired_sets_dict.get(set_num,[""]*7)[6])
retired_sets_dict = {}
try:
with open('retired_sets.csv', mode='r', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
header = next(reader)
for row in reader:
key = row[2]
retired_sets_dict[key] = row
for w in wishlist:
set_num = w[0].split('-')[0]
w.append(retired_sets_dict.get(set_num,[""]*7)[6])
except:
print('No retired list')
if wishlist == None or wishlist == '':
wishlist = ''
conn.commit()