From cc7754a9d78cba712fad35afdd4581c7289e8fa5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B8=D0=BA=D1=82=D0=BE=D1=80=20=D0=93=D0=BB=D0=B0?=
 =?UTF-8?q?=D0=B4=D0=BA=D0=B8=D1=85?= <v.gladkikh@spider.ru>
Date: Tue, 3 Dec 2019 17:58:46 +0300
Subject: [PATCH] Old role migrate1

---
 .../management/commands/add_affilations.py    | 52 ++++++++++++++++---
 1 file changed, 46 insertions(+), 6 deletions(-)

diff --git a/apps/account/management/commands/add_affilations.py b/apps/account/management/commands/add_affilations.py
index 4485d6f1..550847f2 100644
--- a/apps/account/management/commands/add_affilations.py
+++ b/apps/account/management/commands/add_affilations.py
@@ -1,4 +1,5 @@
-from account.models import OldRole
+from account.models import OldRole, Role
+from main.models import SiteSettings
 from django.core.management.base import BaseCommand
 from django.db import connections
 from establishment.management.commands.add_position import namedtuplefetchall
@@ -34,12 +35,51 @@ class Command(BaseCommand):
                 ''')
             return namedtuplefetchall(cursor)
 
-    def handle(self, *args, **kwargs):
+    def add_old_roles(self):
         objects = []
         OldRole.objects.all().delete()
         for s in tqdm(self.map_role_sql(), desc='Add permissions old'):
-               objects.append(
-                   OldRole(new_role=s.new_role, old_role=s.role)
-               )
+            objects.append(
+                OldRole(new_role=s.new_role, old_role=s.role)
+            )
         OldRole.objects.bulk_create(objects)
-        self.stdout.write(self.style.WARNING(f'Migrated old roles.'))
\ No newline at end of file
+        self.stdout.write(self.style.WARNING(f'Migrated old roles.'))
+
+    def site_role_sql(self):
+        with connections['legacy'].cursor() as cursor:
+            cursor.execute('''
+                    select site_id, 
+                           role 
+                    from 	
+                        (
+                            SELECT
+                                DISTINCT 
+                                site_id, 
+                                COALESCE(role, 'GUEST') as role
+                            FROM site_affiliations AS sa
+                        ) t 
+                    where t.role not in ('admin', 'GUEST')
+               ''')
+            return namedtuplefetchall(cursor)
+
+    def add_site_role(self):
+        objects = []
+        for s in tqdm(self.site_role_sql(), desc='Add site role'):
+            old_role = OldRole.objects.get(old_role=s.role)
+            role_choice = getattr(Role, old_role.new_role)
+            sites = SiteSettings.objects.filter(old_id=s.site_id)
+            for site in sites:
+                role = Role.objects.filter(site=site, role=role_choice)
+                if not role.exists():
+                    objects.append(
+                        Role(site=site, role=role_choice)
+                        )
+
+        Role.objects.bulk_create(objects)
+        self.stdout.write(self.style.WARNING(f'Added site roles.'))
+
+
+
+    def handle(self, *args, **kwargs):
+       # self.add_old_roles()
+        self.add_site_role()
\ No newline at end of file