From 2b53efc60cd35b7cb525f2a7eea227258c676360 Mon Sep 17 00:00:00 2001 From: Tom Price Date: Fri, 24 Apr 2015 22:36:19 +0100 Subject: [PATCH] Add updating sequences to importer. --- importer.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/importer.py b/importer.py index 4362e203..c7f3c9e0 100644 --- a/importer.py +++ b/importer.py @@ -345,6 +345,7 @@ def main(): # [x.start() for x in processs] # # Wait for all processs to finish # [x.join() for x in processs] + import_users() import_people(True) import_organisations(True) @@ -354,8 +355,18 @@ def main(): import_rigs(True) import_eventitem(True) import_invoices(True) + + # Do this before doing non rigs else it gets ugly + sql = "SELECT setval(\'\"RIGS_%s_id_seq\"\', (SELECT MAX(id) FROM \"RIGS_%s\"));" % ('event', 'event') + cursor = connections['default'].cursor() + cursor.execute(sql) import_nonrigs(False) + sequences = ['profile', 'person', 'organisation', 'vatrate', 'venue', 'event', 'eventitem', 'invoice', 'payment'] + for seq in sequences: + sql = "SELECT setval(\'\"RIGS_%s_id_seq\"\', (SELECT MAX(id) FROM \"RIGS_%s\"));" % (seq, seq) + cursor = connections['default'].cursor() + cursor.execute(sql) if __name__ == "__main__": main()