25 lines
436 B
Python
25 lines
436 B
Python
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()
|