Initial ability to generate invoices from the fake data tool

1 files changed, 7 insertions, 1 deletions

diff --git a/app/dispatch/misc.py b/app/dispatch/misc.py
index 14bb5e7..e0976bf 100644
--- a/app/dispatch/misc.py
+++ b/app/dispatch/misc.py
@@ -3,11 +3,17 @@ from datetime import datetime, timedelta
 from dateutil import rrule
 import uuid
 
+# Text formatted date. *sigh*
 def get_week_dates(date=None):
-    week_dates = {}
     if date == None:
         date = formats.date_format(datetime.now(), "SHORT_DATE_FORMAT")
     dt = datetime.strptime(date, '%m/%d/%Y')
+
+    return get_week_dates_datetime(dt=dt)
+
+def get_week_dates_datetime(dt=None):
+    week_dates = {}
+
     weekday = dt.weekday()
     if weekday == 6:
         week_dates['start_date'] = dt