added comments

PMT.py

@@ -22,12 +22,13 @@ class PMT:
         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 login POST
         sso_data = {
             "username": self._username,
             "password": self._password,
@@ -43,22 +44,29 @@ class PMT:
                                    json=sso_data,
                                    headers={'Content-Type': 'application/json'})
         
+
+        # Check if it succeeded.
         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)
         
+        # Store the useful information in some class fields
         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 session info
+        print(f"✅ Logged in as {self._username}")
         return self.session, self.context_token, self.user_token
 
     def get_shifts(self, days_ahead=14):
+        # raise if not logged in.
         if not self.session or not self.context_token or not self.user_token:
             raise ValueError("Must call login() first!")
-            
+
+
+        # define date ranges    
         shifts = []
         
         from_date = datetime.now().strftime("%Y-%m-%d")
@@ -66,6 +74,7 @@ class PMT:
         
         print(f"Fetching shifts from {from_date} to {to_date}...")
         
+        # send request for shifts
         headers = {
             'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:148.0) Gecko/20100101 Firefox/148.0',
             'Accept': 'application/json, text/plain, */*',
@@ -88,7 +97,9 @@ class PMT:
         }
         
         resp = self.session.get(shifts_url, headers=headers, params=params)
-        
+
+
+        # check if request succeeded
         if resp.status_code == 200:
             data = resp.json()
             week_shifts = data.get('result', {}).get('shift_instances', [])
@@ -98,6 +109,7 @@ class PMT:
             print(f"❌ Failed: {resp.status_code}")
             print(resp.text[:500])
         
+        # return all shifts fetched
         return shifts
 
     @classmethod
@@ -106,7 +118,7 @@ class PMT:
             print("No upcoming shifts found.")
             return
         
-        print("\n📅 UPCOMING SHIFTS:\n")
+        print("\nUPCOMING SHIFTS:\n")
         print("Date       | Time          | Dept        | Duration | Description")
         print("-" * 80)