Skip to content
Snippets Groups Projects
Commit 9207c5cd authored by Pierre-antoine Comby's avatar Pierre-antoine Comby
Browse files

add aliases import

parent e850c575
No related branches found
No related tags found
1 merge request!19Import nk15
Pipeline #7786 passed with warnings with stages
in 3 minutes and 51 seconds
...@@ -152,6 +152,7 @@ def import_boutons(cur,map_idbde): ...@@ -152,6 +152,7 @@ def import_boutons(cur,map_idbde):
cat.save() cat.save()
button.save() button.save()
@transaction.atomic @transaction.atomic
def import_transaction(cur, map_idbde): def import_transaction(cur, map_idbde):
cur.execute("SELECT * FROM transactions;") cur.execute("SELECT * FROM transactions;")
...@@ -159,6 +160,26 @@ def import_transaction(cur, map_idbde): ...@@ -159,6 +160,26 @@ def import_transaction(cur, map_idbde):
obj_dict = { obj_dict = {
"pk":row["id"], "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): class Command(BaseCommand):
...@@ -171,7 +192,8 @@ class Command(BaseCommand): ...@@ -171,7 +192,8 @@ class Command(BaseCommand):
parser.add_argument('-c', '--comptes', action = 'store_true') parser.add_argument('-c', '--comptes', action = 'store_true')
parser.add_argument('-b', '--boutons', action = 'store_true') parser.add_argument('-b', '--boutons', action = 'store_true')
parser.add_argument('-t', '--transactions', action = 'store_true') parser.add_argument('-t', '--transactions', action = 'store_true')
parser.add_argument('-a', '--aliases', action = 'store_true')
def handle(self, *args, **kwargs): def handle(self, *args, **kwargs):
conn = pg.connect(database="nk15",user="nk15_user") conn = pg.connect(database="nk15",user="nk15_user")
cur = conn.cursor(cursor_factory = pge.DictCursor) cur = conn.cursor(cursor_factory = pge.DictCursor)
...@@ -189,3 +211,6 @@ class Command(BaseCommand): ...@@ -189,3 +211,6 @@ class Command(BaseCommand):
print("boutons table imported") print("boutons table imported")
if kwargs["transactions"]: if kwargs["transactions"]:
import_transaction(cur) import_transaction(cur)
if kwargs["aliases"]:
import_aliases(cur,map_idbde)
print("aliases imported")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment