Now sample_data add members to a created projects.

greenmine/scrum/management/commands/sample_data.py

@@ -9,7 +9,7 @@ from django.utils.timezone import now
 
 from django.contrib.webdesign import lorem_ipsum
 
-from greenmine.base.models import User
+from greenmine.base.models import User, Role
 from greenmine.scrum.models import *
 
 subjects = [
@@ -42,10 +42,12 @@ class Command(BaseCommand):
 
     @transaction.commit_on_success
     def handle(self, *args, **options):
-        users = []
+        users = [User.objects.get(is_superuser=True)]
         for x in range(10):
             users.append(self.create_user(x))
 
+        role = Role.objects.all()[0]
+
         # projects
         for x in xrange(3):
             # create project
@@ -58,6 +60,9 @@ class Command(BaseCommand):
 
             project.save()
 
+            for user in users:
+                Membership.objects.create(project=project, role=role, user=user)
+
             now_date = now() - datetime.timedelta(30)
 
             # create random milestones