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")
--
GitLab