mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 13:32:15 +00:00
243 lines
8.6 KiB
Python
243 lines
8.6 KiB
Python
__author__ = 'Ghost'
|
|
import os
|
|
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "PyRIGS.settings")
|
|
import django
|
|
|
|
django.setup()
|
|
|
|
from django.db import connections
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.db.utils import ConnectionDoesNotExist
|
|
from django.db import transaction
|
|
from RIGS import models
|
|
import reversion
|
|
from datetime import datetime
|
|
|
|
def setup_cursor():
|
|
try:
|
|
cursor = connections['legacy'].cursor()
|
|
return cursor
|
|
except ConnectionDoesNotExist:
|
|
print("Legacy database is not configured")
|
|
print(connections._databases)
|
|
return None
|
|
|
|
|
|
def clean_ascii(text):
|
|
return ''.join([i if ord(i) < 128 else '' for i in text])
|
|
|
|
def import_users():
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT `id`, `login`, `email`, `first_name`, `last_name`, `verified`, `initials`, `phone_number` FROM `users`"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
try:
|
|
object = models.Profile.objects.get(pk=row[0])
|
|
object.first_name = row[3]
|
|
object.last_name = row[4]
|
|
object.initials = row[6]
|
|
object.phone = row[7]
|
|
object.save()
|
|
print("Updated " + str(object))
|
|
except ObjectDoesNotExist:
|
|
object = models.Profile(pk=row[0], username=row[1], email=row[2], first_name=row[3], last_name=row[4],
|
|
is_active=row[5], initials=row[6], phone=row[7])
|
|
object.set_unusable_password()
|
|
object.save()
|
|
print("Created " + str(object))
|
|
|
|
|
|
def import_people():
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT `id`, `name`, `phone`, `email`, `address`, `status` FROM `people`"""
|
|
cursor.execute(sql)
|
|
resp = cursor.fetchall()
|
|
for row in resp:
|
|
email = row[3]
|
|
if email is not "" and "@" not in email:
|
|
email += "@nottingham.ac.uk"
|
|
|
|
notes = ""
|
|
if row[5] != "Normal":
|
|
notes = row[5]
|
|
|
|
person, created = models.Person.objects.get_or_create(pk=row[0], name=row[1], phone=row[2], email=email,
|
|
address=row[4], notes=notes)
|
|
if created:
|
|
print("Created: " + person.__str__())
|
|
with transaction.atomic(), reversion.create_revision():
|
|
person.save()
|
|
else:
|
|
print("Found: " + person.__str__())
|
|
|
|
|
|
def import_organisations():
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT `id`, `name`, `phone`, `address`, `union_account`, `status` FROM `organisations`"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
notes = ""
|
|
if row[5] != "Normal":
|
|
notes = row[5]
|
|
|
|
object, created = models.Organisation.objects.get_or_create(pk=row[0], name=row[1], phone=row[2],
|
|
address=row[3],
|
|
union_account=row[4], notes=notes)
|
|
if created:
|
|
print("Created: " + object.__str__())
|
|
with transaction.atomic(), reversion.create_revision():
|
|
object.save()
|
|
else:
|
|
print("Found: " + object.__str__())
|
|
|
|
|
|
def import_vat_rates():
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT `id`, `start_date`, `start_time`, `comment`, `rate` FROM `vat_rates`"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
start_at = datetime.combine(row[1], row[2])
|
|
object, created = models.VatRate.objects.get_or_create(pk=row[0], start_at=start_at,
|
|
comment=row[3], rate=row[4])
|
|
if created:
|
|
print("Created: " + object.__str__())
|
|
with transaction.atomic(), reversion.create_revision():
|
|
object.save()
|
|
else:
|
|
print("Found: " + object.__str__())
|
|
|
|
|
|
def import_venues(delete=False):
|
|
if delete:
|
|
models.Venue.objects.all().delete()
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT `venue`, `threephasepower` FROM `eventdetails` WHERE `venue` IS NOT NULL"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
print(("Searching for %s", row[0]))
|
|
try:
|
|
object = models.Venue.objects.get(name__iexact=row[0])
|
|
if not object.three_phase_available and row[1]:
|
|
with transaction.atomic(), reversion.create_revision():
|
|
object.three_phase_available = row[1]
|
|
object.save()
|
|
except ObjectDoesNotExist:
|
|
with transaction.atomic(), reversion.create_revision():
|
|
object = models.Venue(name=row[0], three_phase_available=row[1])
|
|
object.save()
|
|
|
|
def import_rigs(delete=False):
|
|
if delete:
|
|
models.Event.objects.all().delete()
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT r.id, event, person_id, organisation_id, venue, description, status, start_date, start_time, end_date, end_time, access_date, access_time, meet_date, meet_time, meet_info, based_on_id, based_on_type, dry_hire, user_id, payment_method, order_no, payment_received, collectorsid FROM eventdetails AS e INNER JOIN rigs AS r ON e.describable_id = r.id WHERE describable_type = 'Rig' AND venue IS NOT NULL"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
print(row)
|
|
person = models.Person.objects.get(pk=row[2])
|
|
if row[3]:
|
|
organisation = models.Organisation.objects.get(pk=row[3])
|
|
else:
|
|
organisation = None
|
|
venue = models.Venue.objects.get(name__iexact=row[4])
|
|
status = {
|
|
'Booked': models.Event.BOOKED,
|
|
'Provisional': models.Event.PROVISIONAL,
|
|
'Cancelled': models.Event.CANCELLED,
|
|
}
|
|
mic = models.Profile.objects.get(pk=row[19])
|
|
if row[16] and row[17] == "Rig":
|
|
try:
|
|
based_on = models.Event.objects.get(pk=row[16])
|
|
except ObjectDoesNotExist:
|
|
based_on = None
|
|
else:
|
|
based_on = None
|
|
with transaction.atomic(), reversion.create_revision():
|
|
try:
|
|
object = models.Event.objects.get(pk=row[0])
|
|
except ObjectDoesNotExist:
|
|
object = models.Event(pk=row[0])
|
|
object.name = clean_ascii(row[1])
|
|
object.person = person
|
|
object.organisation = organisation
|
|
object.venue = venue
|
|
object.notes = clean_ascii(row[5])
|
|
object.status = status[row[6]]
|
|
object.start_date = row[7]
|
|
object.start_time = row[8]
|
|
object.end_date = row[9]
|
|
object.end_time = row[10]
|
|
if row[11] and row[12]:
|
|
object.access_at = datetime.combine(row[11], row[12])
|
|
if row[13] and row[14]:
|
|
object.meet_at = datetime.combine(row[13], row[14])
|
|
object.meet_info = row[15]
|
|
object.based_on = based_on
|
|
object.dry_hire = row[18]
|
|
object.is_rig = True
|
|
object.mic = mic
|
|
object.payment_method = row[20]
|
|
object.purchase_order = row[21]
|
|
object.payment_received = row[22]
|
|
object.collector = row[23]
|
|
object.save()
|
|
|
|
|
|
def import_eventitem(delete=True):
|
|
if delete:
|
|
models.EventItem.objects.all().delete()
|
|
cursor = setup_cursor()
|
|
if cursor is None:
|
|
return
|
|
sql = """SELECT i.id, r.id, i.name, i.description, i.quantity, i.cost, i.sortorder FROM rig_items AS i INNER JOIN eventdetails AS e ON i.eventdetail_id = e.id INNER JOIN rigs AS r ON e.describable_id = r.id"""
|
|
cursor.execute(sql)
|
|
for row in cursor.fetchall():
|
|
print(row)
|
|
with transaction.atomic():
|
|
try:
|
|
event = models.Event.objects.get(pk=row[1])
|
|
except ObjectDoesNotExist:
|
|
continue
|
|
try:
|
|
object = models.EventItem.objects.get(pk=row[0])
|
|
except ObjectDoesNotExist:
|
|
object = models.EventItem(pk=row[0])
|
|
object.event = event
|
|
object.name = clean_ascii(row[2])
|
|
object.description = clean_ascii(row[3])
|
|
object.quantity = row[4]
|
|
object.cost = row[5]
|
|
object.order = row[6] if row[6] else 0
|
|
object.save()
|
|
with reversion.create_revision():
|
|
event.save()
|
|
|
|
|
|
def main():
|
|
# import_users()
|
|
# import_people()
|
|
# import_organisations()
|
|
# import_vat_rates()
|
|
# import_venues(True)
|
|
# import_events()
|
|
# import_rigs(False)
|
|
import_eventitem(True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|