removed some copilot added comments
This commit is contained in:
3
PMT.py
3
PMT.py
@@ -15,7 +15,6 @@ class PMT:
|
|||||||
self.base_url = os.environ.get("URL_PMT")
|
self.base_url = os.environ.get("URL_PMT")
|
||||||
|
|
||||||
def get_week_range(self, week_offset):
|
def get_week_range(self, week_offset):
|
||||||
"""Get Monday date for week_offset weeks from now"""
|
|
||||||
today = datetime.now()
|
today = datetime.now()
|
||||||
days_to_monday = 0 if today.weekday() == 0 else 7 - today.weekday()
|
days_to_monday = 0 if today.weekday() == 0 else 7 - today.weekday()
|
||||||
monday = today + timedelta(days=days_to_monday)
|
monday = today + timedelta(days=days_to_monday)
|
||||||
@@ -57,7 +56,6 @@ class PMT:
|
|||||||
return self.session, self.context_token, self.user_token
|
return self.session, self.context_token, self.user_token
|
||||||
|
|
||||||
def get_shifts(self, days_ahead=14):
|
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:
|
if not self.session or not self.context_token or not self.user_token:
|
||||||
raise ValueError("Must call login() first!")
|
raise ValueError("Must call login() first!")
|
||||||
|
|
||||||
@@ -104,7 +102,6 @@ class PMT:
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def print_shifts(cls, shifts):
|
def print_shifts(cls, shifts):
|
||||||
"""Pretty print shifts (class method since it doesn't need instance)"""
|
|
||||||
if not shifts:
|
if not shifts:
|
||||||
print("No upcoming shifts found.")
|
print("No upcoming shifts found.")
|
||||||
return
|
return
|
||||||
|
|||||||
Reference in New Issue
Block a user