add init discussions and textmessages scripts

JoinTheCityREST/app/Discussion/management/__init__.py

+__author__ = 'Rafal'

JoinTheCityREST/app/Discussion/management/commands/__init__.py

+__author__ = 'Rafal'

JoinTheCityREST/app/Discussion/management/commands/init_discussions.py

+from django.contrib.auth.models import User
+
+from app.Discussion.models import TextMessage, Discussion
+from app.Tag.models import Tag
+
+__author__ = 'Rafal'
+from django.core.management.base import BaseCommand
+
+
+class Command(BaseCommand):
+    help = 'Closes the specified poll for voting'
+
+    def add_arguments(self, parser):
+        parser.add_argument('poll_id', nargs='+', type=int)
+
+    def handle(self, *args, **options):
+        discussions = [
+            ['admin', 'Czy jest w okolicy jakis fryzjer?',['Nie znam','Polecam fryzjera w Kapitolu', 'Lol, on jest do dupy', '@_@'],'Miasteczko AGH'],
+            ]
+        print "Creating Discussions"
+        for discussion in discussions:
+            print "Create {0} ask {1}".format(discussion[0], discussion[1])
+            user = User.objects.filter(username=discussion[0])[0]
+            answers = []
+            for messages in discussion[2]:
+                answers.append(TextMessage.objects.filter(text=messages)[0])
+            tag = Tag.objects.filter(name=discussion[3])[0]
+            new_discussion = Discussion.objects.create(user=user, question=discussion[1], tag=tag)
+            new_discussion.message = answers
+            new_discussion.save()
\ No newline at end of file

JoinTheCityREST/app/Discussion/management/commands/init_textmessages.py

+from django.contrib.auth.models import User
+from app.Discussion.models import TextMessage
+from django.utils import timezone
+
+__author__ = 'Rafal'
+from django.core.management.base import BaseCommand
+
+
+class Command(BaseCommand):
+    help = 'Closes the specified poll for voting'
+
+    def add_arguments(self, parser):
+        parser.add_argument('poll_id', nargs='+', type=int)
+
+    def handle(self, *args, **options):
+        text_messages = [
+            ['Rafal', 'Nie znam'],
+            ['Klaudia', 'Polecam fryzjera w Kapitolu'],
+            ['Antek', 'Lol, on jest do dupy'],
+            ['Rafal', '@_@'],
+            ]
+        print "Creating Messages"
+        for text_message in text_messages:
+            print "Create {0} say : {1}".format(text_message[0], text_message[1])
+            user = User.objects.filter(username=text_message[0])[0]
+            new_text_message = TextMessage.objects.create(user=user,text=text_message[1], created_at=timezone.now())
+            new_text_message.save()
\ No newline at end of file