ynerant
/
nk20-scripts
mirror of https://gitlab.crans.org/bde/nk20-scripts
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge branch 'Update_Scripts' into 'master' 

Update scripts

See merge request bde/nk20-scripts!1
This commit is contained in:
Pierre-antoine Comby 2020-03-22 19:21:12 +01:00
parent be8aa44eb6 146260b358
commit b9fdced3c2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
management/commands/import_nk15.py
View File

@ -18,7 +18,7 @@ from django.contrib.auth.models import User
from note.models import Note, NoteSpecial, NoteUser, NoteClub from note.models import Note, NoteSpecial, NoteUser, NoteClub
from note.models import Alias from note.models import Alias
from note.models import Transaction, TransactionTemplate,\ from note.models import Transaction, TransactionTemplate,\
TemplateCategory, TemplateTransaction, MembershipTransaction TemplateCategory, RecurrentTransaction, MembershipTransaction
from member.models import Profile, Club, Membership from member.models import Profile, Club, Membership
""" """
@ -185,7 +185,7 @@ def import_transaction(cur, map_idbde):
if created: if created:
cat.save() cat.save()
obj_dict["category"] = cat obj_dict["category"] = cat
transac = TemplateTransaction.objects.create(**obj_dict) transac = RecurrentTransaction.objects.create(**obj_dict)
transac.save() transac.save()
elif row["type"] == "adhésion": elif row["type"] == "adhésion":
print("adhesion not supported yet") print("adhesion not supported yet")
@ -231,13 +231,16 @@ class Command(BaseCommand):
parser.add_argument('-b', '--boutons', action = 'store_true', help="import boutons") parser.add_argument('-b', '--boutons', action = 'store_true', help="import boutons")
parser.add_argument('-t', '--transactions', action = 'store_true',help="import transaction") parser.add_argument('-t', '--transactions', action = 'store_true',help="import transaction")
parser.add_argument('-a', '--aliases', action = 'store_true',help="import aliases") parser.add_argument('-a', '--aliases', action = 'store_true',help="import aliases")
parser.add_argument('-d', '--nk15db', action='store', default='nk15', help='NK15 database name')
parser.add_argument('-u', '--nk15user', action='store', default='nk15_user', help='NK15 database owner')
def handle(self, *args, **kwargs): def handle(self, *args, **kwargs):
nk15db, nk15user = kwargs['nk15db'], kwargs['nk15user']
#reset database. #reset database.
call_command("flush") call_command("flush")
self.print_success("flush nk20 database") self.print_success("flush nk20 database")
# connecting to nk15 database # connecting to nk15 database
conn = pg.connect(database="nk15",user="nk15_user") conn = pg.connect(database=nk15db,user=nk15user)
cur = conn.cursor(cursor_factory = pge.DictCursor) cur = conn.cursor(cursor_factory = pge.DictCursor)
if kwargs["special"]: if kwargs["special"]: