75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import datetime
|
|
import os
|
|
import time
|
|
|
|
from dotenv import load_dotenv
|
|
from auth.session import AnnySession
|
|
from booking.client import BookingClient
|
|
from utils.helpers import get_future_datetime
|
|
import pytz
|
|
from config.constants import RESOURCE_ID, TIMEZONE, SSO_PROVIDER, BOOKING_TIMES, RESOURCE_IDS
|
|
|
|
|
|
def main():
|
|
load_dotenv(".env", override=True)
|
|
username = os.getenv("USERNAME")
|
|
password = os.getenv("PASSWORD")
|
|
|
|
tz = pytz.timezone(TIMEZONE)
|
|
|
|
if not username or not password:
|
|
print("❌ Missing USERNAME or PASSWORD in .env")
|
|
return
|
|
|
|
session = AnnySession(username, password, provider_name=SSO_PROVIDER)
|
|
cookies = session.login()
|
|
|
|
if not cookies:
|
|
return
|
|
|
|
booking = BookingClient(cookies)
|
|
|
|
# Only wait for midnight if within 10 minutes, otherwise execute immediately
|
|
now = datetime.datetime.now(tz)
|
|
midnight = (now + datetime.timedelta(days=1)).replace(
|
|
hour=0, minute=0, second=0, microsecond=0
|
|
)
|
|
seconds_until_midnight = (midnight - now).total_seconds()
|
|
max_wait_seconds = 10 * 60 # 10 minutes
|
|
|
|
if 0 < seconds_until_midnight <= max_wait_seconds:
|
|
print("⏳ Waiting {seconds_until_midnight:.0f} seconds until midnight...")
|
|
time.sleep(seconds_until_midnight)
|
|
elif seconds_until_midnight > max_wait_seconds:
|
|
print("⚡ More than 10 min until midnight, executing immediately...")
|
|
|
|
for time_ in BOOKING_TIMES:
|
|
try:
|
|
start = get_future_datetime(hour=time_["start"])
|
|
end = get_future_datetime(hour=time_["end"])
|
|
|
|
# maybe add if reservation fails try next id and if everything fails do ^
|
|
|
|
for seat_id in RESOURCE_IDS:
|
|
resource_id = seat_id
|
|
reservation_success = booking.reserve(resource_id, start, end)
|
|
if reservation_success:
|
|
return True
|
|
|
|
if RESOURCE_ID:
|
|
resource_id = RESOURCE_ID
|
|
reservation_success = booking.reserve(resource_id, start, end)
|
|
if reservation_success:
|
|
return True
|
|
resource_id = booking.find_available_resource(start, end)
|
|
reservation_success = booking.reserve(resource_id, start, end)
|
|
if reservation_success:
|
|
return True
|
|
else:
|
|
print("⚠️ No available slots found. ( or other error )")
|
|
except Exception as e:
|
|
print(f"❌ Error booking slot {time_['start']}-{time_['end']}: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|