From a7d23bca21b0b3f41006f75df2ba7033f19a49cd Mon Sep 17 00:00:00 2001
From: tomtom5152 <tom@codedinternet.com>
Date: Wed, 5 Nov 2014 19:43:30 +0000
Subject: [PATCH] Added venue to importer

---
 RIGS/importer.py | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/RIGS/importer.py b/RIGS/importer.py
index eb58acb2..806694f0 100644
--- a/RIGS/importer.py
+++ b/RIGS/importer.py
@@ -90,10 +90,33 @@ def import_vat_rates():
             print("Found: " + object.__str__())
 
 
+def import_venues():
+    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():
+        object, created = models.Venue.objects.get_or_create(name__iexact=row[0])
+        if created:
+            print("Created: " + object.__str__())
+            with transaction.atomic(), reversion.create_revision():
+                object.three_phase_available = row[1]
+                object.save()
+        else:
+            if not object.three_phase_available and row[1] == 1:
+                print("Updating: " + object.__str__())
+                with transaction.atomic(), reversion.create_revision():
+                    object.three_phase_available = row[1]
+                    object.save()
+            else:
+                print("Found: " + object.__str__())
+
 def main():
     # import_people()
     # import_organisations()
-    import_vat_rates()
+    # import_vat_rates()
+    import_venues()
 
 
 if __name__ == "__main__":