Added retirement info to wishlist

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

14
app.py
View File

@ -5,7 +5,7 @@ from flask_socketio import SocketIO
from threading import Thread
from pprint import pprint as pp
from pathlib import Path
import time,random,string,sqlite3
import time,random,string,sqlite3,csv
import numpy as np
import re #regex
import rebrick #rebrickable api
@ -547,12 +547,24 @@ 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])
if wishlist == None or wishlist == '':
wishlist = ''
conn.commit()

View File

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