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 _get_flags(self, request, goal_slug, suggestion_slug): # noqa
if request.user.is_anonymous():
member = None
else:
member = Member.lookup(request.user, goal_slug)
owns_goal = member and Goal.objects.filter(
slug=goal_slug, owner=member.global_user).exists()
owns_suggestion = (member and suggestion_slug != "none"
and Suggestion.objects.filter(
slug=suggestion_slug,
owner=member.global_user,
goal__slug=goal_slug,
).exists())
data = {
'add_comment': not request.user.is_anonymous(),
'is_member': True if member else False,
'join_goal': not member and goal_slug != "none",
'edit_goal': owns_goal,
'edit_suggestion': owns_suggestion,
'add_review': not owns_suggestion,
'add_goal': goal_slug == "none"
}
return data
def _get_data(self, request, goal_slug, suggestion_slug):
self.member = Member.lookup(request.user, goal_slug)
if self.member:
self.suggestion = get_object_or_404(Suggestion,
slug=suggestion_slug,
goal__slug=goal_slug)
self.review = Review.objects.filter(
revision__suggestion=self.suggestion,
owner=self.member.global_user).first()
def _get_data(self, request, goal_slug, suggestion_slug=None):
self.member = Member.lookup(request.user, goal_slug)
if self.member:
self.goal = get_object_or_404(Goal, slug=goal_slug)
if suggestion_slug:
self.suggestion = get_object_or_404(Suggestion,
slug=suggestion_slug)
else:
self.suggestion = self.queryset.filter(
is_draft=True,
owner=self.member.global_user,
goal__slug=goal_slug).first()
def _get_data(self, request, goal_slug, suggestion_slug):
self.member = Member.lookup(request.user, goal_slug)
if self.member:
self.suggestion = get_object_or_404(
Suggestion,
slug=suggestion_slug,
goal__slug=goal_slug
)
self.review = Review.objects.filter(
revision__suggestion=self.suggestion,
owner=self.member.global_user
).first()
def _get_data(self, request, goal_slug, suggestion_slug=None):
self.member = Member.lookup(request.user, goal_slug)
if self.member:
self.goal = get_object_or_404(
Goal,
slug=goal_slug
)
if suggestion_slug:
self.suggestion = get_object_or_404(
Suggestion,
slug=suggestion_slug
)
else:
self.suggestion = self.queryset.filter(
is_draft=True,
owner=self.member.global_user,
goal__slug=goal_slug
).first()
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 _get_flags(self, request, goal_slug, suggestion_slug): # noqa
if request.user.is_anonymous():
member = None
else:
member = Member.lookup(request.user, goal_slug)
owns_goal = member and Goal.objects.filter(slug=goal_slug, owner=member.global_user).exists()
owns_suggestion = (
member
and suggestion_slug != "none"
and Suggestion.objects.filter(slug=suggestion_slug, owner=member.global_user, goal__slug=goal_slug).exists()
)
data = {
"add_comment": not request.user.is_anonymous(),
"is_member": True if member else False,
"join_goal": not member and goal_slug != "none",
"edit_goal": owns_goal,
"edit_suggestion": owns_suggestion,
"add_review": not owns_suggestion,
"add_goal": goal_slug == "none",
}
return data
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()
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()