Script to fix bad permit type

2 files changed, 32 insertions, 0 deletions

diff --git a/archaeological_files/management/commands/__init__.py b/archaeological_files/management/commands/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/archaeological_files/management/commands/__init__.py
diff --git a/archaeological_files/management/commands/fix_permittype.py b/archaeological_files/management/commands/fix_permittype.py
new file mode 100644
index 000000000..c7013266d
--- /dev/null
+++ b/archaeological_files/management/commands/fix_permittype.py
@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import csv
+import datetime, time
+
+from django.conf import settings
+from django.core.management.base import BaseCommand, CommandError
+
+from archaeological_files.models import File, PermitType
+
+replaced = (('cu', 'CU'),
+            ('pc', 'PC'),
+            ('', 'NP')
+            )
+
+
+class Command(BaseCommand):
+    help = "Update file types"
+
+    def handle(self, *args, **options):
+        self.stdout.write("* Update file types\n")
+        for origin, dest in replaced:
+            permit_type = PermitType.objects.get(txt_idx=dest)
+            if origin:
+                File.objects.filter(permit_type__txt_idx=origin).update(
+                                            permit_type=permit_type)
+                PermitType.objects.filter(txt_idx=origin).update(available=False)
+            else:
+                File.objects.filter(permit_type__isnull=True).update(
+                                            permit_type=permit_type)
+