mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 05:22:16 +00:00
125 lines
4.0 KiB
Python
125 lines
4.0 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 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 main():
|
|
# import_people()
|
|
# import_organisations()
|
|
# import_vat_rates()
|
|
import_venues(True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|