update slug script to ensure unique / ugly hack

2012-02-20 19:40:22 +05:30 · 2012-02-20 19:40:22 +05:30 · 974a57ee1a
commit 974a57ee1a
parent c1aedefbc7
1 changed files with 4 additions and 1 deletions
--- a/chaloBEST/imports/make_slugs.py
+++ b/chaloBEST/imports/make_slugs.py
@ -4,7 +4,10 @@ from mumbai.models import *
 def do():
    for cls in [Road, Area, Stop]:
        for obj in cls.objects.all():
-            obj.slug = slugify(obj.display_name)
+            slug = slugify(obj.display_name)
+            if cls.objects.filter(slug=slug).count() > 1:
+                slug += "2"
+            obj.slug = slug
            obj.save()
    for r in Route.objects.all():
        r.slug = r.alias