From 175390fda7ae5453904516dc9ddca65d2c2f6a4d Mon Sep 17 00:00:00 2001 From: Valentijn Date: Thu, 19 Mar 2026 16:02:53 +0100 Subject: [PATCH] make it a bit more logical structure --- DB.py | 5 +++++ __pycache__/DB.cpython-313.pyc | Bin 2728 -> 3198 bytes __pycache__/PMT.cpython-313.pyc | Bin 7095 -> 6891 bytes main.py | 6 ++++-- 4 files changed, 9 insertions(+), 2 deletions(-) diff --git a/DB.py b/DB.py index c463752..2f50a7a 100644 --- a/DB.py +++ b/DB.py @@ -54,3 +54,8 @@ class Database: self._con.commit() 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") diff --git a/__pycache__/DB.cpython-313.pyc b/__pycache__/DB.cpython-313.pyc index 6d84096d06c6068b48395970c9d13b6e46b28744..57015c2ff48497910a1df74fd06e3c06f874ba54 100644 GIT binary patch delta 577 zcmZ1>`cH!IGcPX}0}$-u-;=qKeIj2Jqt3)kRURpZBKBamB93%+O^%IIt}-*OoSemC z&-h^TDwYk*jG~jHIds{if!c~>C--sKX~+SY#ZQ1l1H%J8q5jOy%n9yaJPhIgQzr8N_5hs7>zSG?=(pjB&~2KaBnL(VReI zz#y2(lo7&GU`S`sWcI7FaB=l<4RKX)3-b3>D9*@CD=Ah8_iznzg>d4FOA?Dp6zmj| zON)w9^Gf1NGILXbVz~vSvOt3#&1qF|NzF+uNlj5m%*jzmD=jH4N`;!E$yx-mxTszb zL<)fjVGyANB1C{hF%ytbP$=R7Qny4>z~;n5jEILg<(8~VVo73BVsWY-Hfc+cAyz=5 zf#C*b7;$j-b9ZuI5Yh3w!r^~|gQuUflk+-<#6=E?D;!dHICw7z=w9W}FX9L4*JK1c z)lZXi@-a>aHZ7n~k?v%6F4KH*AhV6(6B9El? E0Ma{`lK=n! delta 240 zcmew-u|kyZGcPX}0}!12v@6qsZ6aS3quRtwRkkAbbT&=)jkB&YGipvQV6kVsv3Uc_ z24+T)$%!1gY*IiCMKY78aM2f){D6mlax|wgn>>S< z%m<~(Q#cK{G#QIHf$ID;IVMlxb^scCi`z6u2q@CV@QI0;mG=`5Kdbat6){$4#v*>8 za1jfT&}1rNhp~$cKrFBn50G_>!zMRBr8FniuE-F`Wd!145g_q_nURt4HUr;xHho5C L#!n0&608dV<6}0H diff --git a/__pycache__/PMT.cpython-313.pyc b/__pycache__/PMT.cpython-313.pyc index ce3c65e8123f8e6a0a5f6df049de5482fd7bb3a1..59538e8338980da1c33914d6374c8fbfc3404d7a 100644 GIT binary patch delta 641 zcmYjOO=uHA6n?Y2vzg8QB-y_tt!QG>(jJVM5Q-vLQLtEC2}7_TJ&4ja2NQKI#e%dR z1P@-y+^wEG%ld~Pf}ZrWha41oY`u6%n|Sc%>>3*9FyEW+z3+XzH}7VC)%CuvCxM^E z-yd47SNd5AdF+Ao4ehXt*hk&qJ$@{(ABM)}$t7e5jN}meqI}kJT;pFr0GqJOwvt;& zVLd{Ic?*5*AN|Mzu?7!5g(9xPbrK!aqS9 zvwb_OtN$@_#PE3^Ik}AyeJDsYMM}=B0o=gcO{72#39cea+T<`}@r*RA(&MTWO-o<^ z&2WAg*Th(*jOJ+)UST#BYNaB?)U)G2rcR6rZsta9YRYK2SKb1se96l&o%Po9UZ=2n z=Vs@2tG!(CS6WN={lZdbwSB9yZYG1T?kqwNgJf8#&?tAwL~t$t5tT5m9vBn&JQ{eYirwA{0y(oUSa=cjMIQDAi_MI8GP43&LJcUuZdG7X3=N&{l!55`cVQc=({X<~u|YrCZz>@Ic}oAv@1 z5{+EU+(l1bH8q-$c=2TPQe$I`;m`w~@PI&!H=S0b?=bJpeBb+-_uf|Q%_(VDl5Bv= z=dE>h{H1h`8>~df6cbJ8S~guq*|e#koL)pFMVX(~b2)=zP1``ZqFz9nULxb%2e?W6 zyaB&dAMz^J2SMLZx7_0%NpzS)v&!{1E??R~*bL8`e!PkWz7pG9k-q1w<`&6%hRSE>bjlASt)4|5u-N|d#A^WT zbSiU9Bv^^xC@Q8|MhivSS%ExqZd^2!|!%XE_Tos*|yin3ab9`k3wixM?1psh)SOgH?4gc{IbvqZ8$tN z??BxV{p#>G*U1^F3;oqc!MCu-!k^f2BR&rIF2dt{VEYyX7SB&c=4aq8OU+$ICi7jo Prv%{)+yO0BY;Npd@o3K- diff --git a/main.py b/main.py index 3ebe55d..3c52fb4 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ from PMT import PMT from DB import Database from dotenv import load_dotenv +from time import sleep import os def main(): @@ -18,10 +19,11 @@ def main(): days_ahead = int(os.environ.get("DAYS_TO_FETCH")) 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)