valentijn/PMT-Rooster-Fetcher
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make it a bit more logical structure

Browse Source
This commit is contained in:
Valentijn
2026-03-19 16:02:53 +01:00
parent 47bb6585d1
commit 175390fda7
4 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
DB.py
View File

@@ -54,3 +54,8 @@ class Database:
self._con.commit() self._con.commit()
print(f"✅ Inserted {inserted}/{len(shifts)} new shifts") print(f"✅ Inserted {inserted}/{len(shifts)} new shifts")
def delete_future_shifts(self):
cur = self._con.cursor()
cur.execute("DELETE FROM shifts WHERE shift_start > current_timestamp")
self._con.commit()
print(f"✅ Deleted all future shifts")

BIN
__pycache__/DB.cpython-313.pyc
View File

Binary file not shown.

BIN
__pycache__/PMT.cpython-313.pyc
View File

Binary file not shown.

6
main.py
View File

@@ -1,6 +1,7 @@
from PMT import PMT from PMT import PMT
from DB import Database from DB import Database
from dotenv import load_dotenv from dotenv import load_dotenv
from time import sleep
import os import os
def main(): def main():
@@ -18,10 +19,11 @@ def main():
days_ahead = int(os.environ.get("DAYS_TO_FETCH")) days_ahead = int(os.environ.get("DAYS_TO_FETCH"))
shifts = pmt.get_shifts(days_ahead) shifts = pmt.get_shifts(days_ahead)
PMT.print_shifts(shifts)
db.delete_future_shifts() # delete all future shifts before importing, to make sure if a shift is removed, it actually gets removed here too.
# Insert those shifts into the database # Insert the new shifts into the database
db.insert_shifts(shifts) db.insert_shifts(shifts)