diff --git a/generate-work-hours.py b/generate-work-hours.py new file mode 100755 index 0000000..d03830d --- /dev/null +++ b/generate-work-hours.py @@ -0,0 +1,30 @@ +#!/usr/bin/env python3 + +import datetime +import json +import sys + +DATE_FORMAT = '%Y-%m-%d' + +sys.argv.append(9) + +from_date = datetime.datetime.strptime(sys.argv[1], DATE_FORMAT) +to_date = datetime.datetime.strptime(sys.argv[2], DATE_FORMAT) +hours = int(sys.argv[3]) + +def daterange(start_date, end_date): + days = (end_date - start_date).days + for n in range(int(days)): + yield start_date + datetime.timedelta(days=n) + +result = [] + +for d in daterange(from_date, to_date): + week_day = d.isoweekday() + if week_day != 6 and week_day != 7: + result.append({ + 'date': d.strftime(DATE_FORMAT), + 'hours': hours, + }) + +print(json.dumps(result, indent=2)) \ No newline at end of file