first commit
This commit is contained in:
24
main.py
Normal file
24
main.py
Normal file
@@ -0,0 +1,24 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user