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 Alias
from note.models import Transaction, TransactionTemplate,\
TemplateCategory, TemplateTransaction, MembershipTransaction
TemplateCategory, RecurrentTransaction, MembershipTransaction
from member.models import Profile, Club, Membership
"""
@ -185,7 +185,7 @@ def import_transaction(cur, map_idbde):
if created:
cat.save()
obj_dict["category"] = cat
transac = TemplateTransaction.objects.create(**obj_dict)
transac = RecurrentTransaction.objects.create(**obj_dict)
transac.save()
elif row["type"] == "adhésion":
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('-t', '--transactions', action = 'store_true',help="import transaction")
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):
nk15db, nk15user = kwargs['nk15db'], kwargs['nk15user']
#reset database.
call_command("flush")
self.print_success("flush nk20 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)
if kwargs["special"]: