From ca82885fb776759c419f3e1463160417c896eead Mon Sep 17 00:00:00 2001
From: Valentijn <valentijn@herpiederpiee.nl>
Date: Wed, 18 Mar 2026 19:49:14 +0100
Subject: [PATCH] first commit

---
 .gitignore                      |   2 +
 API.py                          |   0
 DB.py                           |  52 +++++++++++++
 PMT.py                          | 127 ++++++++++++++++++++++++++++++++
 __pycache__/DB.cpython-313.pyc  | Bin 0 -> 2728 bytes
 __pycache__/PMT.cpython-313.pyc | Bin 0 -> 7095 bytes
 main.py                         |  24 ++++++
 7 files changed, 205 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 API.py
 create mode 100644 DB.py
 create mode 100644 PMT.py
 create mode 100644 __pycache__/DB.cpython-313.pyc
 create mode 100644 __pycache__/PMT.cpython-313.pyc
 create mode 100644 main.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..4b95af1
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+.env
+shifts.db
\ No newline at end of file
diff --git a/API.py b/API.py
new file mode 100644
index 0000000..e69de29
diff --git a/DB.py b/DB.py
new file mode 100644
index 0000000..429328b
--- /dev/null
+++ b/DB.py
@@ -0,0 +1,52 @@
+import sqlite3
+
+
+class Database:
+    def __init__(self):
+        self._con = sqlite3.connect("shifts.db")
+        self._setup_tables() 
+        pass
+    
+
+    def _setup_tables(self):
+        cur = self._con.cursor()
+        cur.execute('''
+        CREATE TABLE IF NOT EXISTS shifts (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            shift_start DATETIME NOT NULL,
+            shift_end DATETIME NOT NULL,
+            department TEXT NOT NULL,
+            duration TEXT NOT NULL,
+            description TEXT,
+            fetched_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            
+            UNIQUE(shift_start, department)  -- No duplicates
+        );
+        ''')
+        self._con.commit()
+
+
+    def insert_shifts(self, shifts):
+        cur = self._con.cursor()
+        inserted = 0
+
+        for shift in shifts:
+            start = shift.get('start_datetime', 'N/A')
+            end = shift.get('end_datetime', 'N/A')
+
+            department = shift.get('department_name', 'N/A')
+            duration = shift.get('duration', 'N/A')
+            description = shift.get('shift_remark', shift.get('description', 'N/A'))
+            
+            cur.execute("""
+                INSERT OR IGNORE INTO shifts 
+                (shift_start, shift_end, department, duration, description)
+                VALUES (?, ?, ?, ?, ?)
+            """, (start, end, department, duration, description))
+            
+            if cur.rowcount > 0:
+                inserted += 1
+    
+        self._con.commit()
+        print(f"✅ Inserted {inserted}/{len(shifts)} new shifts")
+            
diff --git a/PMT.py b/PMT.py
new file mode 100644
index 0000000..4f748ad
--- /dev/null
+++ b/PMT.py
@@ -0,0 +1,127 @@
+import os
+import requests
+import json
+from datetime import datetime, timedelta
+import sys
+
+class PMT:
+    def __init__(self):
+        self.session = None
+        self.context_token = None
+        self.user_token = None
+
+        self._username = os.environ.get("USERNAME_PMT")
+        self._password = os.environ.get("PASSWORD_PMT")
+        self.base_url = os.environ.get("URL_PMT")
+
+    def get_week_range(self, week_offset):
+        """Get Monday date for week_offset weeks from now"""
+        today = datetime.now()
+        days_to_monday = 0 if today.weekday() == 0 else 7 - today.weekday()
+        monday = today + timedelta(days=days_to_monday)
+        target_monday = monday + timedelta(weeks=week_offset)
+        return target_monday.strftime("%Y-%m-%d")
+
+    def login(self):
+        self.session = requests.Session()
+        
+        login_url = f"{self.base_url}/login"
+        resp = self.session.get(login_url)
+        
+        # Step 2: SSO login POST
+        sso_data = {
+            "username": self._username,
+            "password": self._password,
+            "browserinfo": {
+                "osname": "Linux 148",
+                "browsername": "Firefox",
+                "browserversion": "148",
+                "screenresolution": "1080x1920"
+            }
+        }
+        
+        sso_resp = self.session.post(f"{self.base_url}/pmtLoginSso", 
+                                   json=sso_data,
+                                   headers={'Content-Type': 'application/json'})
+        
+        if sso_resp.status_code != 200 or not sso_resp.json().get('result', {}).get('authenticated'):
+            print("Login failed!")
+            print(sso_resp.text)
+            sys.exit(1)
+        
+        login_data = sso_resp.json()['result']
+        self.context_token = login_data['context_token']
+        self.user_token = login_data['user_token']
+        
+        print(f"✅ Logged in as {self._username} (store AH 8541 Rhoon)")
+        return self.session, self.context_token, self.user_token
+
+    def get_shifts(self, days_ahead=14):
+        """Fetch shifts from now to days_ahead using the REAL endpoint"""
+        if not self.session or not self.context_token or not self.user_token:
+            raise ValueError("Must call login() first!")
+            
+        shifts = []
+        
+        from_date = datetime.now().strftime("%Y-%m-%d")
+        to_date = (datetime.now() + timedelta(days=days_ahead)).strftime("%Y-%m-%d")
+        
+        print(f"Fetching shifts from {from_date} to {to_date}...")
+        
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:148.0) Gecko/20100101 Firefox/148.0',
+            'Accept': 'application/json, text/plain, */*',
+            'Accept-Language': 'en-US,en;q=0.9',
+            'Accept-Encoding': 'gzip, deflate, br, zstd',
+            'Referer': 'https://ah8541.personeelstool.nl/my-overview/my-schedule/11-2026',
+            'Content-Type': 'application/json',
+            'x-api-user': self.user_token,
+            'x-api-context': self.context_token,
+            'Connection': 'keep-alive',
+            'Sec-Fetch-Dest': 'empty',
+            'Sec-Fetch-Mode': 'cors',
+            'Sec-Fetch-Site': 'same-origin'
+        }
+        
+        shifts_url = f"{self.base_url}/api/v3/environments/21/stores/311/employee/98104/simpleShifts"
+        params = {
+            'from_date': from_date,
+            'to_date': to_date
+        }
+        
+        resp = self.session.get(shifts_url, headers=headers, params=params)
+        
+        if resp.status_code == 200:
+            data = resp.json()
+            week_shifts = data.get('result', {}).get('shift_instances', [])
+            shifts.extend(week_shifts)
+            print(f"✅ Got {len(week_shifts)} shifts!")
+        else:
+            print(f"❌ Failed: {resp.status_code}")
+            print(resp.text[:500])
+        
+        return shifts
+
+    @classmethod
+    def print_shifts(cls, shifts):
+        """Pretty print shifts (class method since it doesn't need instance)"""
+        if not shifts:
+            print("No upcoming shifts found.")
+            return
+        
+        print("\n📅 UPCOMING SHIFTS:\n")
+        print("Date       | Time          | Dept        | Duration | Description")
+        print("-" * 80)
+        
+        for shift in sorted(shifts, key=lambda x: x.get('start_datetime', '')):
+            start = shift.get('start_datetime', 'N/A')
+            end = shift.get('end_datetime', 'N/A')
+            dept = shift.get('department_name', 'N/A')
+            duration = shift.get('duration', 'N/A')
+            desc = shift.get('shift_remark', shift.get('description', 'N/A'))
+            
+            date_str = start[:10] if start != 'N/A' else 'N/A'
+            time_str = f"{start[11:16]}-{end[11:16]}" if start != 'N/A' and end != 'N/A' else 'N/A'
+            
+            print(f"{date_str} | {time_str:<12} | {dept:<10} | {duration:<8} | {desc[:40]}")
+
diff --git a/__pycache__/DB.cpython-313.pyc b/__pycache__/DB.cpython-313.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..6d84096d06c6068b48395970c9d13b6e46b28744
GIT binary patch
literal 2728
zcmbVO-EZ4e6u*v>)M-AtwcXaHrMnD-W@&8(+W@f%NLz0UY26XqQ@0{nS>m=HiPK#>
z`>6J?7f>o8N)c5%fwVprfl!f<K>P*p)+w7-nLSJ!FMBKP5U-r;e0L3#xJs_iJ^r2J
zbI(2JXKx3Ctq8{NTfgdWdl7m<7nR{I3)>ikMI<4Ki6er#&><wb!boCIxJn4ZOfkIs
zDP4%ElBzT_<;C`EAS|Mf5Md-lToOZA>Vdc=7G{s+hS@85NXwL$Z?U=Iq?NpsG_|KR
zf>gx`^~IL|+ZRDtL=#l!j6^w<%9>swOy)${T0xmALP3d^dR$39QC49zg?N`Z>oceI
zNy{8aU*g?%i+LrZTiP+ZC6zM_EoIqm6(l>TnwpiLQ6b)pX7Y^fHnq&8Ngs;slM#|P
zBGWn3id;<+5V9Hw8O^Zt%SL2ayQ*b!Gg(+h&WR)aL-bBsBaz|3ftedNr>eT4TdHdB
zt)ye1qVj-y^JSQBp@*G4OJiS-t;CC+M+@%Hd>dTg0T;f@h3|2_AaB!6?ViEU;m#t0
zTu{i4k)V0rF)H~n7==mD2zHEGu5mN3FV1@MCrDi@E{~yv5S0ZiM+Zj)96OE0F&PWv
zF-ewi>2UDjDwdk)X&e(};f#>L=Mu5eXyO8n3m0%yk;h^pI2#p2`ACC><3=^DB(d-?
zJWGy^3eIChWn^SGC>+s?fN5<8+|hHzvM??;ahJ}Mq^0MK|HNo!is&;{SYtqwnw6T?
z(rU6U^kLz2R2h-+kdjEiW>vZ$Nsf-5Ycy@hfg;8}RD{EIsW^gbd*Ctd@5f>er}Hxz
zJ(aXHvue$stch;ogMzTVsXQ@r#BR~9YpJ}Yfg+d9>K5-JetMs!_h~0$OagErlz!~}
zwGyxICu(871=B6G5o({;=ky1mzPq8mVh9)54bDIBpYv~`z`npU6z~N%!^jt$_s#hp
za9ww~uKQe=w1I7^+V`-aG1A)J{uAo#e^oz_m%%Db2ReHjA;Yb>ckKmNjT{Y66AX}F
zg+L8&6O5-0)6xXvt;29lFwJbanr$@}pW@n$A0^-T`SI(D@3^Oegg$pAILPLXAu;cK
zg-{yuJ_=Wbm0J6QkSlidYlQr$DLc^9-Z3higV6+dC=JQ4upuM`Cg|ZhL#eVn=!3^R
z70!2<JR}lY0Cvb}ay1PFZ0T8TmK7sWI{@umRoQK|s#lF<R<pTEliGn&cMvU`Bp=)U
zbgeOGf2j6AnYDm}Bq1T=u>_8t5yuihU7|czVXWuSjg41&4un<TgCLjLf@-Urz(42f
zd~`$+Bz*Y2Be?c!ggu4AM|qmf$KQN{V}_{_OG{&$iOl-3p<OG-$#YIWlf5)|?38Aa
zb}DfscP*968y4xHOV13^4a;WX&BeEt8r<vbz$BdxTrc~#J&qbKlxl4jHfXoj(oMpE
zCcQA3ln*M2EOUe%EV)rjuIt#MLH`mpe+<*EY|+tK;GXR3DF_>#-Al2B*m~c<ny1+L
zCMenuEb$BcvbZV~+utY*J?iee(|x;p{gvZuTCw|7;e(9>y-PP1Zmfh?mEwV8g|i#s
zgLnAb{EE0H{2co+_Dl9RrFdK_hUG&1QJ`%;J{R9a?(RLC9O~#=zPuJ+-~0YEpF23n
zYzEMtj`^E&H<y$5+YWB*=~|911i$3|I*9i7ZXRl2uX|khAs752xIDdlaplVQ9joWR
z>sbx1O{|Ba_qjob*m%Y{B9Z6l#{w-__*iIF)od=E&rms_s#o&KO!-cms!r;}v@*J(
z897xYv<4jdB|J#>K9#-)Ej<M<6pQGWfV@pr{T4aTDseUa93yn1f0Gu(r)bN?dV*Vl
zHc!vrukQB@F(o(!0OivczGFa+Qi*n~<H_bum7;u?ybd~Ak|r&fO_pJpKTzl&Z-g0Q
Kwh)~itA7D)DN3yX

literal 0
HcmV?d00001

diff --git a/__pycache__/PMT.cpython-313.pyc b/__pycache__/PMT.cpython-313.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..ce3c65e8123f8e6a0a5f6df049de5482fd7bb3a1
GIT binary patch
literal 7095
zcmb_hYit`yavqYy_e-Lv7xfzHBwMsaQj}y{vbB!2e#y2ZdpY8s4cSZ3<VfP0n&I{g
zZHw{Z7RX(|3pQALjfG@n12f10k&z#(1A?5B`?3CMAb-eEI1#f3VqE;k$RF8WB*6Yk
z)eK+O%DKZK4XPj2)zv-S)%8`i-q`IH3R3aE{xxyFiK6}!8<j+tD{rQu@)gBWtmXnm
z)KWt<N?S{`P--ve&g+QoJWXgu?bj218`Vd#IuFIthqS6BVp!66^)In8=he!lcvRw~
z#0n>yF^_XeDH>Bb)x126dUFEWUr|>noD?S15FM-aP=rSDM9<QY8(1@|_fQ{ei1Dh4
zHF&5)Wl6-$8i{4e<h97UOSAJYjX;lS%AHu~%71SRK*i4tQ^{(#(yDSRt;(vX1w(}`
zWUAC^Wyo5qS8A;0P+1ntQ62p!$kEox!4Fc>pCCt9C#U)iOO%(MlWk%4bZBmB_H+bx
zO}1T{V%eWw44qP1#&Bo`D_*l~6gg2$2)t~M3B1JJmLigHgX3jOTI5K%t`$Vt$niH5
z1o(7IoFtng*v&`r>?5hDDBcoCTsB>gid-a3l3tBWi(GP1#NA@#$pA_7fn`CI0ym=s
zDk~gRk{mB3KIQ|bxSL#3NUcCKaA|JFcM^Nz90|Yz{i#*i6p1AGgcOOuM*yYL!p{}w
zA^VIfwRU9b&uyiy-YmUk@{~N?&~(0N?|A5W;Ca~npnJnnY(D}t$Fsd?ZJvjg2bPD9
z2aeoou`QUT|Jwcr4<uIA6lmz-4>j#g9<r~f1<2JDs^==x`l{N?Ys1<Qj@^|i+RbW?
zUZ$uh$Y3=WUF;v=_Q7T6bz$v7RcFFFRyU#Xfpne@({*$WOPA^Tus&qoH9a(0eMNWr
zDX$^pKg&tXtiZ>ks|?<2W>Fx_Esnbp5f&Fk=)=0mED~Xb;e}fSznb7O9>H0|AQ8@7
zukid}8o1&cSn&<Sy?TOMChK5;Y{X7rBxsgolPHlzv>ILmX-0yzZZ26b3BV~kP%h|T
zWJOt*Y$%s@DM~;K%VZHRBBLbIf}$MQr-&m_eu<M^s5WoY8i7y5GyIHt(a^SL`_gvb
zk=2*FfAVnU!OGtVn~lZpkq3>BHIGk!JNqxQzr9!(zmRp58e6_@-*6P#j}#h@WDTXZ
z?q9WUb`*NwE400rwU?Zo+s=W4b0GI|(HYE}f}|bfzh$X#yA&Pz9qis8uR~yKsyY^?
zLRGs1(#pvJ+B2%<HPBlz`jsvy_d5M>)slgRQXvII>V`lSsJ6E$*kZM;j-^>WYw**o
z(XRt-+<Q*vkH-S%-fK6FYC(_9tYuikS|@a<$)ScHk7sR1Z&6fy4Aw@60PJ?B%AMNl
zds+K1Re`ZR`yG|rFjy8FKTJ^zsG3S*ot1v3jtc8QjoX##INK^+Ay1`ND*;{`YP|pV
zH?ej!B0St>wh?Qf6OM``pt;Jl>aN$VylL<A1)Xt^R`G+u^WH;ufZAD^r&a>ByVk1I
z#&lpwn&&cxKvGyr@EP+=f=}OOf@9;d@l1kni^6SLhb?n(czpPFaAIWGOUnj9RB*(4
zod~xopyDVuZgNBcrv@=bIF2WrC?wMoz?D}go2n3Gsv=NDNP=G!GPXc!MVi5-u%hsT
z?IhwGFZt$IQ``>?(Nrp#h(%FW;A0U4YGoQr*|^L_0e!@mPvN^{15BAtO0qqgmX;B)
zU<en_*cGwNVl<KD;=L~s=7=3LJd`wAm*0Q=3kH~$xHyyGnW)HQT9`pm5(vjkO*7+%
z#)3>}S%7)H4YG-FpQJfa0+eM{G$wfUh#NB$MqHSo%FA>LU{tn>QdCNdk(dBW1x2M~
zeTpP_Nv09PWu3Sx$~1R7A=Oq@F|=moa1f<s8a^%srdSjrVDh7CO`%y-CZztzyy}%c
z*^E|IsiK_|--kW;jC$7Ev)#J?-&^+=TfGR$&s#gzKE3zpT_N{j-aV8xmz+&s-1+65
zZD&ux*^@IDod>e?j)`jTLi8=Qbz<?{Gf?UoEcNX#^#@*=^_`AaMzh<JHSM%e?zXkb
zdz0Hu{e`Cf4f-4VBYW;=o5}pZ$<p9(Y3OKvaxQ=H;*Oqj5A0A{_XipnWTI^DZEH`#
z+LIeCTK8=%f3xyv<sXG@|5U+0^<?nr`$hl7Z9iM^v-$bhwm)9*$Di36?*_JPeOL~z
z`R@6)Y|Jy8^NZ=fn!fwN=NFzgcdcE!ckO;8ce&8)&ss`d`!}40uJN+((Vx3o@Qf%9
zh{>LB=*{*2ga7aR`QBhYcsB2t-ZGslZ^HWgc2}U#71(tDrsvl^dFP3|>4b_IRcj9$
zi&~{%#sSE`!dnW2-Kk=;qJ4_pt{-1!t#W`m?S0cL9C7e_Ykow{<K;8~iq|ffDs%6J
z88yErgzMgw*s9md>;70d3ss92)>J&5KOmc4z$>F9mJaK{&gsK6<c86z#{)r*ku@RW
zuvXT#_kI9%k9}nAtb=v34Q!*|2G*Dg;mzEYU@jM+u_3Gr(`=Jp%Nv1j7Y{Z=RdlI!
zUgp9^)(seF3Y+%A7PeXO7q7u7nnHN-cBKji^5(F4FZpas9S-g>Kie9p;v{bY-?d}c
z1kemyYN%DW*KG@372X;L>_z)qD5LGC>c)8X{z?ZmTGbW6+v<4&_a3;zcCei^Ra;XS
z=J0l9j-BD8*fJw7Cl;le<H|?^__yFzM$yhQX)(bsF<{!6(CMie1_F<i0CqIfHk%eD
zCKgR58O6p8dYQ!p5v5*&3m~{FnPz1I9A7_2hQ=u}I=|naIWsF{63JvVaL7N*3|<Zf
zk1;AH-X1?3IXrfZAvcc#Ao_>B%vmmWLkNrv2Z!MoWXgCMP<k`HyU}WhK|3EvC8G&`
zh&d2Aka2LlFU$^c{IO5o8}?6RT9-13)DRQr7L$NjL(FwD#AHM%o;k5BNh$GYAP`+f
z6!ND4>ja+Tk^oLZ($6OYE2};MfcIvCyM=Wzw#>!TNiGl!`bLIF4inU|jO7N$rF_w3
z;wC5Sxs{Z(D$_B6h#9L0QK3&D3BX;iEn^7-ru(Lr06S&FR4m4&q>NK7eKX*mr=v?;
zm2jE|K!XjHjUjH4BOD>%?7>{OebH3Hhj^c{tMziQDq8^6c`k-9o^h~T%%_}*?-ZbE
zeR~#gvcAnGBrbD208#@tM+4>179&tGFcJ(XFf9f~gTVlNSyEW#xWL4CaCj^rCZNr+
zib{dUV8(C=v6GXH5T;<ot0hNZ&9WgCCD9d;j6;J=AjPR@3B<C1-Fyt(3N)qYCY%){
zCSxiqQEwWd|M!3UIdevFE{-yDiSGehy$z~QQ1voo%ZJfqnmbL1K$I|x^x#yzn4u#_
zPGDAZA^r@TimPA%)d5A9br5HhDIj~TY7AznT@2Zv>NNQPhoSXSKg*($(S0B+QcpvM
z0U)>!Y6zz?i1>P<24!`TnFXUTl|gn?qB6C84EV)W#ff;1PQn*=a%0;+IRwAQ^WR3k
zjXZO8=DUs*UE_J%`18i5tl@>bb8Y6!nfr6w?t=yQ!Hx4p_mQmmSyRW_<d>88kN@i2
z=Fvj`@j}z_ZP#?cHT~36bj@dtFA$Rla*5o+CcXL7&2x|0yklz1^!|>4>WFDdEfb}l
zK&fZE)Ow}V>4%4ehk*wHcvn!7y5!kc>K-U{_m{d4zG}BOS+d5RZggJk5A1h;mfsi8
zH*;Cb^Y*?{YxgU?&Sl9Ob}UqbyUyjYJUNqh0@z>3($B5VZEHut+OaMct$n4Y*0qUy
z6Zem0PvNwK4+eAN8=<0Sc-u2p@QmdToq4J)dM;!ylx(gqW`8-mesIgS4?UFWui0Ej
zvCEh5^lvQZT}QTT<2Y$c*8aVbvb*li<lNi6!2<Xn%t+ofx@8-~QT=P)d*1a^>#XAG
zJa;x{?LYQC(3w?ykJoT6;CpDQJ_6|I0?^b{^AQ+oNn1y+i!SP@bj?!NQR}dI0q0Z_
zU_QJAvc?H*6E*!$5dXa6eFU+*65ZkZ$X+ilfb}&V`W-yhIv)Kycx-h%hIjDT-$8fO
z##YyB#Ml=j+b9HaP}j0fn!2cA&ENx6dcgGuUqa;q3jlQRKxhg*4WO6nqdH}jxwfle
z6L2(z&8*vrrJik8N(0-Xlt!4V6{U=tYiBb1&lF{)@Fcx$nW;;JlcZHf2`(xyHyBI8
ziyyPXNy|c<0RWA0OhRJf0w?nOB?hnrV;<GYc{A;E0+UX~gq3=r6w-X$KbLNWQ?~3p
zy#EU(eCg!H*>iJe8Fu>InR)i8B{OvjtZCVT++pV7si<a7?l7kSM=EV-rHK;BG1vj2
zVj`sg!X@ATb9(DF2aPO-WN-FCrg-Mn8v%rJVm09Pmi<WZN_2Cw12%?83RWm@=ag5Y
z#CY(2rn#*?n5xfMrpaV<<$64NT<$|tsT@Jsc={)k0{AB4aj2*~63i#!o5^R?X76tX
zem#&MpMKi6H8QK*YLKnF!L5=gFl2OdfhpMr08m?lGtL2ANeF-uB^Wow%R4ID6!RP*
z+)9+(kgf6B_P)eNqnBvsU*g2F#+T8-0c3QO!4b@cGup}VmuND)It7dXQv?Fhv--L4
zoF7Yy<O&YeE0aKhTO;cboM>2*L=xBwRWYXiw;=-0hO!Cw4QnE<prQo~qi&n6exu}u
z-CKte#RfR3&#3=4fSCkin0G;h=Lfrr;Nb9XnGQU|$_HDf5LB|u@RZ!#v3{m7IJ?z6
zmpxr_x38VMcP`)U-_#e~hk@W~RjldU#_6JKBzv;d)P|-!--QfKhryt?cCM}7Tg`bk
z!o}9n>~xu7{iBWWH&-5A+5GegSM<#kn`X18o(~+@;Qs!jyXkdy{rbJk<7VKVIR0ed
zx7|-q=Gn{tHnTpE)8<Cjht|7upKLU3gf_Jso()@}^-%t!Sa$janD)ze&#%XFgBzo{
z?hSEs?6Ge1z~hroOix{pKLr_==JVmp`42DWFI{>1=Xuw{?`&6JH&ZPaHL9W*lvf7z
zy+QFE6n{f`jX*S*^J-pV>9yjsJgN#j%UdGw5{n)pKG8=apQNM7a*s0-QM~44g6DW4
z5+Qiu$$revVFqCgy!<NWM71U)gOuh7oGx7GSR^We4qQ)5oEV8bqLd{nr%do6QOWR4
zRSZD(m(**mQ9tt9Wq_=K(NAf<KcyMcx9nWiP=>}=TDSh1=8X=jZ<MMk3&nVi1cOX6
zi5WUbsyC(hQ14??nd>jGh5=XQa;l$voctVUc=5$P$aZuZjpjSb{2k@|j<Ws-HTWH6
S`5*e0MsrQ`no<fy#{U9BkQ@sD

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..1f7c151
--- /dev/null
+++ b/main.py
@@ -0,0 +1,24 @@
+from PMT import PMT
+from DB import Database
+from dotenv import load_dotenv
+import os
+
+def main():
+    load_dotenv(override=True)
+
+    db = Database()
+    pmt = PMT()
+
+
+    pmt.login()
+
+    days_ahead = int(os.environ.get("DAYS_TO_FETCH"))
+    shifts = pmt.get_shifts(days_ahead)
+    PMT.print_shifts(shifts)
+
+    db.insert_shifts(shifts)
+    
+    print(f"\nFound {len(shifts)} shifts total.")
+    
+if __name__ == "__main__":
+    main()