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