from PMT import PMT
from DB import Database
from API import Webserver
from dotenv import load_dotenv
from time import sleep
import os

def main():
    # Load the environment
    load_dotenv(override=True)


    # Set up database, webserver and PMT connection
    db = Database()
    pmt = PMT()
    webserver = Webserver("0.0.0.0", 8080, db)

    # fetch PMT shifts
    pmt.login()

    days_ahead = int(os.environ.get("DAYS_TO_FETCH"))
    shifts = pmt.get_shifts(days_ahead)

    db.delete_future_shifts() # delete all future shifts before importing, to make sure if a shift is removed, it actually gets removed here too.


    # Insert the new shifts into the database
    db.insert_shifts(shifts)
    

    webserver.run()

if __name__ == "__main__":
    main()