diff options
| -rwxr-xr-x | satscalc.py | 38 |
1 files changed, 32 insertions, 6 deletions
diff --git a/satscalc.py b/satscalc.py index 878ba76..6e733cc 100755 --- a/satscalc.py +++ b/satscalc.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -import time +import datetime # TODO: Prompt user for base rate, call time, wrap time, whether overtime was warned, and if it was a weekend. # This should be enough to calculate the pay for a single day's work. @@ -9,12 +9,38 @@ hourly_rate = baserate/7.5 print("Hourly rate:", str(hourly_rate)) -# TODO: Investigate if storing multiple calltimes and wraptimes will require a custom datastructure +print("\nTime to add your hours of work.") + +def timeInput(prompt): + while True: + i = input(prompt) + if i=="": return "" + elif len(i)==4: break + print("Wrong format. Try again.") + + t = datetime.datetime.strptime(i,"%H%M") + + return t while True: - # TODO: Check for valid format - calltime = input("Your call time (hhmm): ") - if calltime == "done": + calltime = timeInput("Your call time (24-hour format, hhmm. Leave blank to continue.): ") + if calltime == "": break - wraptime = input("Your wrap time (hhmm): ") + wraptime = timeInput("Your wrap time (24-hour format, hhmm): ") + + # Make sure we're not traveling backwards in time + if wraptime < calltime: + wraptime = wraptime.replace(day=wraptime.day+1) + difference = wraptime-calltime + hours_worked = difference.total_seconds()/3600 + # TODO: This part makes no sense. Don't worry. It'll be made to make sense. + + input("Was this a weekend? [y/n]: ") + + if hours_worked > 8: + print("Overtime detected.") + input("Was this overtime warned about before 12 o'clock the day before? [y/n]: ") + +print("\nDifference:",difference) +print("Hours worked:",hours_worked) |