Add daily visits stats.

[wolnelektury.git] / 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 (file)
index 0000000..efe2500
--- /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)
+