LIL-551. Upload clients to roistat

apps/user/management/commands/users_to_roistat.py

@@ -0,0 +1,32 @@
+import requests
+
+from django.conf import settings
+from django.contrib.auth import get_user_model
+from django.core.management.base import BaseCommand, CommandError
+
+# User = get_user_model()
+
+from apps.user.models import User
+
+
+class Command(BaseCommand):
+    help = 'Upload users to Roistat'
+
+    def handle(self, *args, **options):
+        users_queryset = User.objects.all()
+        users = [
+            {
+                'id': str(user.id),
+                'name': user.get_full_name(),
+                'phone': str(user.phone),
+                'email': user.email,
+                'birth_date': user.birthday.strftime('%d%m%Y') if user.birthday else None,
+            }
+            for user in users_queryset
+        ]
+        project = settings.ROISTAT_PROJECT
+        key = settings.ROISTAT_KEY
+        url = settings.ROISTAT_API_URL + f'/project/clients/import?key={key}&project={project}'
+        resp = requests.post(url, json=users)
+        self.stdout.write(str(resp))
+        self.stdout.write(str(resp.json()))

apps/user/tasks.py

@@ -1,6 +1,9 @@
+import requests
+
 from mixpanel import Mixpanel
 
 from django.conf import settings
+from django.contrib.auth import get_user_model
 
 from project.celery import app
 
@@ -20,3 +23,23 @@ def user_to_mixpanel(user_id, email, phone, first_name, last_name, date_joined,
             'subscriptions': subscriptions,
         }
     )
+
+
+@app.task
+def users_to_roistat():
+    User = get_user_model()
+    users_queryset = User.objects.all()
+    users = [
+        {
+            'id': str(user.id),
+            'name': user.get_full_name(),
+            'phone': str(user.phone),
+            'email': user.email,
+            'birth_date': user.birthday.strftime('%d%m%Y') if user.birthday else None,
+        }
+        for user in users_queryset
+    ]
+    project = settings.ROISTAT_PROJECT
+    key = settings.ROISTAT_KEY
+    url = settings.ROISTAT_API_URL + f'/project/clients/import?key={key}&project={project}'
+    resp = requests.post(url, json=users)

project/settings.py

@@ -222,6 +222,11 @@ CELERY_BEAT_SCHEDULE = {
         'schedule': timedelta(minutes=5) if DEBUG else crontab(minute=0, hour=0),
         'args': (),
     },
+    'update_users_in_roistat': {
+        'task': 'apps.user.tasks.users_to_roistat',
+        'schedule': timedelta(hours=1),
+        'args': (),
+    },
 }
 
 try: