Merge pull request #401 from davidfsmith/master

Fix for deleterevisions on Py3
etianen · May 7, 2015 · 9584eec · 9584eec
2 parents e419121 + 561ea9b
commit 9584eec
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/reversion/management/commands/deleterevisions.py b/src/reversion/management/commands/deleterevisions.py
@@ -195,6 +195,10 @@ def handle(self, *app_labels, **options):
 
         # Ask confirmation
         if confirmation:
+            try:
+                raw_input = raw_input
+            except NameError:
+                raw_input = input
             choice = raw_input("Are you sure you want to delete theses revisions? [y|N] ")
             if choice.lower() != "y":
                 print("Aborting revision deletion.")
@@ -212,4 +216,4 @@ def handle(self, *app_labels, **options):
             for item in revision_query:
                 item.delete()
 
-        print("Done")
+        print("Done")