FrederikBaerentsen/BrickTracker
4
3
Fork 3
Code Issues 9 Pull Requests 1 Actions Packages Projects 1 Releases Wiki Activity

Added retirement info to wishlist

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

14
app.py
View File

@ -5,7 +5,7 @@ from flask_socketio import SocketIO
from threading import Thread from threading import Thread
from pprint import pprint as pp from pprint import pprint as pp
from pathlib import Path from pathlib import Path
import time,random,string,sqlite3 import time,random,string,sqlite3,csv
import numpy as np import numpy as np
import re #regex import re #regex
import rebrick #rebrickable api import rebrick #rebrickable api
@ -547,12 +547,24 @@ def wishlist():
cursor.close() cursor.close()
conn.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') conn = sqlite3.connect('app.db')
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute('SELECT * from wishlist;') cursor.execute('SELECT * from wishlist;')
results = cursor.fetchall() results = cursor.fetchall()
wishlist = [list(i) for i in results] 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])
if wishlist == None or wishlist == '': if wishlist == None or wishlist == '':
wishlist = '' wishlist = ''
conn.commit() conn.commit()

2
templates/wishlist.html
View File

@ -234,6 +234,7 @@ background-color: white;
<th class="fixed-width">Name</th> <th class="fixed-width">Name</th>
<th style="text-align: center;" class="fixed-width hidden-mobile">Year</th> <th style="text-align: center;" class="fixed-width hidden-mobile">Year</th>
<th style="text-align: center;" class="fixed-width hidden-mobile">Parts</th> <th style="text-align: center;" class="fixed-width hidden-mobile">Parts</th>
<th class="fixed-width hidden-mobile">Retirement Date</th>
<th style="text-align: center;" class="sorttable_nosort">Delete</th> <th style="text-align: center;" class="sorttable_nosort">Delete</th>
</tr> </tr>
</thead> </thead>
@ -245,6 +246,7 @@ background-color: white;
<td>{{ sets[1] }}</td> <td>{{ sets[1] }}</td>
<td style="text-align:center;" class="hidden-mobile">{{ sets[2] }}</td> <td style="text-align:center;" class="hidden-mobile">{{ sets[2] }}</td>
<td style="text-align:center;" class="hidden-mobile">{{ sets[4] }}</td> <td style="text-align:center;" class="hidden-mobile">{{ sets[4] }}</td>
<td class="hidden-mobile">{{ sets[8] }}</td>
<td style="text-align:center;"> <td style="text-align:center;">
<form method="POST" style="margin:0"> <form method="POST" style="margin:0">
<input type="hidden" name="set_num" value="{{ sets[0] }}"> <input type="hidden" name="set_num" value="{{ sets[0] }}">