def goal_new():
form = GoalForm()
if form.add_subgoal.data:
form.subgoals.append_entry()
form.validate = False
elif any([sg.delete_subgoal.data for sg in form.subgoals.entries]):
for sg in form.subgoals.entries:
if sg.delete_subgoal.data:
form.subgoals.remove(sg)
form.validate = False
else:
if form.validate_on_submit():
goal = Goal(title=form.title.data,
description=form.description.data,
duedate=form.duedate.data,
user_id=current_user.id)
for sg in form.subgoals.entries:
goal.subgoals.append(
Subgoal(title=sg.title.data, duedate=sg.duedate.data))
db.session.add(goal)
db.session.commit()
flash("Goal has been successfully created!")
return redirect(url_for('goal_view', goal_id=goal.id))
return render_template('goal_new.html',
title="Master Lao - Create New Goal",
form=form)
def create_goal(db, title, user_id, description=None, duedate=None):
goal = Goal(title=title, user_id=user_id,
description=description, duedate=duedate)
db.session.add(goal)
db.session.commit()
return goal
def test_goal(test_client, init_database):
u = User.query.filter_by(username=conftest.TEST_USER_USERNAME).first()
current_time = datetime.utcnow()
goal = Goal(title="My Exercise",
motivation="Why am i motivated to do this",
acceptance_criteria="My acceptance criteria",
reward="how will i reward myself",
frequency=5,
frequency_activity_type=1,
timestamp=current_time,
last_updated=current_time,
user_id=u.id)
db.session.add(goal)
db.session.commit()
load_goal = Goal.query.filter_by(user_id=u.id).first()
assert Goal.query.filter_by(user_id=u.id).count() == 1
assert load_goal.title == goal.title
assert load_goal.motivation == goal.motivation
assert load_goal.acceptance_criteria == goal.acceptance_criteria
assert load_goal.reward == goal.reward
assert load_goal.frequency == goal.frequency
assert load_goal.frequency_activity_type == goal.frequency_activity_type
# ignore microseconds when comparing the timestamps
assert "Goal" in repr(load_goal)
assert "Goal" in str(load_goal)
def set_goal():
"""
sets a user's exercise goal
:return:
:rtype:
"""
form = GoalForm()
is_add_goal = True
if form.validate_on_submit():
goal = Goal(title=form.title.data,
motivation=form.motivation.data,
acceptance_criteria=form.acceptance_criteria.data,
reward=form.reward.data,
frequency=form.frequency.data,
frequency_activity_type=form.frequency_activity_type.data,
duration_activity_type=form.duration_activity_type.data,
duration=form.duration.data,
distance_activity_type=form.distance_activity_type.data,
distance=form.distance.data,
user_id=current_user.get_id())
db.session.add(goal)
db.session.commit()
flash('Well done on setting yourself a goal')
return redirect(url_for('main.exercise_log'))
return render_template('goal/edit_goal.html',
title='Set a goal',
form=form,
is_add_goal=is_add_goal)
def init_db():
print("Start db init")
db.session.remove()
db.drop_all()
db.create_all()
u1 = User(username="******", first_name="Ivan",
last_name="Fedorovsky", email="*****@*****.**")
u2 = User(username="******", first_name="Dzimurhan",
last_name="Zaliphan", email="*****@*****.**")
u3 = User(username="******", first_name="Zope",
last_name="Zope", email="*****@*****.**")
u4 = User(username="******", first_name="Lashar",
last_name="Lasharan", email="*****@*****.**")
u1.set_password('Password')
u2.set_password('Password')
u3.set_password('Password')
u4.set_password('Password')
g1 = Goal(title="Goal1")
g2 = Goal(title="Goal2", description="Description",
duedate=datetime.utcnow() + relativedelta(months=1))
g3 = Goal(title="Goal3", description="Description",
duedate=datetime.utcnow() + relativedelta(months=2))
s1 = Subgoal(title="Subgoal Test #1", duedate=datetime.utcnow())
s2 = Subgoal(title="Subgoal Test #2")
s3 = Subgoal(title="Subgoal Test #3")
g3.subgoals.append(s1)
g3.subgoals.append(s2)
g3.subgoals.append(s3)
u1.goals.append(g1)
u1.goals.append(g2)
u1.goals.append(g3)
db.session.add_all([u1, u2, u3, u4])
db.session.commit()
assert User.query.count() == 4
print("Finish db init")
def create_goal():
form = GoalForm()
form['csrf_token'].data = request.cookies['csrf_token']
if form.validate_on_submit():
data = Goal()
form.populate_obj(data)
data.userId = current_user.id
db.session.add(data)
db.session.commit()
return data.to_dict()
return 'invalid info'
def goals():
print(f"REQ: {request.form}")
if request.method == 'POST':
result = request.form.copy()
result['user_id'] = session['USER_ID']
new_goal = Goal()
new_goal.populate(result)
db.session.add(new_goal)
db.session.commit()
stats = get_goal_stats(session['USER_ID'])
print(f"GOAL: {stats}")
return render_template('goals.html', stats=stats, users=get_users())
def add_goal():
u = User.query.filter_by(username=TEST_USER_USERNAME).first()
current_time = datetime.utcnow()
goal = Goal(title="My Exercise",
motivation="Why am i motivated to do this",
acceptance_criteria="My acceptance criteria",
reward="how will i reward myself",
frequency=5,
frequency_activity_type=1,
timestamp=current_time,
last_updated=current_time,
user_id=u.id)
db.session.add(goal)
db.session.commit()
def add_goal():
form = GoalForm()
if form.validate_on_submit():
goal = Goal(weight=form.weight.data,
days=form.days.data,
start_date=form.start_date.data,
person=current_user)
db.session.add(goal)
db.session.commit()
flash('Your Goal has been created!', 'success')
return redirect(url_for('display_goal'))
return render_template('add_goal.html',
title='New Goal',
form=form,
legend='New Goal')
def addGoal(current_user):
data = request.json
splitTargetDate = data['targetDate'].split('-')
joinedTargetDate = ' '.join(splitTargetDate)
datetimeOfTarget = datetime.strptime(joinedTargetDate, '%Y %m %d')
userSketchbook = Sketchbook.query.filter(
Sketchbook.owner_id == current_user.id).first()
newGoal = Goal(owner_id=current_user.id,
Sketchbook_id=userSketchbook.id,
title=data['title'],
description=data['description'],
target=data['target'],
targetdate=datetimeOfTarget,
timestamp=datetime.now())
db.session.add(newGoal)
db.session.commit()
# check for goals that have passed finished date, and delete them
goalsForCurrUser = Goal.query.filter(
Goal.owner_id == current_user.id).all()
currDate = datetime.now()
for goal in goalsForCurrUser:
if goal.targetdate < currDate:
datapoints = Datapoint.query.filter(
Datapoint.goal_id == goal.id).all()
for datapoint in datapoints:
db.session.delete(datapoint)
db.session.delete(goal)
db.session.commit()
returnDict = {
newGoal.Sketchbook_id: {
newGoal.id: {
'id': newGoal.id,
'owner_id': newGoal.owner_id,
'sketchbook_id': newGoal.Sketchbook_id,
'title': newGoal.title,
'description': newGoal.description,
'target': newGoal.target,
'targetdate': newGoal.targetdate,
'timestamp': newGoal.timestamp
}
}
}
return returnDict
def new(request):
current_goal = Goal.objects.filter(user=request.user).first()
if current_goal:
return redirect('app_goals_timeline', goal_id=current_goal.id)
if request.method == 'GET':
return render(request, 'goals/new.html',
{'timezones': pytz.common_timezones})
if request.method != 'POST':
return HttpResponseNotAllowed(['GET', 'POST'])
is_valid = 'text' in request.POST and \
'first_step' in request.POST and \
'timezone' in request.POST
if not is_valid:
return HttpResponseBadRequest()
start = timezone.now()
goal = Goal(user=request.user,
timezone=request.POST['timezone'],
text=request.POST['text'],
start=start)
with transaction.atomic():
logger.info('Creating goal user=%s tz=%s' %
(goal.user.email, goal.timezone))
goal.save()
logger.info('Creating first step goal=%s user=%s' %
(goal.id, request.user.email))
first_step = goal.create_step(request.POST['first_step'],
goal.start,
commit=True)
new_goal.send('app.views.goals.new', goal=goal)
new_step.send('app.views.goals.new', step=first_step)
return redirect('app_steps_start', goal_id=goal.id, step_id=first_step.id)
def load_goals(user_id):
if (request.method == 'GET'):
goals = db.session.query(Goal).filter(Goal.user_id == user_id)
goals_dict = {}
for goal in goals:
goals_dict[goal.id] = goal.to_dict()
return {'goals': goals_dict}, 200
elif (request.method == 'POST'):
data = request.get_json()
goal = Goal(description=data['description'],
amount=data['amount'],
completion_year=data['completion_year'],
completion_month=data['completion_month'],
user_id=user_id)
db.session.add(goal)
db.session.commit()
goals = db.session.query(Goal).filter(Goal.user_id == user_id)
goals_dict = {}
for goal in goals:
goals_dict[goal.id] = goal.to_dict()
return {'goals': goals_dict}, 200
elif (request.method == 'PATCH'):
data = request.get_json()
id = data['id']
goal = Goal.query.get(id)
goal.is_complete = not goal.is_complete
db.session.add(goal)
db.session.commit()
goals = db.session.query(Goal).filter(Goal.user_id == user_id)
goals_dict = {}
for goal in goals:
goals_dict[goal.id] = goal.to_dict()
return {'goals': goals_dict}, 200
elif (request.method == 'DELETE'):
data = request.get_json()
id = data['id']
goal = Goal.query.get(id)
db.session.delete(goal)
db.session.commit()
goals = db.session.query(Goal).filter(Goal.user_id == user_id)
goals_dict = {}
for goal in goals:
goals_dict[goal.id] = goal.to_dict()
return {'goals': goals_dict}, 200
def goals():
if request.method == 'POST':
new_goal = Goal()
new_goal.populate(request.form)
db.session.add(new_goal)
db.session.commit()
return "True"
elif request.method == 'GET':
user_id = request.args.get('user_id')
action = request.args.get('type')
recent_goal = Goal.query.filter_by(user_id=user_id).order_by(
Goal.set_time)[::-1]
if not recent_goal:
return "No goals found"
recent_goal = recent_goal[0]
print(recent_goal)
if action == "steps":
return str(recent_goal.steps)
elif action == "distance":
return str(recent_goal.distance)
return "Incorrect parameters"
def sync():
upload_goals = request.get_json()
goals = upload_goals.get('goals')
for goal in goals:
title = goal.get('title')
timestamp = goal.get('timestamp')
level = goal.get('level')
parent = goal.get('parent')
start = goal.get('start')
over = goal.get('over')
items = goal.get('items')
username = goal.get('username')
user_name = username
mGoal = Goal(level, parent, title, start, over, timestamp, 9, items,
username)
existGoal = Goal.query.filter_by(title=title,
username=username).first()
if goal.get('status') == 1:
if existGoal is not None:
db.session.delete(existGoal) # 增
db.session.add(mGoal) # 增
elif goal.get('status') == 2:
if existGoal is not None:
db.session.delete(existGoal) # 删除
elif goal.get('status') == 3:
if existGoal is not None:
if existGoal.timestamp < timestamp:
db.session.delete(existGoal)
db.session.add(mGoal) # 改
db.session.commit()
goals_in_db = Goal.query.filter_by(
username=upload_goals.get('username')).all()
contList = []
for goal in goals_in_db:
contList.append(goal.serialize())
return formattingData(200, "success", contList)
"""
run one time to load data from json file teachers.json to database test.db
"""
import json
from app.models import Teacher, Goal, db
from app import app
db.init_app(app)
app.app_context().push()
for goal, rus, favicon in (("travel", "путешествй", '⛱'),
("study", "учебы", '🏫'), ("work", "рабoты", '🏢'),
("relocate", "переезда", '🚜')):
gl = Goal(title=goal,
title_rus=f"Для {rus}",
image_location=f"{goal}.png",
favicon=favicon)
db.session.add(gl)
with open("json_data/teachers.json", "r") as f:
teachers = json.load(f)
for t in teachers:
teacher = Teacher(
name=t["name"],
about=t["about"],
rating=t["rating"],
picture=t["picture"],
price=t["price"],
)
with open("data.json", "r", encoding="utf-8") as f:
data = json.loads(f.read())
with open("data.json", "r", encoding="utf-8") as f:
data = json.loads(f.read())
# Импорт целей
db.session.execute(text("DELETE FROM goals;"))
goals = []
g = []
ind = 0
for i, val in data["goals"].items():
goals.append(
Goal(id=ind,
name=i,
desc=val,
style=data["goalstyle"][i],
icon=data["goalicon"][i]))
g.append(i)
db.session.add(goals[ind])
ind += 1
# Импорт справочника Дни
i = 1
s = "INSERT INTO days(id, day, day_desc)\n VALUES"
for d, desc in data["days"].items():
s += f"({i}, '{d}', '{desc}'),\n"
i += 1
db.session.execute(text("DELETE FROM days;"))
db.session.execute(text(s[0:-2] + ";"))
dotenv_path = os.path.join(os.path.dirname(__file__), ".env")
if os.path.exists(dotenv_path):
load_dotenv(dotenv_path)
engine = create_engine(os.getenv("DEVELOPMENT_DATABASE_URI"))
session = sessionmaker()
session.configure(bind=engine)
s = session()
goal_in_db = dict()
for g_key, g_value in goals.items():
# Model GOAL
g = Goal(name=g_key, name_ru=g_value)
s.add(g)
goal_in_db[g_key] = g
for teacher in teachers:
# Model TEACHER
t = Teacher(
name=teacher["name"],
about=teacher["about"],
rating=teacher["rating"],
picture=teacher["picture"],
price=teacher["price"],
)
# Model Teacher_goal
assert total_distance_by_exercise_type == {
1: 0,
2: 0,
3: 0,
4: 10,
5: 0,
6: 0
}
weekly_totals_test_data = [
([
Goal(title="My Exercise",
motivation="Why am i motivated to do this",
acceptance_criteria="My acceptance criteria",
reward="how will i reward myself",
frequency=5,
frequency_activity_type=-1,
user_id=1)
], [
Activity(id=1,
type=1,
title='title',
duration=20,
iso_timestamp='2020-06-16T04:58:33.302785+05:00')
], [(20, 0, 0)]),
([
Goal(title="My Exercise",
motivation="Why am i motivated to do this",
acceptance_criteria="My acceptance criteria",
reward="how will i reward myself",
def build_goal_from_json(json):
return Goal(title=json["title"])