Skip to content

Commit 9f210f7

Browse files
committed
Refactored check for db file
1 parent 52f208b commit 9f210f7

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

Diff for: main.py

+8-8
Original file line numberDiff line numberDiff line change
@@ -26,17 +26,17 @@ def main():
2626
goto_waitinglist_page()
2727

2828
persons_on_waiting_list = extract_persons_on_waiting_list()
29+
30+
if not os.path.exists(PATH_TO_DB):
31+
create_and_seed_db(persons_on_waiting_list)
32+
exit()
33+
2934
saved_persons_on_waiting_list = fetch_saved_waiting_list_persons()
3035

31-
# TODO: This whole section sucks ass - ie. it's confusing - so refactor that thang ..
3236
persons_not_in_saved_waiting_list = []
33-
if os.path.exists(PATH_TO_DB):
34-
for person in persons_on_waiting_list:
35-
if person not in saved_persons_on_waiting_list:
36-
persons_not_in_saved_waiting_list.append(person)
37-
38-
else:
39-
create_and_seed_db(persons_on_waiting_list)
37+
for person in persons_on_waiting_list:
38+
if person not in saved_persons_on_waiting_list:
39+
persons_not_in_saved_waiting_list.append(person)
4040

4141
if persons_not_in_saved_waiting_list:
4242
save_new_persons_to_db(persons_not_in_saved_waiting_list)

0 commit comments

Comments
 (0)