]> git.0d.be Git - panikdb.git/blobdiff - panikdb/aa/management/commands/load-csv-users.py
commands: use print as a function
[panikdb.git] / panikdb / aa / management / commands / load-csv-users.py
index 8721db956a561d4fb010318b04358ceb24a15121..663483b3d529c34f648dc6bb59441fc184c9ac1d 100644 (file)
@@ -13,18 +13,21 @@ class Command(BaseCommand):
     def handle(self, filename, verbosity, **kwargs):
         self.verbose = (verbosity > 1)
         for line in csv.reader(file(filename)):
-            lastname, firstname, email = [unicode(x.strip(), 'utf-8') for x in line[:3]]
-            emission_slugs = [x.strip().lower() for x in line[6:] if x]
+            email, lastname, firstname = [unicode(x.strip(), 'utf-8') for x in line[:3]]
+            emission_slugs = [x.strip().lower() for x in line[4:9] if x]
             if not email or not emission_slugs:
                 continue
             if firstname and lastname:
                 username = slugify(u'%s-%s' % (firstname, lastname))
             elif firstname or lastname:
                 username = slugify(firstname or lastname)
+                if len(username) < 8:
+                    username = '%s.%s' % (username, slugify(unicode(emission_slugs[0])))
             else:
                 continue
 
-            print username, email, emission_slugs
+            username = username[:28]
+
             try:
                 user = User.objects.get(username=username)
             except User.DoesNotExist:
@@ -37,6 +40,6 @@ class Command(BaseCommand):
                 try:
                     emissions.append(Emission.objects.get(slug=slug))
                 except Emission.DoesNotExist:
-                    print 'E: missing emission:', slug
+                    print('E: missing emission:', slug)
             user.emissions = emissions
             user.save()