anny-booking-automation/main.py

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 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(f"⏳ 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:
                    print("✅ Reservation success for %d" % resource_id)
                    # return True
                else:
                    print("❌ Reservation failure for %d" % resource_id)

            # 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()