def update(self, instance, validated_data): # noqa
revision = (instance.get_current_revision()
if validated_data['is_draft'] else Revision())
revision.title = validated_data['title']
revision.description = validated_data['description']
revision.suggestion = instance
revision.save()
instance.type = validated_data['type']
instance.is_draft = validated_data['is_draft']
instance.slug = slugify(validated_data['title'])
instance.save()
return instance
def _create_draft_suggestion(self):
suggestion = Suggestion()
suggestion.owner = self.member.global_user
suggestion.goal = self.goal
suggestion.is_draft = True
suggestion.save()
revision = Revision()
revision.title = "not set"
revision.description = "not set"
revision.suggestion = suggestion
revision.save()
return suggestion
def _create_draft_suggestion(self):
suggestion = Suggestion()
suggestion.owner = self.member.global_user
suggestion.goal = self.goal
suggestion.is_draft = True
suggestion.save()
revision = Revision()
revision.title = "not set"
revision.description = "not set"
revision.suggestion = suggestion
revision.save()
return suggestion
def __create_suggestions(self):
self.yoga_bend = Suggestion()
self.yoga_bend.goal = self.become_a_yogi
self.yoga_bend.owner = self.user_mnieber
self.yoga_bend.is_draft = False
self.yoga_bend.image = "suggestions/KroukTom-710x300-crop.jpg"
self.yoga_bend.slug = "bend-backwards"
self.yoga_bend.save()
yoga_bend = Revision()
yoga_bend.title = "Bend backwards"
yoga_bend.description = yoga_bend_content
yoga_bend.suggestion = self.yoga_bend
yoga_bend.save()
self.yoga_meditate = Suggestion()
self.yoga_meditate.goal = self.become_a_yogi
self.yoga_meditate.owner = self.user_mnieber
self.yoga_meditate.is_draft = False
self.yoga_meditate.image = "suggestions/Yoga_Nidra.jpg"
self.yoga_meditate.slug = "meditate-often"
self.yoga_meditate.save()
yoga_meditate = Revision()
yoga_meditate.title = "Meditate often"
yoga_meditate.description = yoga_meditate_content
yoga_meditate.suggestion = self.yoga_meditate
yoga_meditate.save()
self.yoga_twist = Suggestion()
self.yoga_twist.goal = self.become_a_yogi
self.yoga_twist.owner = self.user_anders_om
self.yoga_twist.is_draft = False
self.yoga_twist.image = "suggestions/MC_AM06_00_sized2.jpg"
self.yoga_twist.slug = "keep-twisting"
self.yoga_twist.save()
self.rev_yoga_twist = Revision()
self.rev_yoga_twist.title = "Keep twisting"
self.rev_yoga_twist.description = yoga_twist_content
self.rev_yoga_twist.suggestion = self.yoga_twist
self.rev_yoga_twist.save()
time.sleep(1)
self.rev_yoga_twist_2 = Revision()
self.rev_yoga_twist_2.title = "Keep twisting"
self.rev_yoga_twist_2.description = yoga_twist_2_content
self.rev_yoga_twist_2.suggestion = self.yoga_twist
self.rev_yoga_twist_2.save()
self.yoga_acrobatics = Suggestion()
self.yoga_acrobatics.goal = self.become_a_yogi
self.yoga_acrobatics.owner = self.user_marie_houana
self.yoga_acrobatics.is_draft = False
self.yoga_acrobatics.image = "suggestions/yoga-acrobats.jpg"
self.yoga_acrobatics.slug = "yoga-acrobatics"
self.yoga_acrobatics.save()
yoga_acrobatics = Revision()
yoga_acrobatics.title = "Acrobatics"
yoga_acrobatics.description = yoga_acrobatics_content
yoga_acrobatics.suggestion = self.yoga_acrobatics
yoga_acrobatics.save()
class Command(BaseCommand):
help = \
'Populates the database with example suggestions'
def add_arguments(self, parser):
pass
def __migrate(self):
if settings.DATABASES['default']['NAME'] == ":memory:":
management.call_command('migrate', '--run-syncdb')
else:
management.call_command('migrate', '--run-syncdb')
management.call_command(
'makemigrations', 'goal', 'suggestion', 'review')
management.call_command('migrate', '--fake-initial')
def __make_global_user(self, username, password, first_name, last_name):
user = User.objects.get_or_create(username=username)[0]
user.set_password(password)
user.first_name = first_name
user.last_name = last_name
user.save()
global_user = GlobalUser()
global_user.user = user
global_user.save()
return global_user
def __create_superuser(self):
management.call_command(
'createsuperuser',
'--noinput',
username='******',
email='*****@*****.**'
)
self.user_mnieber = self.__make_global_user(
'mnieber', '***REMOVED***', 'Maarten', 'Nieber')
def __create_users(self):
self.user_anders_om = self.__make_global_user(
"andersom", "***REMOVED***", "Anders", "Om")
self.user_marie_houana = self.__make_global_user(
"mhouana", "***REMOVED***", "Maria", "Houana")
def __create_goal(self):
self.become_a_yogi = Goal()
self.become_a_yogi.owner = self.user_marie_houana
self.become_a_yogi.title = "Become a yogi"
self.become_a_yogi.is_draft = False
self.become_a_yogi.save()
def __create_members(self):
self.yogi_mnieber = Member()
self.yogi_mnieber.global_user = self.user_mnieber
self.yogi_mnieber.goal = self.become_a_yogi
self.yogi_mnieber.save()
self.yogi_anders_om = Member()
self.yogi_anders_om.global_user = self.user_anders_om
self.yogi_anders_om.goal = self.become_a_yogi
self.yogi_anders_om.save()
self.yogi_marie_houana = Member()
self.yogi_marie_houana.global_user = self.user_marie_houana
self.yogi_marie_houana.goal = self.become_a_yogi
self.yogi_marie_houana.save()
def __create_suggestions(self):
self.yoga_bend = Suggestion()
self.yoga_bend.goal = self.become_a_yogi
self.yoga_bend.owner = self.user_mnieber
self.yoga_bend.is_draft = False
self.yoga_bend.image = "suggestions/KroukTom-710x300-crop.jpg"
self.yoga_bend.slug = "bend-backwards"
self.yoga_bend.save()
yoga_bend = Revision()
yoga_bend.title = "Bend backwards"
yoga_bend.description = yoga_bend_content
yoga_bend.suggestion = self.yoga_bend
yoga_bend.save()
self.yoga_meditate = Suggestion()
self.yoga_meditate.goal = self.become_a_yogi
self.yoga_meditate.owner = self.user_mnieber
self.yoga_meditate.is_draft = False
self.yoga_meditate.image = "suggestions/Yoga_Nidra.jpg"
self.yoga_meditate.slug = "meditate-often"
self.yoga_meditate.save()
yoga_meditate = Revision()
yoga_meditate.title = "Meditate often"
yoga_meditate.description = yoga_meditate_content
yoga_meditate.suggestion = self.yoga_meditate
yoga_meditate.save()
self.yoga_twist = Suggestion()
self.yoga_twist.goal = self.become_a_yogi
self.yoga_twist.owner = self.user_anders_om
self.yoga_twist.is_draft = False
self.yoga_twist.image = "suggestions/MC_AM06_00_sized2.jpg"
self.yoga_twist.slug = "keep-twisting"
self.yoga_twist.save()
self.rev_yoga_twist = Revision()
self.rev_yoga_twist.title = "Keep twisting"
self.rev_yoga_twist.description = yoga_twist_content
self.rev_yoga_twist.suggestion = self.yoga_twist
self.rev_yoga_twist.save()
time.sleep(1)
self.rev_yoga_twist_2 = Revision()
self.rev_yoga_twist_2.title = "Keep twisting"
self.rev_yoga_twist_2.description = yoga_twist_2_content
self.rev_yoga_twist_2.suggestion = self.yoga_twist
self.rev_yoga_twist_2.save()
self.yoga_acrobatics = Suggestion()
self.yoga_acrobatics.goal = self.become_a_yogi
self.yoga_acrobatics.owner = self.user_marie_houana
self.yoga_acrobatics.is_draft = False
self.yoga_acrobatics.image = "suggestions/yoga-acrobats.jpg"
self.yoga_acrobatics.slug = "yoga-acrobatics"
self.yoga_acrobatics.save()
yoga_acrobatics = Revision()
yoga_acrobatics.title = "Acrobatics"
yoga_acrobatics.description = yoga_acrobatics_content
yoga_acrobatics.suggestion = self.yoga_acrobatics
yoga_acrobatics.save()
def __create_reviews(self):
self.review_yoga_twist = Review()
self.review_yoga_twist.owner = self.user_marie_houana
self.review_yoga_twist.revision = self.rev_yoga_twist
self.review_yoga_twist.rating = 2.5
self.review_yoga_twist.description = "Not bad!"
self.review_yoga_twist.is_draft = False
self.review_yoga_twist.save()
self.review_yoga_twist_2 = Review()
self.review_yoga_twist_2.owner = self.user_mnieber
self.review_yoga_twist_2.revision = self.rev_yoga_twist_2
self.review_yoga_twist_2.rating = 4
self.review_yoga_twist_2.description = "Good, good, good"
self.review_yoga_twist_2.is_draft = False
self.review_yoga_twist_2.save()
def __create_comments(self):
self.comment_1 = Comment()
self.comment_1.owner = self.user_mnieber
self.comment_1.body = "Thanks for the feedback"
self.comment_1.review = self.review_yoga_twist
self.comment_1.is_draft = False
self.comment_1.save()
self.comment_2 = Comment()
self.comment_2.owner = self.user_anders_om
self.comment_2.body = "I see your point"
self.comment_2.review = self.review_yoga_twist
self.comment_2.reply_to = self.comment_1
self.comment_2.is_draft = False
self.comment_2.save()
def handle(self, *args, **options):
management.call_command('initdb')
self.__migrate()
self.__create_superuser()
self.__create_users()
self.__create_goal()
self.__create_members()
self.__create_suggestions()
self.__create_reviews()
self.__create_comments()
def __create_suggestions(self):
self.yoga_bend = Suggestion()
self.yoga_bend.goal = self.become_a_yogi
self.yoga_bend.owner = self.user_mnieber
self.yoga_bend.is_draft = False
self.yoga_bend.image = "suggestions/KroukTom-710x300-crop.jpg"
self.yoga_bend.slug = "bend-backwards"
self.yoga_bend.save()
yoga_bend = Revision()
yoga_bend.title = "Bend backwards"
yoga_bend.description = yoga_bend_content
yoga_bend.suggestion = self.yoga_bend
yoga_bend.save()
self.yoga_meditate = Suggestion()
self.yoga_meditate.goal = self.become_a_yogi
self.yoga_meditate.owner = self.user_mnieber
self.yoga_meditate.is_draft = False
self.yoga_meditate.image = "suggestions/Yoga_Nidra.jpg"
self.yoga_meditate.slug = "meditate-often"
self.yoga_meditate.save()
yoga_meditate = Revision()
yoga_meditate.title = "Meditate often"
yoga_meditate.description = yoga_meditate_content
yoga_meditate.suggestion = self.yoga_meditate
yoga_meditate.save()
self.yoga_twist = Suggestion()
self.yoga_twist.goal = self.become_a_yogi
self.yoga_twist.owner = self.user_anders_om
self.yoga_twist.is_draft = False
self.yoga_twist.image = "suggestions/MC_AM06_00_sized2.jpg"
self.yoga_twist.slug = "keep-twisting"
self.yoga_twist.save()
self.rev_yoga_twist = Revision()
self.rev_yoga_twist.title = "Keep twisting"
self.rev_yoga_twist.description = yoga_twist_content
self.rev_yoga_twist.suggestion = self.yoga_twist
self.rev_yoga_twist.save()
time.sleep(1)
self.rev_yoga_twist_2 = Revision()
self.rev_yoga_twist_2.title = "Keep twisting"
self.rev_yoga_twist_2.description = yoga_twist_2_content
self.rev_yoga_twist_2.suggestion = self.yoga_twist
self.rev_yoga_twist_2.save()
self.yoga_acrobatics = Suggestion()
self.yoga_acrobatics.goal = self.become_a_yogi
self.yoga_acrobatics.owner = self.user_marie_houana
self.yoga_acrobatics.is_draft = False
self.yoga_acrobatics.image = "suggestions/yoga-acrobats.jpg"
self.yoga_acrobatics.slug = "yoga-acrobatics"
self.yoga_acrobatics.save()
yoga_acrobatics = Revision()
yoga_acrobatics.title = "Acrobatics"
yoga_acrobatics.description = yoga_acrobatics_content
yoga_acrobatics.suggestion = self.yoga_acrobatics
yoga_acrobatics.save()
class Command(BaseCommand):
help = (
'Populates the database with example suggestions'
)
def add_arguments(self, parser):
pass
def __migrate(self):
if settings.DATABASES['default']['NAME'] == ":memory:":
management.call_command('migrate', '--run-syncdb')
else:
management.call_command('migrate', '--run-syncdb')
management.call_command(
'makemigrations', 'goal', 'suggestion', 'review')
management.call_command('migrate', '--fake-initial')
def __make_global_user(self, username, password, first_name, last_name):
user = User.objects.get_or_create(username=username)[0]
user.set_password(password)
user.first_name = first_name
user.last_name = last_name
user.save()
global_user = GlobalUser()
global_user.user = user
global_user.save()
return global_user
def __create_superuser(self):
management.call_command(
'createsuperuser',
'--noinput',
username='******',
email='*****@*****.**'
)
self.user_mnieber = self.__make_global_user(
'mnieber', '***REMOVED***', 'Maarten', 'Nieber')
def __create_users(self):
self.user_anders_om = self.__make_global_user(
"andersom", "***REMOVED***", "Anders", "Om")
self.user_marie_houana = self.__make_global_user(
"mhouana", "***REMOVED***", "Maria", "Houana")
def __create_goal(self):
self.become_a_yogi = Goal()
self.become_a_yogi.owner = self.user_marie_houana
self.become_a_yogi.title = "Become a yogi"
self.become_a_yogi.is_draft = False
self.become_a_yogi.save()
def __create_members(self):
self.yogi_mnieber = Member()
self.yogi_mnieber.global_user = self.user_mnieber
self.yogi_mnieber.goal = self.become_a_yogi
self.yogi_mnieber.save()
self.yogi_anders_om = Member()
self.yogi_anders_om.global_user = self.user_anders_om
self.yogi_anders_om.goal = self.become_a_yogi
self.yogi_anders_om.save()
self.yogi_marie_houana = Member()
self.yogi_marie_houana.global_user = self.user_marie_houana
self.yogi_marie_houana.goal = self.become_a_yogi
self.yogi_marie_houana.save()
def __create_suggestions(self):
self.yoga_bend = Suggestion()
self.yoga_bend.goal = self.become_a_yogi
self.yoga_bend.owner = self.user_mnieber
self.yoga_bend.is_draft = False
self.yoga_bend.image = "suggestions/KroukTom-710x300-crop.jpg"
self.yoga_bend.slug = "bend-backwards"
self.yoga_bend.save()
yoga_bend = Revision()
yoga_bend.title = "Bend backwards"
yoga_bend.description = yoga_bend_content
yoga_bend.suggestion = self.yoga_bend
yoga_bend.save()
self.yoga_meditate = Suggestion()
self.yoga_meditate.goal = self.become_a_yogi
self.yoga_meditate.owner = self.user_mnieber
self.yoga_meditate.is_draft = False
self.yoga_meditate.image = "suggestions/Yoga_Nidra.jpg"
self.yoga_meditate.slug = "meditate-often"
self.yoga_meditate.save()
yoga_meditate = Revision()
yoga_meditate.title = "Meditate often"
yoga_meditate.description = yoga_meditate_content
yoga_meditate.suggestion = self.yoga_meditate
yoga_meditate.save()
self.yoga_twist = Suggestion()
self.yoga_twist.goal = self.become_a_yogi
self.yoga_twist.owner = self.user_anders_om
self.yoga_twist.is_draft = False
self.yoga_twist.image = "suggestions/MC_AM06_00_sized2.jpg"
self.yoga_twist.slug = "keep-twisting"
self.yoga_twist.save()
self.rev_yoga_twist = Revision()
self.rev_yoga_twist.title = "Keep twisting"
self.rev_yoga_twist.description = yoga_twist_content
self.rev_yoga_twist.suggestion = self.yoga_twist
self.rev_yoga_twist.save()
time.sleep(1)
self.rev_yoga_twist_2 = Revision()
self.rev_yoga_twist_2.title = "Keep twisting"
self.rev_yoga_twist_2.description = yoga_twist_2_content
self.rev_yoga_twist_2.suggestion = self.yoga_twist
self.rev_yoga_twist_2.save()
self.yoga_acrobatics = Suggestion()
self.yoga_acrobatics.goal = self.become_a_yogi
self.yoga_acrobatics.owner = self.user_marie_houana
self.yoga_acrobatics.is_draft = False
self.yoga_acrobatics.image = "suggestions/yoga-acrobats.jpg"
self.yoga_acrobatics.slug = "yoga-acrobatics"
self.yoga_acrobatics.save()
yoga_acrobatics = Revision()
yoga_acrobatics.title = "Acrobatics"
yoga_acrobatics.description = yoga_acrobatics_content
yoga_acrobatics.suggestion = self.yoga_acrobatics
yoga_acrobatics.save()
def __create_reviews(self):
self.review_yoga_twist = Review()
self.review_yoga_twist.owner = self.user_marie_houana
self.review_yoga_twist.revision = self.rev_yoga_twist
self.review_yoga_twist.rating = 2.5
self.review_yoga_twist.description = "Not bad!"
self.review_yoga_twist.is_draft = False
self.review_yoga_twist.save()
self.review_yoga_twist_2 = Review()
self.review_yoga_twist_2.owner = self.user_mnieber
self.review_yoga_twist_2.revision = self.rev_yoga_twist_2
self.review_yoga_twist_2.rating = 4
self.review_yoga_twist_2.description = "Good, good, good"
self.review_yoga_twist_2.is_draft = False
self.review_yoga_twist_2.save()
def __create_comments(self):
self.comment_1 = Comment()
self.comment_1.owner = self.user_mnieber
self.comment_1.body = "Thanks for the feedback"
self.comment_1.review = self.review_yoga_twist
self.comment_1.is_draft = False
self.comment_1.save()
self.comment_2 = Comment()
self.comment_2.owner = self.user_anders_om
self.comment_2.body = "I see your point"
self.comment_2.review = self.review_yoga_twist
self.comment_2.reply_to = self.comment_1
self.comment_2.is_draft = False
self.comment_2.save()
def handle(self, *args, **options):
management.call_command('initdb')
self.__migrate()
self.__create_superuser()
self.__create_users()
self.__create_goal()
self.__create_members()
self.__create_suggestions()
self.__create_reviews()
self.__create_comments()
