def test_goals_by_season(self):
# 2 goals in season 1 by player 1
goal_1 = Goal(scored_by=self.playfor_1, scored_for=self.team_1,
game=self.game_season_1)
goal_1.save()
goal_2 = Goal(scored_by=self.playfor_1, scored_for=self.team_1,
game=self.game_season_1)
goal_2.save()
# 1 goal by player 1 in season 2
goal_x = Goal(scored_by=self.playfor_1, scored_for=self.team_1,
game=self.game_season_2)
goal_x.save()
# 1 goal in season 2 by player 2
goal_3 = Goal(scored_by=self.playfor_2, scored_for=self.team_1,
game=self.game_season_2)
goal_3.save()
# 1 own goal in season 1 by player 3
goal_4 = Goal(scored_by=self.playfor_3,
scored_for=self.team_2, game=self.game_season_1)
goal_4.save()
self.assertIn(self.player_1, self.season_1.scorers())
self.assertNotIn(self.player_2, self.season_1.scorers())
self.assertNotIn(self.player_3, self.season_1.scorers())
self.assertEqual(3, self.season_1.count_goals())
self.assertEqual(2, self.season_1.scorers()[0].num_scored)
def goal_new():
from models import Goal
form = NewGoalForm(request.form)
new_status = "form"
if request.method == "POST":
try:
new_goal = Goal(title=form.title.data, user=current_user.self(), description=form.description.data)
new_goal.save()
new_status = "success"
except OperationError:
new_status = "fail"
return render_template("goal/new.html", form=form, new_status=new_status)
def goal_copy(goal_id):
from models import Goal, Milestone
goal = Goal.objects.get_or_404(id=goal_id)
if (goal.user.id == current_user.get_id()):
abort(410)
else:
new_goal = Goal(title=goal.title, description=goal.description, user=current_user.self(), original=goal)
new_goal.save()
milestones = Milestone.objects(goal=goal.id)
for milestone in milestones:
new_milestone = Milestone(goal=new_goal, message=milestone.message)
new_milestone.save()
return redirect("/goal/%s" % new_goal.id)
def add_problem(request, patient_id):
role = UserProfile.objects.get(user=request.user).role
authenticated = True if (role == 'physician' or role == 'admin') else False
if 'problem_name' in request.POST:
problem = Problem(patient=User.objects.get(id=patient_id), problem_name=request.POST['problem_name'], concept_id=request.POST['concept_id'], authenticated=authenticated)
problem.save()
elif 'goal' in request.POST:
goal = Goal(patient=User.objects.get(id=patient_id), goal=request.POST['goal'])
goal.save()
elif 'todo' in request.POST:
print 'todo'
print request.POST
todo = ToDo(patient=User.objects.get(id=patient_id), todo=request.POST['todo'])
todo.save()
return HttpResponse('added')
def newgoal():
goals = Goal.objects(people=g.user.id)
me = User.objects(id=g.user.id).first()
friends = Set()
if goals:
for goal in goals:
friends.update(goal.people)
friends.update(goal.completed)
if friends:
friends.remove(me)
else:
friends.add(User.objects(username='******').first())
form = GoalForm()
if form.validate_on_submit():
me = User.objects(id=g.user.id).first()
if form.description.data:
goal = Goal(name=form.name.data,
description=form.description.data,
end=form.end.data,
people=[me])
else:
goal = Goal(name=form.name.data, end=form.end.data, people=[me])
goal.save()
feeditem = GoalRequest(
user=me,
goal=goal,
message='Your friend invited you to join their goal')
people = []
for email in form.people.data:
friend = User.objects(username=email).first()
friend.feed.append(feeditem)
friend.goalrequest.append(goal)
friend.save()
flash('Goal added!')
return redirect(url_for('goals'))
else:
print "Nope"
return render_template('newgoal.html', users=friends, form=form)
def newgoal():
goals = Goal.objects(people = g.user.id)
me = User.objects(id = g.user.id).first()
friends = Set()
if goals:
for goal in goals:
friends.update(goal.people)
friends.update(goal.completed)
if friends:
friends.remove(me)
else:
friends.add(User.objects(username = '******').first())
form = GoalForm()
if form.validate_on_submit():
me = User.objects(id = g.user.id).first()
if form.description.data:
goal = Goal(name = form.name.data, description = form.description.data, end = form.end.data, people = [me] )
else:
goal = Goal(name = form.name.data, end = form.end.data, people = [me] )
goal.save()
feeditem = GoalRequest(user=me,goal=goal,message='Your friend invited you to join their goal')
people = []
for email in form.people.data:
friend = User.objects(username = email).first()
friend.feed.append(feeditem)
friend.goalrequest.append(goal)
friend.save()
flash ('Goal added!')
return redirect(url_for('goals'))
else:
print "Nope"
return render_template('newgoal.html',users = friends, form = form)
def save(self, current_user, goal=None):
if goal:
goal.updated_by = current_user
for f in ['is_achieved', 'title']:
if self.cleaned_data.get(f) is not None:
setattr(goal, f, self.cleaned_data[f])
goal.updated_at = datetime.utcnow()
goal.updated_by = current_user
else:
goal = Goal(title=self.cleaned_data.get("title", ""),
created_by=current_user,
updated_by=current_user,
is_achieved=self.cleaned_data.get(
"is_achieved", False))
goal.save()
return goal
def add_problem(request, patient_id):
role = UserProfile.objects.get(user=request.user).role
authenticated = True if (role == 'physician' or role == 'admin') else False
if 'problem_name' in request.POST:
problem = Problem(patient=User.objects.get(id=patient_id),
problem_name=request.POST['problem_name'],
concept_id=request.POST['concept_id'],
authenticated=authenticated)
problem.save()
elif 'goal' in request.POST:
goal = Goal(patient=User.objects.get(id=patient_id),
goal=request.POST['goal'])
goal.save()
elif 'todo' in request.POST:
# print 'todo'
# print request.POST
todo = ToDo(patient=User.objects.get(id=patient_id),
todo=request.POST['todo'])
todo.save()
return HttpResponse('added')
def add_test_data():
mipt_team = Team(name=u"Физтех", start_date=datetime.date.today())
mipt_team.save()
anytime_team = Team(name=u"Энитайм", start_date=datetime.date.today())
anytime_team.save()
den = Person(first_name=u"Денис",
last_name=u"Щигельский",
position=Person.BACK,
is_active=True,
is_captain=False,
start_date=datetime.date.today(),
cell_phone=u"+79151164158",
email=u"*****@*****.**",
team=mipt_team)
den.save()
stan = Person(first_name=u"Илья",
last_name=u"Станиславский",
position=Person.HALFBACK,
is_active=True,
is_captain=False,
start_date=datetime.date.today(),
cell_phone=u"+79670614948",
team=mipt_team)
stan.save()
burov = Person(first_name=u"Александр",
last_name=u"Буров",
position=Person.FORWARD,
is_active=True,
is_captain=True,
start_date=datetime.date.today(),
cell_phone=u"89197711249",
team=mipt_team)
burov.save()
ahyan = Person(first_name=u"Ара",
last_name=u"Ахян",
position=Person.FORWARD,
is_active=True,
is_captain=True,
start_date=datetime.date.today(),
cell_phone=u"89123711249",
team=anytime_team)
ahyan.save()
mipt_anytime = Match(date=datetime.date.today(),
time=datetime.datetime.now(),
home_team=mipt_team,
guest_team=anytime_team,
guest_team_score=0,
home_team_score=0)
mipt_anytime.save()
anytime_mipt = Match(date=datetime.date.today().replace(day=30),
time=datetime.datetime.now(),
home_team=anytime_team,
guest_team=mipt_team,
guest_team_score=0,
home_team_score=0)
anytime_mipt.save()
g1 = Goal(player_scored=stan,
player_assisted=burov,
own_goal=False,
match=mipt_anytime,
minute=11,
is_penalty=False)
g1.save()
print mipt_anytime.home_team_score, mipt_anytime.guest_team_score
g2 = Goal(player_scored=stan,
player_assisted=den,
own_goal=False,
match=mipt_anytime,
minute=15,
is_penalty=False)
g2.save()
print mipt_anytime.home_team_score, mipt_anytime.guest_team_score
g3 = Goal(player_scored=burov,
own_goal=True,
match=mipt_anytime,
minute=58,
is_penalty=False)
g3.save()
print mipt_anytime.home_team_score, mipt_anytime.guest_team_score
g4 = Goal(player_scored=ahyan,
own_goal=False,
match=mipt_anytime,
minute=59,
is_penalty=False)
g4.save()
print mipt_anytime.home_team_score, mipt_anytime.guest_team_score
card1 = Card(type='Y',
person=den,
minute=24)
card1.save()
def add(request):
name = request.POST.get('goal', None)
if name and name != '':
goal = Goal(name=name, user=request.user)
goal.save()
return HttpResponseRedirect(reverse('user_index', kwargs={'username':request.user.username}))
