From 9207c5cdec7df347faa53013836a0d0eb5d03efa Mon Sep 17 00:00:00 2001
From: Pierre-antoine Comby <comby@crans.org>
Date: Sun, 23 Feb 2020 21:11:42 +0100
Subject: [PATCH] add aliases import

---
 .../member/management/commands/import_nk15.py | 27 ++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/apps/member/management/commands/import_nk15.py b/apps/member/management/commands/import_nk15.py
index 12399862..4b8d57aa 100644
--- a/apps/member/management/commands/import_nk15.py
+++ b/apps/member/management/commands/import_nk15.py
@@ -152,6 +152,7 @@ def import_boutons(cur,map_idbde):
             cat.save()
         button.save()
 
+
 @transaction.atomic
 def import_transaction(cur, map_idbde):
     cur.execute("SELECT * FROM transactions;")
@@ -159,6 +160,26 @@ def import_transaction(cur, map_idbde):
         obj_dict = {
             "pk":row["id"],
         }
+       
+@transaction.atomic
+def import_aliases(cur,map_idbde):
+    cur.execute("SELECT * FROM aliases ORDER by id")
+    for row in cur:
+        alias_name = row["alias"]
+        alias_name_good = (alias_name[:252]+'...') if len(alias_name) > 255 else alias_name
+        obj_dict = {
+            "note_id":map_idbde[row["idbde"]],
+            "name":alias_name_good,
+        }
+        try:
+            with transaction.atomic():
+                alias =  Alias.objects.create(**obj_dict)
+        except IntegrityError as e:
+            if "unique" in e.args[0]:
+                continue
+            else:
+                raise(e)
+        alias.save()
 
 
 class Command(BaseCommand):
@@ -171,7 +192,8 @@ class Command(BaseCommand):
         parser.add_argument('-c', '--comptes', action = 'store_true')
         parser.add_argument('-b', '--boutons', action = 'store_true')
         parser.add_argument('-t', '--transactions', action = 'store_true')
-
+        parser.add_argument('-a', '--aliases', action = 'store_true')
+       
     def handle(self, *args, **kwargs):
         conn = pg.connect(database="nk15",user="nk15_user")
         cur = conn.cursor(cursor_factory = pge.DictCursor)
@@ -189,3 +211,6 @@ class Command(BaseCommand):
             print("boutons table imported")
         if kwargs["transactions"]:
             import_transaction(cur)
+        if kwargs["aliases"]:
+            import_aliases(cur,map_idbde)
+            print("aliases imported")