ynerant/nk20-scripts
1
0
Fork 0
mirror of https://gitlab.crans.org/bde/nk20-scripts synced 2024-12-27 01:42:24 +00:00
Code Issues Releases Wiki Activity

Merge branch 'Update_Scripts' into 'master'

Browse Source 
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
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"]: