Changeset a3b83ea for asadb

Timestamp:

Jan 22, 2012, 11:03:38 PM (14 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, space-access, stable, stage

Children:

8a8ba4c

Parents:

2ca295b

git-author:

Alex Dehnert <adehnert@…> (01/22/12 23:03:38)

git-committer:

Alex Dehnert <adehnert@…> (01/22/12 23:03:38)

Message:

Use transaction.commit_on_success for import_db.py

As usual, we get significant speed improvements (~4:45 -> 5 seconds,
so about a 60x improvement...).

File:

• asadb/util/import_db.py (modified) (3 diffs)

asadb/util/import_db.py

@@ -11 +11 @@
     os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'
 
+from django.db import transaction
 import django.contrib.auth.models
 import reversion
@@ -84 +85 @@
     g.save()
 
-if __name__ == '__main__':
-    indb = sys.stdin
-    reader = csv.DictReader(indb)
+
+@transaction.commit_on_success
+def import_groups(reader):
     with reversion.create_revision():
         for line in reader:
@@ -94 +95 @@
         reversion.set_user(importer)
         reversion.set_comment("Groups importer")
+
+
+if __name__ == '__main__':
+    indb = sys.stdin
+    reader = csv.DictReader(indb)
+    import_groups(reader)