Moved fixture creation scripts to manage commands.

Author: kevinastone

Makefile

@@ -7,8 +7,8 @@ create_database:
 	./manage.py createsuperuser --username=root [email protected] --noinput
 
 make_fixtures:
-	DJANGO_SETTINGS_MODULE='example.settings' ./scripts/create_users.py
-	DJANGO_SETTINGS_MODULE='example.settings' ./scripts/create_posts.py
-	DJANGO_SETTINGS_MODULE='example.settings' ./scripts/create_photos.py
+	./manage.py create_users
+	./manage.py create_posts
+	./manage.py create_photos
 
 all: clean create_database make_fixtures

example/api/management/__init__.py

@@ -0,0 +1,19 @@
+import os
+import os.path
+
+from django.core.management.base import BaseCommand, CommandError
+from django.conf import settings
+
+from example.api.models import Photo, Post
+
+
+class Command(BaseCommand):
+	sample_dir = 'samples'
+
+	def handle(self, *args, **options):
+		sample_images = [os.path.join(self.sample_dir, fn) for fn in os.listdir(os.path.join(settings.MEDIA_ROOT, self.sample_dir))]
+
+		posts = Post.objects.all()
+
+		for i, image in enumerate(sample_images):
+			Photo.objects.create(post=posts[i % posts.count()], image=image)

example/api/management/commands/__init__.py

@@ -0,0 +1,14 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.conf import settings
+
+from example.api.models import Post, User
+
+bodies = ['This is a great post', 'Another thing I wanted to share']
+
+
+class Command(BaseCommand):
+    def handle(self, *args, **options):
+        users = User.objects.all()
+
+        for i, body in enumerate(bodies):
+            Post.objects.create(author=users[i % users.count()], title="Title #{}".format(i + 1), body=body)

example/api/management/commands/create_photos.py

@@ -0,0 +1,12 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.conf import settings
+
+from example.api.models import User
+
+
+class Command(BaseCommand):
+    def handle(self, *args, **options):
+        users = ['Bob', 'Sally', 'Joe', 'Rachel']
+        for user in users:
+            username = user.lower()
+            User.objects.create(username=username, email="{}@example.com".format(username), first_name=user)