X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/c8d364d4d3331d03a079d200887e5462279fd875..a3f3fa4837a3364acc1d69f5f17b5153ca3af057:/src/stats/management/commands/populate_visits.py

diff --git a/src/stats/management/commands/populate_visits.py b/src/stats/management/commands/populate_visits.py
new file mode 100644
index 000000000..efe2500db
--- /dev/null
+++ b/src/stats/management/commands/populate_visits.py
@@ -0,0 +1,25 @@
+from datetime import date, timedelta
+from django.core.management.base import BaseCommand
+from stats.models import Visits, DayVisits
+
+
+class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '-s' ,'--since', metavar='YYYY-MM-DD',
+            required=True
+        )
+        parser.add_argument(
+            '-u' ,'--until', metavar='YYYY-MM-DD',
+            required=True
+        )
+
+        
+    def handle(self, **options):
+        since = date(*[int(p) for p in options['since'].split('-')])
+        until = date(*[int(p) for p in options['until'].split('-')])
+        while since < until:
+            print(since)
+            DayVisits.build_day(since)
+            since += timedelta(1)
+