diff --git a/finddates.py b/finddates.py index 1135baa..c9b6d56 100644 --- a/finddates.py +++ b/finddates.py @@ -1,8 +1,13 @@ +import settings import fredmssql def get_birthdays(start, thru): - db = fredmssql.FredDB(DATABASE_USER, DATABASE_PASS, DATABASE_NAME) + db = fredmssql.FredDB( + settings.DATABASE_USER, + settings.DATABASE_PASS, + settings.DATABASE_NAME + ) s = (" SELECT e.EmpUID, e.LastName, e.FirstName, e.BirthDate " " FROM Employee e " "LEFT JOIN EmpCstFV c ON e.EmpUID = c.EmpUID AND c.CstID = 105 " @@ -26,7 +31,11 @@ def get_birthdays(start, thru): def get_anniversary(start, thru): - db = fredmssql.FredDB(DATABASE_USER, DATABASE_PASS, DATABASE_NAME) + db = fredmssql.FredDB( + settings.DATABASE_USER, + settings.DATABASE_PASS, + settings.DATABASE_NAME + ) s = (" SELECT e.EmpUID, e.LastName, e.FirstName, e.HireDate " " FROM Employee e " "LEFT JOIN EmpCstFV c ON e.EmpUID = c.EmpUID AND c.CstID = 105 " diff --git a/settings.py b/settings.py new file mode 100644 index 0000000..2873243 --- /dev/null +++ b/settings.py @@ -0,0 +1,12 @@ +import os +from dotenv import load_dotenv + +load_dotenv() + +DATABASE_NAME = os.getenv("DATABASE_NAME") +DATABASE_USER = os.getenv("DATABASE_USER") +DATABASE_PASS = os.getenv("DATABASE_PASS") + + +if __name__ == "__main__": + pass diff --git a/workdays.py b/workdays.py index cb0e71d..fbab216 100644 --- a/workdays.py +++ b/workdays.py @@ -61,3 +61,4 @@ def get_date_range(date=''): if __name__ == "__main__": pass +