arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-28 19:02:16 +00:00
Code Issues Projects Releases Wiki Activity

Fixed added the ability to import

Browse Source
This commit is contained in:
tomtom5152
2015-01-28 18:45:40 +00:00
parent fd36688147
commit dd88b5361f
1 changed files with 50 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
importer.py
View File

@@ -34,6 +34,7 @@ def setup_cursor():
def clean_ascii(text): def clean_ascii(text):
return ''.join([i if ord(i) < 128 else '' for i in text]) return ''.join([i if ord(i) < 128 else '' for i in text])
def import_users(delete=False): def import_users(delete=False):
if (delete): if (delete):
models.Event.objects.get(is_rig=False).delete() models.Event.objects.get(is_rig=False).delete()
@@ -157,6 +158,7 @@ def import_venues(delete=False):
object = models.Venue(name=name, three_phase_available=row[1]) object = models.Venue(name=name, three_phase_available=row[1])
object.save() object.save()
def import_rigs(delete=False): def import_rigs(delete=False):
if delete: if delete:
models.Event.objects.all().delete() models.Event.objects.all().delete()
@@ -214,6 +216,8 @@ def import_rigs(delete=False):
object.purchase_order = row[21] object.purchase_order = row[21]
object.payment_received = row[22] object.payment_received = row[22]
object.collector = row[23] object.collector = row[23]
if object.dry_hire and object.end_time < datetime.date.today():
object.checked_in_by = mic
object.save() object.save()
@@ -246,6 +250,7 @@ def import_eventitem(delete=True):
with reversion.create_revision(): with reversion.create_revision():
event.save() event.save()
def import_nonrigs(delete=False): def import_nonrigs(delete=False):
if (delete): if (delete):
try: try:
@@ -255,15 +260,26 @@ def import_nonrigs(delete=False):
cursor = setup_cursor() cursor = setup_cursor()
if cursor is None: if cursor is None:
return return
sql = """SELECT name, start_date, start_time, end_date, end_time, description, user_id FROM non_rigs WHERE active = 1;""" sql = """SELECT id, name, start_date, start_time, end_date, end_time, description, user_id FROM non_rigs WHERE active = 1;"""
cursor.execute(sql) cursor.execute(sql)
for row in cursor.fetchall(): for row in cursor.fetchall():
print(row) print(row)
mic = models.Profile.objects.get(pk=row[6]) mic = models.Profile.objects.get(pk=row[6])
with transaction.atomic(), reversion.create_revision(): with transaction.atomic(), reversion.create_revision():
event = models.Event(pk=None, name=row[0], start_date=row[1], start_time=row[2], end_date=row[3], end_time=row[4], description=row[5], mic=mic) try:
print(event) object = models.Event.objects.get(pk=row[0])
event.save() except ObjectDoesNotExist:
object = models.Event()
object.name = row[1]
object.start_date = row[2]
object.start_time = row[3]
object.end_date = row[4]
object.end_time = row[5]
object.description = row[6]
object.mic = row[7]
print(object)
object.save()
def main(): def main():
# import_users() # import_users()