API - reformat python files w/ black

This commit is contained in:
Sam 2019-08-28 13:25:39 +02:00
parent d2adeed805
commit d914951c61
30 changed files with 2924 additions and 3170 deletions

View File

@ -42,8 +42,9 @@ def create_app():
if app.debug:
logging.getLogger('sqlalchemy').setLevel(logging.WARNING)
logging.getLogger('sqlalchemy'
).handlers = logging.getLogger('werkzeug').handlers
logging.getLogger('sqlalchemy').handlers = logging.getLogger(
'werkzeug'
).handlers
logging.getLogger('sqlalchemy.orm').setLevel(logging.WARNING)
logging.getLogger('flake8').propagate = False
appLog.setLevel(logging.DEBUG)
@ -58,7 +59,7 @@ def create_app():
)
response.headers.add(
'Access-Control-Allow-Methods',
'GET,PUT,POST,DELETE,PATCH,OPTIONS'
'GET,PUT,POST,DELETE,PATCH,OPTIONS',
)
return response

View File

@ -8,16 +8,25 @@ from flask import Blueprint, current_app, jsonify, request, send_file
from sqlalchemy import exc
from ..users.utils import (
User, authenticate, can_view_activity, verify_extension
User,
authenticate,
can_view_activity,
verify_extension,
)
from .models import Activity
from .utils import (
ActivityException, create_activity, edit_activity, get_absolute_file_path,
get_datetime_with_tz, process_files
ActivityException,
create_activity,
edit_activity,
get_absolute_file_path,
get_datetime_with_tz,
process_files,
)
from .utils_format import convert_in_duration
from .utils_gpx import (
ActivityGPXException, extract_segment_from_gpx_file, get_chart_data
ActivityGPXException,
extract_segment_from_gpx_file,
get_chart_data,
)
activities_blueprint = Blueprint('activities', __name__)
@ -177,8 +186,9 @@ def get_activities(auth_user_id):
_, date_from = get_datetime_with_tz(user.timezone, date_from)
date_to = params.get('to')
if date_to:
date_to = datetime.strptime(f'{date_to} 23:59:59',
'%Y-%m-%d %H:%M:%S')
date_to = datetime.strptime(
f'{date_to} 23:59:59', '%Y-%m-%d %H:%M:%S'
)
_, date_to = get_datetime_with_tz(user.timezone, date_to)
distance_from = params.get('distance_from')
distance_to = params.get('distance_to')
@ -191,46 +201,59 @@ def get_activities(auth_user_id):
order = params.get('order')
sport_id = params.get('sport_id')
per_page = int(params.get('per_page')) if params.get('per_page') else 5
activities = Activity.query.filter(
activities = (
Activity.query.filter(
Activity.user_id == auth_user_id,
Activity.sport_id == sport_id if sport_id else True,
Activity.activity_date >= date_from if date_from else True,
Activity.activity_date < date_to + timedelta(seconds=1)
if date_to else True,
Activity.distance >= int(distance_from) if distance_from else True,
if date_to
else True,
Activity.distance >= int(distance_from)
if distance_from
else True,
Activity.distance <= int(distance_to) if distance_to else True,
Activity.moving >= convert_in_duration(duration_from)
if duration_from else True,
if duration_from
else True,
Activity.moving <= convert_in_duration(duration_to)
if duration_to else True,
if duration_to
else True,
Activity.ave_speed >= float(ave_speed_from)
if ave_speed_from else True,
if ave_speed_from
else True,
Activity.ave_speed <= float(ave_speed_to)
if ave_speed_to else True,
if ave_speed_to
else True,
Activity.max_speed >= float(max_speed_from)
if max_speed_from else True,
if max_speed_from
else True,
Activity.max_speed <= float(max_speed_to)
if max_speed_to else True,
).order_by(
if max_speed_to
else True,
)
.order_by(
Activity.activity_date.asc()
if order == 'asc'
else Activity.activity_date.desc()
).paginate(
page, per_page, False
).items
)
.paginate(page, per_page, False)
.items
)
response_object = {
'status': 'success',
'data': {
'activities': [activity.serialize(params)
for activity in activities]
}
'activities': [
activity.serialize(params) for activity in activities
]
},
}
code = 200
except Exception as e:
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code
@ -327,7 +350,8 @@ def get_activity(auth_user_id, activity_id):
if activity:
response_object, code = can_view_activity(
auth_user_id, activity.user_id)
auth_user_id, activity.user_id
)
if response_object:
return jsonify(response_object), code
@ -340,9 +364,7 @@ def get_activity(auth_user_id, activity_id):
response_object = {
'status': status,
'data': {
'activities': activities_list
}
'data': {'activities': activities_list},
}
return jsonify(response_object), code
@ -353,13 +375,14 @@ def get_activity_data(auth_user_id, activity_id, data_type, segment_id=None):
content = ''
if activity:
response_object, code = can_view_activity(
auth_user_id, activity.user_id)
auth_user_id, activity.user_id
)
if response_object:
return jsonify(response_object), code
if not activity.gpx or activity.gpx == '':
response_object = {
'status': 'fail',
'message': f'No gpx file for this activity (id: {activity_id})'
'message': f'No gpx file for this activity (id: {activity_id})',
}
return jsonify(response_object), 400
@ -372,19 +395,16 @@ def get_activity_data(auth_user_id, activity_id, data_type, segment_id=None):
content = f.read()
if segment_id is not None:
content = extract_segment_from_gpx_file(
content,
segment_id
content, segment_id
)
except ActivityGPXException as e:
appLog.error(e.message)
response_object = {'status': e.status,
'message': e.message}
response_object = {'status': e.status, 'message': e.message}
code = 404 if e.status == 'not found' else 500
return jsonify(response_object), code
except Exception as e:
appLog.error(e)
response_object = {'status': 'error',
'message': 'internal error'}
response_object = {'status': 'error', 'message': 'internal error'}
return jsonify(response_object), 500
status = 'success'
@ -395,11 +415,15 @@ def get_activity_data(auth_user_id, activity_id, data_type, segment_id=None):
message = f'Activity not found (id: {activity_id})'
code = 404
response_object = {'status': status,
response_object = {
'status': status,
'message': message,
'data': ({'chart_data': content}
'data': (
{'chart_data': content}
if data_type == 'chart'
else {'gpx': content})}
else {'gpx': content}
),
}
return jsonify(response_object), code
@ -520,7 +544,7 @@ def get_activity_chart_data(auth_user_id, activity_id):
@activities_blueprint.route(
'/activities/<int:activity_id>/gpx/segment/<int:segment_id>',
methods=['GET']
methods=['GET'],
)
@authenticate
def get_segment_gpx(auth_user_id, activity_id, segment_id):
@ -570,7 +594,7 @@ def get_segment_gpx(auth_user_id, activity_id, segment_id):
@activities_blueprint.route(
'/activities/<int:activity_id>/chart_data/segment/<int:segment_id>',
methods=['GET']
methods=['GET'],
)
@authenticate
def get_segment_chart_data(auth_user_id, activity_id, segment_id):
@ -673,7 +697,7 @@ def get_map(map_id):
if not activity:
response_object = {
'status': 'fail',
'message': 'Map does not exist'
'message': 'Map does not exist',
}
return jsonify(response_object), 404
else:
@ -681,10 +705,7 @@ def get_map(map_id):
return send_file(absolute_map_filepath)
except Exception as e:
appLog.error(e)
response_object = {
'status': 'error',
'message': 'internal error.'
}
response_object = {'status': 'error', 'message': 'internal error.'}
return jsonify(response_object), 500
@ -808,17 +829,13 @@ def post_activity(auth_user_id):
activity_data = json.loads(request.form["data"])
if not activity_data or activity_data.get('sport_id') is None:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
activity_file = request.files['file']
upload_dir = os.path.join(
current_app.config['UPLOAD_FOLDER'],
'activities',
str(auth_user_id))
current_app.config['UPLOAD_FOLDER'], 'activities', str(auth_user_id)
)
folders = {
'extract_dir': os.path.join(upload_dir, 'extract'),
'tmp_dir': os.path.join(upload_dir, 'tmp'),
@ -832,27 +849,21 @@ def post_activity(auth_user_id):
response_object = {
'status': 'created',
'data': {
'activities': [new_activity.serialize()
for new_activity in new_activities]
}
'activities': [
new_activity.serialize()
for new_activity in new_activities
]
},
}
code = 201
else:
response_object = {
'status': 'fail',
'data': {
'activities': []
}
}
response_object = {'status': 'fail', 'data': {'activities': []}}
code = 400
except ActivityException as e:
db.session.rollback()
if e.e:
appLog.error(e.e)
response_object = {
'status': e.status,
'message': e.message,
}
response_object = {'status': e.status, 'message': e.message}
code = 500 if e.status == 'error' else 400
shutil.rmtree(folders['extract_dir'], ignore_errors=True)
@ -975,14 +986,14 @@ def post_activity_no_gpx(auth_user_id):
"""
activity_data = request.get_json()
if not activity_data or activity_data.get('sport_id') is None \
or activity_data.get('duration') is None \
or activity_data.get('distance') is None \
or activity_data.get('activity_date') is None:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
if (
not activity_data
or activity_data.get('sport_id') is None
or activity_data.get('duration') is None
or activity_data.get('distance') is None
or activity_data.get('activity_date') is None
):
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
try:
@ -993,9 +1004,7 @@ def post_activity_no_gpx(auth_user_id):
response_object = {
'status': 'created',
'data': {
'activities': [new_activity.serialize()]
}
'data': {'activities': [new_activity.serialize()]},
}
return jsonify(response_object), 201
@ -1004,7 +1013,7 @@ def post_activity_no_gpx(auth_user_id):
appLog.error(e)
response_object = {
'status': 'fail',
'message': 'Error during activity save.'
'message': 'Error during activity save.',
}
return jsonify(response_object), 500
@ -1130,17 +1139,15 @@ def update_activity(auth_user_id, activity_id):
"""
activity_data = request.get_json()
if not activity_data:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
try:
activity = Activity.query.filter_by(id=activity_id).first()
if activity:
response_object, code = can_view_activity(
auth_user_id, activity.user_id)
auth_user_id, activity.user_id
)
if response_object:
return jsonify(response_object), code
@ -1148,17 +1155,13 @@ def update_activity(auth_user_id, activity_id):
db.session.commit()
response_object = {
'status': 'success',
'data': {
'activities': [activity.serialize()]
}
'data': {'activities': [activity.serialize()]},
}
code = 200
else:
response_object = {
'status': 'not found',
'data': {
'activities': []
}
'data': {'activities': []},
}
code = 404
except (exc.IntegrityError, exc.OperationalError, ValueError) as e:
@ -1166,7 +1169,7 @@ def update_activity(auth_user_id, activity_id):
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code
@ -1213,31 +1216,32 @@ def delete_activity(auth_user_id, activity_id):
activity = Activity.query.filter_by(id=activity_id).first()
if activity:
response_object, code = can_view_activity(
auth_user_id, activity.user_id)
auth_user_id, activity.user_id
)
if response_object:
return jsonify(response_object), code
db.session.delete(activity)
db.session.commit()
response_object = {
'status': 'no content'
}
response_object = {'status': 'no content'}
code = 204
else:
response_object = {
'status': 'not found',
'data': {
'activities': []
}
'data': {'activities': []},
}
code = 404
except (exc.IntegrityError, exc.OperationalError, ValueError, OSError) \
as e:
except (
exc.IntegrityError,
exc.OperationalError,
ValueError,
OSError,
) as e:
db.session.rollback()
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code

View File

@ -21,42 +21,38 @@ record_types = [
def update_records(user_id, sport_id, connection, session):
record_table = Record.__table__
new_records = Activity.get_user_activity_records(
user_id,
sport_id)
new_records = Activity.get_user_activity_records(user_id, sport_id)
for record_type, record_data in new_records.items():
if record_data['record_value']:
record = Record.query.filter_by(
user_id=user_id,
sport_id=sport_id,
record_type=record_type,
user_id=user_id, sport_id=sport_id, record_type=record_type
).first()
if record:
value = convert_value_to_integer(
record_type, record_data['record_value']
)
connection.execute(record_table.update().where(
record_table.c.id == record.id,
).values(
connection.execute(
record_table.update()
.where(record_table.c.id == record.id)
.values(
value=value,
activity_id=record_data['activity'].id,
activity_date=record_data['activity'].activity_date,
))
)
)
else:
new_record = Record(
activity=record_data['activity'],
record_type=record_type
activity=record_data['activity'], record_type=record_type
)
new_record.value = record_data['record_value']
session.add(new_record)
else:
connection.execute(record_table.delete().where(
record_table.c.user_id == user_id,
).where(
record_table.c.sport_id == sport_id,
).where(
record_table.c.record_type == record_type,
))
connection.execute(
record_table.delete()
.where(record_table.c.user_id == user_id)
.where(record_table.c.sport_id == sport_id)
.where(record_table.c.record_type == record_type)
)
class Sport(db.Model):
@ -65,12 +61,12 @@ class Sport(db.Model):
label = db.Column(db.String(50), unique=True, nullable=False)
img = db.Column(db.String(255), unique=True, nullable=True)
is_default = db.Column(db.Boolean, default=False, nullable=False)
activities = db.relationship('Activity',
lazy=True,
backref=db.backref('sports', lazy='joined'))
records = db.relationship('Record',
lazy=True,
backref=db.backref('sports', lazy='joined'))
activities = db.relationship(
'Activity', lazy=True, backref=db.backref('sports', lazy='joined')
)
records = db.relationship(
'Record', lazy=True, backref=db.backref('sports', lazy='joined')
)
def __repr__(self):
return f'<Sport {self.label!r}>'
@ -83,31 +79,24 @@ class Sport(db.Model):
'id': self.id,
'label': self.label,
'img': self.img,
'_can_be_deleted':
len(self.activities) == 0 and not self.is_default
'_can_be_deleted': len(self.activities) == 0
and not self.is_default,
}
class Activity(db.Model):
__tablename__ = "activities"
id = db.Column(
db.Integer,
primary_key=True,
autoincrement=True)
user_id = db.Column(
db.Integer,
db.ForeignKey('users.id'),
nullable=False)
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
sport_id = db.Column(
db.Integer,
db.ForeignKey('sports.id'),
nullable=False)
db.Integer, db.ForeignKey('sports.id'), nullable=False
)
title = db.Column(db.String(255), nullable=True)
gpx = db.Column(db.String(255), nullable=True)
creation_date = db.Column(
db.DateTime, default=datetime.datetime.utcnow)
creation_date = db.Column(db.DateTime, default=datetime.datetime.utcnow)
modification_date = db.Column(
db.DateTime, onupdate=datetime.datetime.utcnow)
db.DateTime, onupdate=datetime.datetime.utcnow
)
activity_date = db.Column(db.DateTime, nullable=False)
duration = db.Column(db.Interval, nullable=False)
pauses = db.Column(db.Interval, nullable=True)
@ -125,20 +114,18 @@ class Activity(db.Model):
weather_start = db.Column(JSON, nullable=True)
weather_end = db.Column(JSON, nullable=True)
notes = db.Column(db.String(500), nullable=True)
segments = db.relationship('ActivitySegment',
segments = db.relationship(
'ActivitySegment',
lazy=True,
cascade='all, delete',
backref=db.backref(
'activities',
lazy='joined',
single_parent=True))
records = db.relationship('Record',
backref=db.backref('activities', lazy='joined', single_parent=True),
)
records = db.relationship(
'Record',
lazy=True,
cascade='all, delete',
backref=db.backref(
'activities',
lazy='joined',
single_parent=True))
backref=db.backref('activities', lazy='joined', single_parent=True),
)
def __str__(self):
return f'<Activity \'{self.sports.label}\' - {self.activity_date}>'
@ -162,58 +149,80 @@ class Activity(db.Model):
max_speed_from = params.get('max_speed_from') if params else None
max_speed_to = params.get('max_speed_to') if params else None
sport_id = params.get('sport_id') if params else None
previous_activity = Activity.query.filter(
previous_activity = (
Activity.query.filter(
Activity.id != self.id,
Activity.user_id == self.user_id,
Activity.sport_id == sport_id if sport_id else True,
Activity.activity_date <= self.activity_date,
Activity.activity_date >= datetime.datetime.strptime(
date_from, '%Y-%m-%d'
) if date_from else True,
Activity.activity_date <= datetime.datetime.strptime(
date_to, '%Y-%m-%d'
) if date_to else True,
Activity.distance >= int(distance_from) if distance_from else True,
Activity.activity_date
>= datetime.datetime.strptime(date_from, '%Y-%m-%d')
if date_from
else True,
Activity.activity_date
<= datetime.datetime.strptime(date_to, '%Y-%m-%d')
if date_to
else True,
Activity.distance >= int(distance_from)
if distance_from
else True,
Activity.distance <= int(distance_to) if distance_to else True,
Activity.duration >= convert_in_duration(duration_from)
if duration_from else True,
if duration_from
else True,
Activity.duration <= convert_in_duration(duration_to)
if duration_to else True,
if duration_to
else True,
Activity.ave_speed >= float(ave_speed_from)
if ave_speed_from else True,
if ave_speed_from
else True,
Activity.ave_speed <= float(ave_speed_to)
if ave_speed_to else True,
if ave_speed_to
else True,
Activity.max_speed >= float(max_speed_from)
if max_speed_from else True,
if max_speed_from
else True,
Activity.max_speed <= float(max_speed_to)
if max_speed_to else True,
).order_by(
Activity.activity_date.desc()
).first()
next_activity = Activity.query.filter(
if max_speed_to
else True,
)
.order_by(Activity.activity_date.desc())
.first()
)
next_activity = (
Activity.query.filter(
Activity.id != self.id,
Activity.user_id == self.user_id,
Activity.sport_id == sport_id if sport_id else True,
Activity.activity_date >= self.activity_date,
Activity.activity_date >= datetime.datetime.strptime(
date_from, '%Y-%m-%d'
) if date_from else True,
Activity.activity_date <= datetime.datetime.strptime(
date_to, '%Y-%m-%d'
) if date_to else True,
Activity.distance >= int(distance_from) if distance_from else True,
Activity.activity_date
>= datetime.datetime.strptime(date_from, '%Y-%m-%d')
if date_from
else True,
Activity.activity_date
<= datetime.datetime.strptime(date_to, '%Y-%m-%d')
if date_to
else True,
Activity.distance >= int(distance_from)
if distance_from
else True,
Activity.distance <= int(distance_to) if distance_to else True,
Activity.duration >= convert_in_duration(duration_from)
if duration_from else True,
if duration_from
else True,
Activity.duration <= convert_in_duration(duration_to)
if duration_to else True,
if duration_to
else True,
Activity.ave_speed >= float(ave_speed_from)
if ave_speed_from else True,
if ave_speed_from
else True,
Activity.ave_speed <= float(ave_speed_to)
if ave_speed_to else True,
).order_by(
Activity.activity_date.asc()
).first()
if ave_speed_to
else True,
)
.order_by(Activity.activity_date.asc())
.first()
)
return {
"id": self.id,
"user_id": self.user_id,
@ -233,15 +242,19 @@ class Activity(db.Model):
"max_speed": float(self.max_speed) if self.max_speed else None,
"ave_speed": float(self.ave_speed) if self.ave_speed else None,
"with_gpx": self.gpx is not None,
"bounds": [float(bound) for bound in self.bounds] if self.bounds else [], # noqa
"previous_activity": previous_activity.id if previous_activity else None, # noqa
"bounds": [float(bound) for bound in self.bounds]
if self.bounds
else [], # noqa
"previous_activity": previous_activity.id
if previous_activity
else None, # noqa
"next_activity": next_activity.id if next_activity else None,
"segments": [segment.serialize() for segment in self.segments],
"records": [record.serialize() for record in self.records],
"map": self.map_id if self.map else None,
"weather_start": self.weather_start,
"weather_end": self.weather_end,
"notes": self.notes
"notes": self.notes,
}
@classmethod
@ -255,37 +268,41 @@ class Activity(db.Model):
records = {}
for record_type, column in record_types_columns.items():
column_sorted = getattr(getattr(Activity, column), 'desc')()
record_activity = Activity.query.filter_by(
user_id=user_id,
sport_id=sport_id,
).order_by(
column_sorted,
Activity.activity_date,
).first()
record_activity = (
Activity.query.filter_by(user_id=user_id, sport_id=sport_id)
.order_by(column_sorted, Activity.activity_date)
.first()
)
records[record_type] = dict(
record_value=(getattr(record_activity, column)
if record_activity else None),
activity=record_activity)
record_value=(
getattr(record_activity, column)
if record_activity
else None
),
activity=record_activity,
)
return records
@listens_for(Activity, 'after_insert')
def on_activity_insert(mapper, connection, activity):
@listens_for(db.Session, 'after_flush', once=True)
def receive_after_flush(session, context):
update_records(activity.user_id, activity.sport_id, connection, session) # noqa
update_records(
activity.user_id, activity.sport_id, connection, session
) # noqa
@listens_for(Activity, 'after_update')
def on_activity_update(mapper, connection, activity):
if object_session(activity).is_modified(activity, include_collections=True): # noqa
if object_session(activity).is_modified(
activity, include_collections=True
): # noqa
@listens_for(db.Session, 'after_flush', once=True)
def receive_after_flush(session, context):
sports_list = [activity.sport_id]
records = Record.query.filter_by(
activity_id=activity.id,
).all()
records = Record.query.filter_by(activity_id=activity.id).all()
for rec in records:
if rec.sport_id not in sports_list:
sports_list.append(rec.sport_id)
@ -295,7 +312,6 @@ def on_activity_update(mapper, connection, activity):
@listens_for(Activity, 'after_delete')
def on_activity_delete(mapper, connection, old_record):
@listens_for(db.Session, 'after_flush', once=True)
def receive_after_flush(session, context):
if old_record.map:
@ -307,12 +323,9 @@ def on_activity_delete(mapper, connection, old_record):
class ActivitySegment(db.Model):
__tablename__ = "activity_segments"
activity_id = db.Column(
db.Integer,
db.ForeignKey('activities.id'),
primary_key=True)
segment_id = db.Column(
db.Integer,
primary_key=True)
db.Integer, db.ForeignKey('activities.id'), primary_key=True
)
segment_id = db.Column(db.Integer, primary_key=True)
duration = db.Column(db.Interval, nullable=False)
pauses = db.Column(db.Interval, nullable=True)
moving = db.Column(db.Interval, nullable=True)
@ -325,8 +338,10 @@ class ActivitySegment(db.Model):
ave_speed = db.Column(db.Numeric(6, 2), nullable=True) # km/h
def __str__(self):
return (f'<Segment \'{self.segment_id}\' '
f'for activity \'{self.activity_id}\'>')
return (
f'<Segment \'{self.segment_id}\' '
f'for activity \'{self.activity_id}\'>'
)
def __init__(self, segment_id, activity_id):
self.segment_id = segment_id
@ -345,38 +360,35 @@ class ActivitySegment(db.Model):
"descent": float(self.descent) if self.descent else None,
"ascent": float(self.ascent) if self.ascent else None,
"max_speed": float(self.max_speed) if self.max_speed else None,
"ave_speed": float(self.ave_speed) if self.ave_speed else None
"ave_speed": float(self.ave_speed) if self.ave_speed else None,
}
class Record(db.Model):
__tablename__ = "records"
__table_args__ = (db.UniqueConstraint(
'user_id', 'sport_id', 'record_type', name='user_sports_records'),)
id = db.Column(
db.Integer,
primary_key=True,
autoincrement=True)
user_id = db.Column(
db.Integer,
db.ForeignKey('users.id'),
nullable=False)
__table_args__ = (
db.UniqueConstraint(
'user_id', 'sport_id', 'record_type', name='user_sports_records'
),
)
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
sport_id = db.Column(
db.Integer,
db.ForeignKey('sports.id'),
nullable=False)
db.Integer, db.ForeignKey('sports.id'), nullable=False
)
activity_id = db.Column(
db.Integer,
db.ForeignKey('activities.id'),
nullable=False)
db.Integer, db.ForeignKey('activities.id'), nullable=False
)
record_type = db.Column(Enum(*record_types, name="record_types"))
activity_date = db.Column(db.DateTime, nullable=False)
_value = db.Column("value", db.Integer, nullable=True)
def __str__(self):
return (f'<Record {self.sports.label} - '
return (
f'<Record {self.sports.label} - '
f'{self.record_type} - '
f"{self.activity_date.strftime('%Y-%m-%d')}>")
f"{self.activity_date.strftime('%Y-%m-%d')}>"
)
def __init__(self, activity, record_type):
self.user_id = activity.user_id
@ -421,19 +433,19 @@ class Record(db.Model):
@listens_for(Record, 'after_delete')
def on_record_delete(mapper, connection, old_record):
@listens_for(db.Session, 'after_flush', once=True)
def receive_after_flush(session, context):
activity = old_record.activities
new_records = Activity.get_user_activity_records(
activity.user_id,
activity.sport_id)
activity.user_id, activity.sport_id
)
for record_type, record_data in new_records.items():
if record_data['record_value'] \
and record_type == old_record.record_type:
if (
record_data['record_value']
and record_type == old_record.record_type
):
new_record = Record(
activity=record_data['activity'],
record_type=record_type
activity=record_data['activity'], record_type=record_type
)
new_record.value = record_data['record_value']
session.add(new_record)

View File

@ -104,15 +104,13 @@ def get_records(auth_user_id):
"""
records = Record.query.filter_by(user_id=auth_user_id)\
.order_by(
Record.sport_id.asc(),
Record.record_type.asc(),
).all()
records = (
Record.query.filter_by(user_id=auth_user_id)
.order_by(Record.sport_id.asc(), Record.record_type.asc())
.all()
)
response_object = {
'status': 'success',
'data': {
'records': [record.serialize() for record in records]
}
'data': {'records': [record.serialize() for record in records]},
}
return jsonify(response_object), 200

View File

@ -87,9 +87,7 @@ def get_sports(auth_user_id):
sports = Sport.query.order_by(Sport.id).all()
response_object = {
'status': 'success',
'data': {
'sports': [sport.serialize() for sport in sports]
}
'data': {'sports': [sport.serialize() for sport in sports]},
}
return jsonify(response_object), 200
@ -161,34 +159,25 @@ def get_sport(auth_user_id, sport_id):
if sport:
response_object = {
'status': 'success',
'data': {
'sports': [sport.serialize()]
}
'data': {'sports': [sport.serialize()]},
}
code = 200
else:
response_object = {
'status': 'not found',
'data': {
'sports': []
}
}
response_object = {'status': 'not found', 'data': {'sports': []}}
code = 404
return jsonify(response_object), code
# no administration - no documentation for now
@sports_blueprint.route('/sports', methods=['POST'])
@authenticate_as_admin
def post_sport(auth_user_id):
"""Post a sport"""
sport_data = request.get_json()
if not sport_data or sport_data.get('label') is None:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
try:
@ -197,9 +186,7 @@ def post_sport(auth_user_id):
db.session.commit()
response_object = {
'status': 'created',
'data': {
'sports': [new_sport.serialize()]
}
'data': {'sports': [new_sport.serialize()]},
}
code = 201
except (exc.IntegrityError, exc.OperationalError, ValueError) as e:
@ -207,7 +194,7 @@ def post_sport(auth_user_id):
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code
@ -219,10 +206,7 @@ def update_sport(auth_user_id, sport_id):
"""Update a sport"""
sport_data = request.get_json()
if not sport_data or sport_data.get('label') is None:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
sports_list = []
@ -231,23 +215,16 @@ def update_sport(auth_user_id, sport_id):
if sport:
sport.label = sport_data.get('label')
db.session.commit()
sports_list.append({
'id': sport.id,
'label': sport.label
})
sports_list.append({'id': sport.id, 'label': sport.label})
response_object = {
'status': 'success',
'data': {
'sports': sports_list
}
'data': {'sports': sports_list},
}
code = 200
else:
response_object = {
'status': 'not found',
'data': {
'sports': sports_list
}
'data': {'sports': sports_list},
}
code = 404
except (exc.IntegrityError, exc.OperationalError, ValueError) as e:
@ -255,7 +232,7 @@ def update_sport(auth_user_id, sport_id):
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code
@ -270,24 +247,17 @@ def delete_sport(auth_user_id, sport_id):
if sport:
db.session.delete(sport)
db.session.commit()
response_object = {
'status': 'no content'
}
response_object = {'status': 'no content'}
code = 204
else:
response_object = {
'status': 'not found',
'data': {
'sports': []
}
}
response_object = {'status': 'not found', 'data': {'sports': []}}
code = 404
except exc.IntegrityError as e:
db.session.rollback()
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Associated activities exist.'
'message': 'Error. Associated activities exist.',
}
code = 500
except (exc.OperationalError, ValueError) as e:
@ -295,7 +265,7 @@ def delete_sport(auth_user_id, sport_id):
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code

View File

@ -18,7 +18,7 @@ def get_activities(user_id, filter_type):
if not user:
response_object = {
'status': 'not found',
'message': 'User does not exist.'
'message': 'User does not exist.',
}
return jsonify(response_object), 404
@ -29,8 +29,9 @@ def get_activities(user_id, filter_type):
_, date_from = get_datetime_with_tz(user.timezone, date_from)
date_to = params.get('to')
if date_to:
date_to = datetime.strptime(f'{date_to} 23:59:59',
'%Y-%m-%d %H:%M:%S')
date_to = datetime.strptime(
f'{date_to} 23:59:59', '%Y-%m-%d %H:%M:%S'
)
_, date_to = get_datetime_with_tz(user.timezone, date_to)
sport_id = params.get('sport_id')
time = params.get('time')
@ -42,19 +43,22 @@ def get_activities(user_id, filter_type):
if not sport:
response_object = {
'status': 'not found',
'message': 'Sport does not exist.'
'message': 'Sport does not exist.',
}
return jsonify(response_object), 404
activities = Activity.query.filter(
activities = (
Activity.query.filter(
Activity.user_id == user_id,
Activity.activity_date >= date_from if date_from else True,
Activity.activity_date < date_to + timedelta(seconds=1)
if date_to else True,
if date_to
else True,
Activity.sport_id == sport_id if sport_id else True,
).order_by(
Activity.activity_date.asc()
).all()
)
.order_by(Activity.activity_date.asc())
.all()
)
activities_list = {}
for activity in activities:
@ -63,21 +67,25 @@ def get_activities(user_id, filter_type):
if sport_id not in activities_list:
activities_list[sport_id] = {
'nb_activities': 0,
'total_distance': 0.,
'total_distance': 0.0,
'total_duration': 0,
}
activities_list[sport_id]['nb_activities'] += 1
activities_list[sport_id]['total_distance'] += \
float(activity.distance)
activities_list[sport_id]['total_duration'] += \
convert_timedelta_to_integer(activity.moving)
activities_list[sport_id]['total_distance'] += float(
activity.distance
)
activities_list[sport_id][
'total_duration'
] += convert_timedelta_to_integer(activity.moving)
else:
if time == 'week':
activity_date = activity.activity_date - timedelta(
days=(activity.activity_date.isoweekday()
days=(
activity.activity_date.isoweekday()
if activity.activity_date.isoweekday() < 7
else 0)
else 0
)
)
time_period = datetime.strftime(activity_date, "%Y-%m-%d")
elif time == 'weekm': # week start Monday
@ -86,13 +94,17 @@ def get_activities(user_id, filter_type):
)
time_period = datetime.strftime(activity_date, "%Y-%m-%d")
elif time == 'month':
time_period = datetime.strftime(activity.activity_date, "%Y-%m") # noqa
time_period = datetime.strftime(
activity.activity_date, "%Y-%m"
) # noqa
elif time == 'year' or not time:
time_period = datetime.strftime(activity.activity_date, "%Y") # noqa
time_period = datetime.strftime(
activity.activity_date, "%Y"
) # noqa
else:
response_object = {
'status': 'fail',
'message': 'Invalid time period.'
'message': 'Invalid time period.',
}
return jsonify(response_object), 400
sport_id = activity.sport_id
@ -101,27 +113,27 @@ def get_activities(user_id, filter_type):
if sport_id not in activities_list[time_period]:
activities_list[time_period][sport_id] = {
'nb_activities': 0,
'total_distance': 0.,
'total_distance': 0.0,
'total_duration': 0,
}
activities_list[time_period][sport_id]['nb_activities'] += 1
activities_list[time_period][sport_id]['total_distance'] += \
float(activity.distance)
activities_list[time_period][sport_id]['total_duration'] += \
convert_timedelta_to_integer(activity.moving)
activities_list[time_period][sport_id][
'total_distance'
] += float(activity.distance)
activities_list[time_period][sport_id][
'total_duration'
] += convert_timedelta_to_integer(activity.moving)
response_object = {
'status': 'success',
'data': {
'statistics': activities_list
}
'data': {'statistics': activities_list},
}
code = 200
except Exception as e:
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
code = 500
return jsonify(response_object), code

View File

@ -35,7 +35,8 @@ def get_datetime_with_tz(timezone, activity_date, gpx_data=None):
fmt = '%Y-%m-%d %H:%M:%S'
activity_date_string = activity_date.strftime(fmt)
activity_date_tmp = utc_tz.localize(
datetime.strptime(activity_date_string, fmt))
datetime.strptime(activity_date_string, fmt)
)
activity_date_tz = activity_date_tmp.astimezone(user_tz)
else:
activity_date_tz = user_tz.localize(activity_date)
@ -59,27 +60,34 @@ def update_activity_data(activity, gpx_data):
return activity
def create_activity(
user, activity_data, gpx_data=None
):
activity_date = gpx_data['start'] if gpx_data else datetime.strptime(
activity_data.get('activity_date'), '%Y-%m-%d %H:%M')
def create_activity(user, activity_data, gpx_data=None):
activity_date = (
gpx_data['start']
if gpx_data
else datetime.strptime(
activity_data.get('activity_date'), '%Y-%m-%d %H:%M'
)
)
activity_date_tz, activity_date = get_datetime_with_tz(
user.timezone, activity_date, gpx_data)
user.timezone, activity_date, gpx_data
)
duration = gpx_data['duration'] if gpx_data \
duration = (
gpx_data['duration']
if gpx_data
else timedelta(seconds=activity_data.get('duration'))
distance = gpx_data['distance'] if gpx_data \
else activity_data.get('distance')
title = gpx_data['name'] if gpx_data \
else activity_data.get('title')
)
distance = (
gpx_data['distance'] if gpx_data else activity_data.get('distance')
)
title = gpx_data['name'] if gpx_data else activity_data.get('title')
new_activity = Activity(
user_id=user.id,
sport_id=activity_data.get('sport_id'),
activity_date=activity_date,
distance=distance,
duration=duration
duration=duration,
)
new_activity.notes = activity_data.get('notes')
@ -91,7 +99,8 @@ def create_activity(
activity_datetime = (
activity_date_tz.strftime(fmt)
if activity_date_tz
else new_activity.activity_date.strftime(fmt))
else new_activity.activity_date.strftime(fmt)
)
new_activity.title = f'{sport.label} - {activity_datetime}'
if gpx_data:
@ -100,18 +109,18 @@ def create_activity(
update_activity_data(new_activity, gpx_data)
else:
new_activity.moving = duration
new_activity.ave_speed = (None
new_activity.ave_speed = (
None
if duration.seconds == 0
else float(new_activity.distance) /
(duration.seconds / 3600))
else float(new_activity.distance) / (duration.seconds / 3600)
)
new_activity.max_speed = new_activity.ave_speed
return new_activity
def create_segment(activity_id, segment_data):
new_segment = ActivitySegment(
activity_id=activity_id,
segment_id=segment_data['idx']
activity_id=activity_id, segment_id=segment_data['idx']
)
new_segment.duration = segment_data['duration']
new_segment.distance = segment_data['distance']
@ -127,7 +136,8 @@ def update_activity(activity):
(case of a modified gpx file, see issue #7)
"""
gpx_data, _, _ = get_gpx_info(
get_absolute_file_path(activity.gpx), False, False)
get_absolute_file_path(activity.gpx), False, False
)
updated_activity = update_activity_data(activity, gpx_data)
updated_activity.duration = gpx_data['duration']
updated_activity.distance = gpx_data['distance']
@ -156,21 +166,26 @@ def edit_activity(activity, activity_data, auth_user_id):
if not activity.gpx:
if activity_data.get('activity_date'):
activity_date = datetime.strptime(
activity_data.get('activity_date'), '%Y-%m-%d %H:%M')
activity_data.get('activity_date'), '%Y-%m-%d %H:%M'
)
_, activity.activity_date = get_datetime_with_tz(
user.timezone, activity_date)
user.timezone, activity_date
)
if activity_data.get('duration'):
activity.duration = timedelta(
seconds=activity_data.get('duration'))
seconds=activity_data.get('duration')
)
activity.moving = activity.duration
if activity_data.get('distance'):
activity.distance = activity_data.get('distance')
activity.ave_speed = (None if activity.duration.seconds == 0
else float(activity.distance) /
(activity.duration.seconds / 3600))
activity.ave_speed = (
None
if activity.duration.seconds == 0
else float(activity.distance) / (activity.duration.seconds / 3600)
)
activity.max_speed = activity.ave_speed
return activity
@ -188,14 +203,12 @@ def get_new_file_path(
if not extension:
extension = f".{old_filename.rsplit('.', 1)[1].lower()}"
_, new_filename = tempfile.mkstemp(
prefix=f'{activity_date}_{sport}_',
suffix=extension
prefix=f'{activity_date}_{sport}_', suffix=extension
)
dir_path = os.path.join('activities', str(auth_user_id))
if not os.path.exists(dir_path):
os.makedirs(dir_path)
file_path = os.path.join(dir_path,
new_filename.split('/')[-1])
file_path = os.path.join(dir_path, new_filename.split('/')[-1])
return file_path
@ -228,7 +241,7 @@ def process_one_gpx_file(params, filename):
auth_user_id=auth_user_id,
activity_date=gpx_data['start'],
old_filename=filename,
sport=params['sport_label']
sport=params['sport_label'],
)
absolute_gpx_filepath = get_absolute_file_path(new_filepath)
os.rename(params['file_path'], absolute_gpx_filepath)
@ -238,7 +251,7 @@ def process_one_gpx_file(params, filename):
auth_user_id=auth_user_id,
activity_date=gpx_data['start'],
extension='.png',
sport=params['sport_label']
sport=params['sport_label'],
)
absolute_map_filepath = get_absolute_file_path(map_filepath)
generate_map(absolute_map_filepath, map_data)
@ -249,7 +262,8 @@ def process_one_gpx_file(params, filename):
try:
new_activity = create_activity(
params['user'], params['activity_data'], gpx_data)
params['user'], params['activity_data'], gpx_data
)
new_activity.map = map_filepath
new_activity.map_id = get_map_hash(map_filepath)
new_activity.weather_start = weather_data[0]
@ -280,8 +294,9 @@ def process_zip_archive(common_params, extract_dir):
gpx_files_ok = 0
for gpx_file in os.listdir(extract_dir):
if ('.' in gpx_file and gpx_file.rsplit('.', 1)[1].lower()
in current_app.config.get('ACTIVITY_ALLOWED_EXTENSIONS')):
if '.' in gpx_file and gpx_file.rsplit('.', 1)[
1
].lower() in current_app.config.get('ACTIVITY_ALLOWED_EXTENSIONS'):
gpx_files_ok += 1
if gpx_files_ok > gpx_files_limit:
break
@ -303,7 +318,7 @@ def process_files(auth_user_id, activity_data, activity_file, folders):
raise ActivityException(
'error',
f"Sport id: {activity_data.get('sport_id')} does not exist",
None
None,
)
user = User.query.filter_by(id=auth_user_id).first()

View File

@ -36,8 +36,9 @@ def get_gpx_data(parsed_gpx, max_speed, start, stopped_time_btwn_seg):
mv = parsed_gpx.get_moving_data()
gpx_data['moving_time'] = timedelta(seconds=mv.moving_time)
gpx_data['stop_time'] = (timedelta(seconds=mv.stopped_time)
+ stopped_time_btwn_seg)
gpx_data['stop_time'] = (
timedelta(seconds=mv.stopped_time) + stopped_time_btwn_seg
)
distance = mv.moving_distance + mv.stopped_distance
gpx_data['distance'] = distance / 1000
@ -52,10 +53,7 @@ def get_gpx_info(gpx_file, update_map_data=True, update_weather_data=True):
if gpx is None:
return None
gpx_data = {
'name': gpx.tracks[0].name,
'segments': []
}
gpx_data = {'name': gpx.tracks[0].name, 'segments': []}
max_speed = 0
start = 0
map_data = []
@ -90,12 +88,12 @@ def get_gpx_info(gpx_file, update_map_data=True, update_weather_data=True):
weather_data.append(get_weather(point))
if update_map_data:
map_data.append([
point.longitude, point.latitude
])
segment_max_speed = (segment.get_moving_data().max_speed
map_data.append([point.longitude, point.latitude])
segment_max_speed = (
segment.get_moving_data().max_speed
if segment.get_moving_data().max_speed
else 0)
else 0
)
if segment_max_speed > max_speed:
max_speed = segment_max_speed
@ -115,7 +113,7 @@ def get_gpx_info(gpx_file, update_map_data=True, update_weather_data=True):
bounds.min_latitude,
bounds.min_longitude,
bounds.max_latitude,
bounds.max_longitude
bounds.max_longitude,
]
return gpx_data, map_data, weather_data
@ -126,16 +124,10 @@ def get_gpx_segments(track_segments, segment_id=None):
segment_index = segment_id - 1
if segment_index > (len(track_segments) - 1):
raise ActivityGPXException(
'not found',
f'No segment with id \'{segment_id}\'',
None
'not found', f'No segment with id \'{segment_id}\'', None
)
if segment_index < 0:
raise ActivityGPXException(
'error',
'Incorrect segment id',
None
)
raise ActivityGPXException('error', 'Incorrect segment id', None)
segments = [track_segments[segment_index]]
else:
segments = track_segments
@ -160,28 +152,41 @@ def get_chart_data(gpx_file, segment_id=None):
for point_idx, point in enumerate(segment.points):
if segment_idx == 0 and point_idx == 0:
first_point = point
distance = (point.distance_3d(previous_point)
if (point.elevation
distance = (
point.distance_3d(previous_point)
if (
point.elevation
and previous_point
and previous_point.elevation)
and previous_point.elevation
)
else point.distance_2d(previous_point)
)
distance = 0 if distance is None else distance
distance += previous_distance
speed = (round((segment.get_speed(point_idx) / 1000)*3600, 2)
speed = (
round((segment.get_speed(point_idx) / 1000) * 3600, 2)
if segment.get_speed(point_idx) is not None
else 0)
chart_data.append({
'distance': (round(distance / 1000, 2)
if distance is not None else 0),
else 0
)
chart_data.append(
{
'distance': (
round(distance / 1000, 2)
if distance is not None
else 0
),
'duration': point.time_difference(first_point),
'elevation': (round(point.elevation, 1)
if point.elevation is not None else 0),
'elevation': (
round(point.elevation, 1)
if point.elevation is not None
else 0
),
'latitude': point.latitude,
'longitude': point.longitude,
'speed': speed,
'time': point.time,
})
}
)
previous_point = point
previous_distance = distance
@ -194,8 +199,7 @@ def extract_segment_from_gpx_file(content, segment_id):
return None
track_segment = get_gpx_segments(
gpx_content.tracks[0].segments,
segment_id
gpx_content.tracks[0].segments, segment_id
)
gpx = gpxpy.gpx.GPX()
@ -207,8 +211,8 @@ def extract_segment_from_gpx_file(content, segment_id):
for point_idx, point in enumerate(track_segment[0].points):
gpx_segment.points.append(
gpxpy.gpx.GPXTrackPoint(
point.latitude,
point.longitude,
elevation=point.elevation))
point.latitude, point.longitude, elevation=point.elevation
)
)
return gpx.to_xml()

View File

@ -17,7 +17,7 @@ def get_weather(point):
point.latitude,
point.longitude,
time=point_time,
units='si'
units='si',
)
weather = forecast.currently()
return {

View File

@ -5,15 +5,14 @@ from flask import current_app
class BaseConfig:
"""Base configuration"""
DEBUG = False
TESTING = False
SQLALCHEMY_TRACK_MODIFICATIONS = False
BCRYPT_LOG_ROUNDS = 13
TOKEN_EXPIRATION_DAYS = 30
TOKEN_EXPIRATION_SECONDS = 0
UPLOAD_FOLDER = os.path.join(
current_app.root_path, 'uploads'
)
UPLOAD_FOLDER = os.path.join(current_app.root_path, 'uploads')
PICTURE_ALLOWED_EXTENSIONS = {'jpg', 'png', 'gif'}
ACTIVITY_ALLOWED_EXTENSIONS = {'gpx', 'zip'}
REGISTRATION_ALLOWED = (
@ -23,6 +22,7 @@ class BaseConfig:
class DevelopmentConfig(BaseConfig):
"""Development configuration"""
DEBUG = True
SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL')
SECRET_KEY = 'development key'
@ -33,6 +33,7 @@ class DevelopmentConfig(BaseConfig):
class TestingConfig(BaseConfig):
"""Testing configuration"""
DEBUG = True
TESTING = True
SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_TEST_URL')

View File

@ -50,9 +50,7 @@ def user_1():
@pytest.fixture()
def user_1_admin():
admin = User(
username='admin',
email='admin@example.com',
password='12345678'
username='admin', email='admin@example.com', password='12345678'
)
admin.admin = True
db.session.add(admin)
@ -114,7 +112,7 @@ def activity_cycling_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('01/01/2018', '%d/%m/%Y'),
distance=10,
duration=datetime.timedelta(seconds=1024)
duration=datetime.timedelta(seconds=1024),
)
activity.max_speed = 10
activity.ave_speed = 10
@ -126,10 +124,7 @@ def activity_cycling_user_1():
@pytest.fixture()
def activity_cycling_user_1_segment():
activity_segment = ActivitySegment(
activity_id=1,
segment_id=0
)
activity_segment = ActivitySegment(activity_id=1, segment_id=0)
activity_segment.duration = datetime.timedelta(seconds=6000)
activity_segment.moving = activity_segment.duration
activity_segment.distance = 5
@ -145,7 +140,7 @@ def activity_running_user_1():
sport_id=2,
activity_date=datetime.datetime.strptime('01/04/2018', '%d/%m/%Y'),
distance=12,
duration=datetime.timedelta(seconds=6000)
duration=datetime.timedelta(seconds=6000),
)
activity.moving = activity.duration
db.session.add(activity)
@ -160,7 +155,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('20/03/2017', '%d/%m/%Y'),
distance=5,
duration=datetime.timedelta(seconds=1024)
duration=datetime.timedelta(seconds=1024),
)
activity.ave_speed = float(activity.distance) / (1024 / 3600)
activity.moving = activity.duration
@ -171,7 +166,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('01/06/2017', '%d/%m/%Y'),
distance=10,
duration=datetime.timedelta(seconds=3456)
duration=datetime.timedelta(seconds=3456),
)
activity.ave_speed = float(activity.distance) / (3456 / 3600)
activity.moving = activity.duration
@ -182,7 +177,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('01/01/2018', '%d/%m/%Y'),
distance=10,
duration=datetime.timedelta(seconds=1024)
duration=datetime.timedelta(seconds=1024),
)
activity.ave_speed = float(activity.distance) / (1024 / 3600)
activity.moving = activity.duration
@ -193,7 +188,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('23/02/2018', '%d/%m/%Y'),
distance=1,
duration=datetime.timedelta(seconds=600)
duration=datetime.timedelta(seconds=600),
)
activity.ave_speed = float(activity.distance) / (600 / 3600)
activity.moving = activity.duration
@ -204,7 +199,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('23/02/2018', '%d/%m/%Y'),
distance=10,
duration=datetime.timedelta(seconds=1000)
duration=datetime.timedelta(seconds=1000),
)
activity.ave_speed = float(activity.distance) / (1000 / 3600)
activity.moving = activity.duration
@ -215,7 +210,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('01/04/2018', '%d/%m/%Y'),
distance=8,
duration=datetime.timedelta(seconds=6000)
duration=datetime.timedelta(seconds=6000),
)
activity.ave_speed = float(activity.distance) / (6000 / 3600)
activity.moving = activity.duration
@ -226,7 +221,7 @@ def seven_activities_user_1():
sport_id=1,
activity_date=datetime.datetime.strptime('09/05/2018', '%d/%m/%Y'),
distance=10,
duration=datetime.timedelta(seconds=3000)
duration=datetime.timedelta(seconds=3000),
)
activity.ave_speed = float(activity.distance) / (3000 / 3600)
activity.moving = activity.duration
@ -242,7 +237,7 @@ def activity_cycling_user_2():
sport_id=1,
activity_date=datetime.datetime.strptime('23/01/2018', '%d/%m/%Y'),
distance=15,
duration=datetime.timedelta(seconds=3600)
duration=datetime.timedelta(seconds=3600),
)
activity.moving = activity.duration
db.session.add(activity)

View File

@ -6,7 +6,10 @@ from fittrackee_api.activities.models import Activity
def assert_activity_data_with_gpx(data):
assert 'creation_date' in data['data']['activities'][0]
assert 'Tue, 13 Mar 2018 12:44:45 GMT' == data['data']['activities'][0]['activity_date'] # noqa
assert (
'Tue, 13 Mar 2018 12:44:45 GMT'
== data['data']['activities'][0]['activity_date']
) # noqa
assert 1 == data['data']['activities'][0]['user_id']
assert '0:04:10' == data['data']['activities'][0]['duration']
assert data['data']['activities'][0]['ascent'] == 0.4
@ -50,37 +53,29 @@ def test_edit_an_activity_with_gpx(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=2,
title="Activity test",
)),
data=json.dumps(dict(sport_id=2, title="Activity test")),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -94,14 +89,11 @@ def test_edit_an_activity_with_gpx(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
notes="test notes"
)),
data=json.dumps(dict(notes="test notes")),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -118,45 +110,34 @@ def test_edit_an_activity_with_gpx_different_user(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='toto@toto.com',
password='87654321'
)),
content_type='application/json'
data=json.dumps(dict(email='toto@toto.com', password='87654321')),
content_type='application/json',
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=2,
title="Activity test",
)),
data=json.dumps(dict(sport_id=2, title="Activity test")),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -171,36 +152,29 @@ def test_edit_an_activity_with_gpx_partial(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=2,
)),
data=json.dumps(dict(sport_id=2)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -218,34 +192,29 @@ def test_edit_an_activity_with_gpx_invalid_payload(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -261,43 +230,39 @@ def test_edit_an_activity_with_gpx_incorrect_data(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=2,
)),
data=json.dumps(dict(sport_id=2)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert response.status_code == 500
assert 'error' in data['status']
assert 'Error. Please try again or contact the administrator.' in data['message'] # noqa
assert (
'Error. Please try again or contact the administrator.'
in data['message']
) # noqa
def test_edit_an_activity_wo_gpx(
@ -306,28 +271,26 @@ def test_edit_an_activity_wo_gpx(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/activities/no_gpx',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=1,
duration=3600,
activity_date='2018-05-14 14:05',
distance=7,
title='Activity test'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
title='Activity test',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -336,7 +299,10 @@ def test_edit_an_activity_wo_gpx(
assert 'created' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Mon, 14 May 2018 14:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Mon, 14 May 2018 14:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 1
assert data['data']['activities'][0]['duration'] == '1:00:00'
@ -382,18 +348,19 @@ def test_edit_an_activity_wo_gpx(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=2,
duration=3600,
activity_date='2018-05-15 15:05',
distance=8,
title='Activity test'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
title='Activity test',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -401,7 +368,10 @@ def test_edit_an_activity_wo_gpx(
assert 'success' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Tue, 15 May 2018 15:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Tue, 15 May 2018 15:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 2
assert data['data']['activities'][0]['duration'] == '1:00:00'
@ -447,14 +417,11 @@ def test_edit_an_activity_wo_gpx(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
notes='test notes'
)),
data=json.dumps(dict(notes='test notes')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -462,7 +429,10 @@ def test_edit_an_activity_wo_gpx(
assert 'success' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Tue, 15 May 2018 15:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Tue, 15 May 2018 15:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 2
assert data['data']['activities'][0]['duration'] == '1:00:00'
@ -512,52 +482,48 @@ def test_edit_an_activity_wo_gpx_different_user(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities/no_gpx',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=1,
duration=3600,
activity_date='2018-05-14 14:05',
distance=7,
title='Activity test'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
title='Activity test',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='toto@toto.com',
password='87654321'
)),
content_type='application/json'
data=json.dumps(dict(email='toto@toto.com', password='87654321')),
content_type='application/json',
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=2,
duration=3600,
activity_date='2018-05-15 15:05',
distance=8,
title='Activity test'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
title='Activity test',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -572,27 +538,25 @@ def test_edit_an_activity_wo_gpx_timezone(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/activities/no_gpx',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=1,
duration=3600,
activity_date='2018-05-14 14:05',
distance=7
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
distance=7,
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -601,11 +565,17 @@ def test_edit_an_activity_wo_gpx_timezone(
assert 'created' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Mon, 14 May 2018 12:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Mon, 14 May 2018 12:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 1
assert data['data']['activities'][0]['duration'] == '1:00:00'
assert data['data']['activities'][0]['title'] == 'Cycling - 2018-05-14 14:05:00' # noqa
assert (
data['data']['activities'][0]['title']
== 'Cycling - 2018-05-14 14:05:00'
) # noqa
assert data['data']['activities'][0]['ascent'] is None
assert data['data']['activities'][0]['ave_speed'] == 7.0
assert data['data']['activities'][0]['descent'] is None
@ -643,18 +613,19 @@ def test_edit_an_activity_wo_gpx_timezone(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=2,
duration=3600,
activity_date='2018-05-15 15:05',
distance=8,
title='Activity test'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
title='Activity test',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -662,7 +633,10 @@ def test_edit_an_activity_wo_gpx_timezone(
assert 'success' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Tue, 15 May 2018 13:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Tue, 15 May 2018 13:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 2
assert data['data']['activities'][0]['duration'] == '1:00:00'
@ -702,32 +676,28 @@ def test_edit_an_activity_wo_gpx_timezone(
assert records[3]['value'] == 8.0
def test_edit_an_activity_wo_gpx_partial(
app, user_1, sport_1_cycling
):
def test_edit_an_activity_wo_gpx_partial(app, user_1, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/activities/no_gpx',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=1,
duration=3600,
activity_date='2018-05-14 14:05',
distance=7
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
distance=7,
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -736,12 +706,17 @@ def test_edit_an_activity_wo_gpx_partial(
assert 'created' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0][
'activity_date'] == 'Mon, 14 May 2018 14:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Mon, 14 May 2018 14:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 1
assert data['data']['activities'][0]['duration'] == '1:00:00'
assert data['data']['activities'][0]['title'] == 'Cycling - 2018-05-14 14:05:00' # noqa
assert (
data['data']['activities'][0]['title']
== 'Cycling - 2018-05-14 14:05:00'
) # noqa
assert data['data']['activities'][0]['ascent'] is None
assert data['data']['activities'][0]['ave_speed'] == 7.0
assert data['data']['activities'][0]['descent'] is None
@ -779,15 +754,11 @@ def test_edit_an_activity_wo_gpx_partial(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=1,
distance=10
)),
data=json.dumps(dict(sport_id=1, distance=10)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -796,11 +767,17 @@ def test_edit_an_activity_wo_gpx_partial(
assert 'success' in data['status']
assert len(data['data']['activities']) == 1
assert 'creation_date' in data['data']['activities'][0]
assert data['data']['activities'][0]['activity_date'] == 'Mon, 14 May 2018 14:05:00 GMT' # noqa
assert (
data['data']['activities'][0]['activity_date']
== 'Mon, 14 May 2018 14:05:00 GMT'
) # noqa
assert data['data']['activities'][0]['user_id'] == 1
assert data['data']['activities'][0]['sport_id'] == 1
assert data['data']['activities'][0]['duration'] == '1:00:00'
assert data['data']['activities'][0]['title'] == 'Cycling - 2018-05-14 14:05:00' # noqa
assert (
data['data']['activities'][0]['title']
== 'Cycling - 2018-05-14 14:05:00'
) # noqa
assert data['data']['activities'][0]['ascent'] is None
assert data['data']['activities'][0]['ave_speed'] == 10.0
assert data['data']['activities'][0]['descent'] is None
@ -842,21 +819,17 @@ def test_edit_an_activity_wo_gpx_invalid_payload(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -872,62 +845,55 @@ def test_edit_an_activity_wo_gpx_incorrect_data(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
sport_id=1,
duration=3600,
activity_date='15/2018',
distance=10
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
data=json.dumps(
dict(
sport_id=1, duration=3600, activity_date='15/2018', distance=10
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert response.status_code == 500
assert 'error' in data['status']
assert 'Error. Please try again or contact the administrator.' \
assert (
'Error. Please try again or contact the administrator.'
in data['message']
)
def test_edit_an_activity_no_activity(
app, user_1, sport_1_cycling
):
def test_edit_an_activity_no_activity(app, user_1, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
sport_id=1,
duration=3600,
activity_date='2018-05-15 14:05',
distance=10
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
distance=10,
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -943,24 +909,20 @@ def test_refresh_an_activity_with_gpx(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
# Edit some activity data
@ -971,14 +933,11 @@ def test_refresh_an_activity_with_gpx(
response = client.patch(
'/api/activities/1',
content_type='application/json',
data=json.dumps(dict(
refresh=True,
)),
data=json.dumps(dict(refresh=True)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())

View File

@ -15,76 +15,64 @@ def test_delete_an_activity_with_gpx(app, user_1, sport_1_cycling, gpx_file):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
response = client.delete(
'/api/activities/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
assert response.status_code == 204
def test_delete_an_activity_with_gpx_different_user(
app, user_1, user_2, sport_1_cycling, gpx_file):
app, user_1, user_2, sport_1_cycling, gpx_file
):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='toto@toto.com',
password='87654321'
)),
content_type='application/json'
data=json.dumps(dict(email='toto@toto.com', password='87654321')),
content_type='application/json',
)
response = client.delete(
'/api/activities/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -100,19 +88,15 @@ def test_delete_an_activity_wo_gpx(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/activities/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
assert response.status_code == 204
@ -123,19 +107,15 @@ def test_delete_an_activity_wo_gpx_different_user(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='toto@toto.com',
password='87654321'
)),
content_type='application/json'
data=json.dumps(dict(email='toto@toto.com', password='87654321')),
content_type='application/json',
)
response = client.delete(
'/api/activities/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -149,19 +129,15 @@ def test_delete_an_activity_no_activity(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/activities/9999',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert response.status_code == 404
@ -169,28 +145,25 @@ def test_delete_an_activity_no_activity(app, user_1):
def test_delete_an_activity_with_gpx_invalid_file(
app, user_1, sport_1_cycling, gpx_file):
app, user_1, sport_1_cycling, gpx_file
):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
client.post(
'/api/activities',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data='{"sport_id": 1}'
data='{"sport_id": 1}',
),
headers=dict(
content_type='multipart/form-data',
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
gpx_filepath = get_gpx_filepath(1)
@ -200,15 +173,16 @@ def test_delete_an_activity_with_gpx_invalid_file(
response = client.delete(
'/api/activities/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert response.status_code == 500
assert 'error' in data['status']
assert 'Error. Please try again or contact the administrator.' \
assert (
'Error. Please try again or contact the administrator.'
in data['message']
)

View File

@ -1,8 +1,4 @@
def test_add_activity(
app, sport_1_cycling, user_1, activity_cycling_user_1
):
def test_add_activity(app, sport_1_cycling, user_1, activity_cycling_user_1):
activity_cycling_user_1.title = 'Test'
assert 1 == activity_cycling_user_1.id
@ -12,7 +8,9 @@ def test_add_activity(
assert 10.0 == float(activity_cycling_user_1.distance)
assert '0:17:04' == str(activity_cycling_user_1.duration)
assert 'Test' == activity_cycling_user_1.title
assert '<Activity \'Cycling\' - 2018-01-01 00:00:00>' == str(activity_cycling_user_1) # noqa
assert '<Activity \'Cycling\' - 2018-01-01 00:00:00>' == str(
activity_cycling_user_1
) # noqa
serialized_activity = activity_cycling_user_1.serialize()
assert 1 == serialized_activity['id']
@ -44,7 +42,12 @@ def test_add_activity(
def test_add_segment(
app, sport_1_cycling, user_1, activity_cycling_user_1,
activity_cycling_user_1_segment
app,
sport_1_cycling,
user_1,
activity_cycling_user_1,
activity_cycling_user_1_segment,
):
assert '<Segment \'0\' for activity \'1\'>' == str(activity_cycling_user_1_segment) # noqa
assert '<Segment \'0\' for activity \'1\'>' == str(
activity_cycling_user_1_segment
) # noqa

View File

@ -7,13 +7,15 @@ def test_user_registration(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='justatest',
email='test@test.com',
password='12345678',
password_conf='12345678'
)),
content_type='application/json'
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -27,13 +29,15 @@ def test_user_registration_user_already_exists(app, user_1):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='test',
email='test@test.com',
password='12345678',
password_conf='12345678'
)),
content_type='application/json'
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -46,17 +50,21 @@ def test_user_registration_invalid_short_username(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='t',
email='test@test.com',
password='12345678',
password_conf='12345678'
)),
content_type='application/json'
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == "Errors: Username: 3 to 12 characters required.\n" # noqa
assert (
data['message'] == "Errors: Username: 3 to 12 characters required.\n"
) # noqa
assert response.content_type == 'application/json'
assert response.status_code == 400
@ -65,17 +73,21 @@ def test_user_registration_invalid_long_username(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='testestestestestest',
email='test@test.com',
password='12345678',
password_conf='12345678'
)),
content_type='application/json'
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == "Errors: Username: 3 to 12 characters required.\n" # noqa
assert (
data['message'] == "Errors: Username: 3 to 12 characters required.\n"
) # noqa
assert response.content_type == 'application/json'
assert response.status_code == 400
@ -84,13 +96,15 @@ def test_user_registration_invalid_email(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='test',
email='test@test',
password='12345678',
password_conf='12345678'
)),
content_type='application/json'
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -103,17 +117,21 @@ def test_user_registration_invalid_short_password(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='test',
email='test@test.com',
password='1234567',
password_conf='1234567'
)),
content_type='application/json'
password_conf='1234567',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == "Errors: Password: 8 characters required.\n" # noqa
assert (
data['message'] == "Errors: Password: 8 characters required.\n"
) # noqa
assert response.content_type == 'application/json'
assert response.status_code == 400
@ -122,17 +140,22 @@ def test_user_registration_mismatched_password(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='test',
email='test@test.com',
password='12345678',
password_conf='87654321'
)),
content_type='application/json'
password_conf='87654321',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == "Errors: Password and password confirmation don\'t match.\n" # noqa
assert (
data['message']
== "Errors: Password and password confirmation don\'t match.\n"
) # noqa
assert response.content_type == 'application/json'
assert response.status_code == 400
@ -142,7 +165,7 @@ def test_user_registration_invalid_json(app):
response = client.post(
'/api/auth/register',
data=json.dumps(dict()),
content_type='application/json'
content_type='application/json',
)
data = json.loads(response.data.decode())
assert response.status_code, 400
@ -154,10 +177,13 @@ def test_user_registration_invalid_json_keys_no_username(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
email='test@test.com',
password='12345678',
password_conf='12345678')),
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
@ -170,10 +196,11 @@ def test_user_registration_invalid_json_keys_no_email(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
username='test',
password='12345678',
password_conf='12345678')),
data=json.dumps(
dict(
username='test', password='12345678', password_conf='12345678'
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
@ -186,10 +213,13 @@ def test_user_registration_invalid_json_keys_no_password(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username='test',
email='test@test.com',
password_conf='12345678')),
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
@ -202,10 +232,9 @@ def test_user_registration_invalid_json_keys_no_password_conf(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
username='test',
email='test@test.com',
password='12345678')),
data=json.dumps(
dict(username='test', email='test@test.com', password='12345678')
),
content_type='application/json',
)
data = json.loads(response.data.decode())
@ -218,17 +247,22 @@ def test_user_registration_invalid_data(app):
client = app.test_client()
response = client.post(
'/api/auth/register',
data=json.dumps(dict(
data=json.dumps(
dict(
username=1,
email='test@test.com',
password='12345678',
password_conf='12345678'
)),
password_conf='12345678',
)
),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert response.status_code == 500
assert 'Error. Please try again or contact the administrator.' in data['message'] # noqa
assert (
'Error. Please try again or contact the administrator.'
in data['message']
) # noqa
assert 'error' in data['status']
@ -258,11 +292,8 @@ def test_login_registered_user(app, user_1):
client = app.test_client()
response = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -276,11 +307,8 @@ def test_login_no_registered_user(app):
client = app.test_client()
response = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -294,7 +322,7 @@ def test_login_invalid_payload(app):
response = client.post(
'/api/auth/login',
data=json.dumps(dict()),
content_type='application/json'
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -307,11 +335,8 @@ def test_login_registered_user_invalid_password(app, user_1):
client = app.test_client()
response = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='123456789'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='123456789')),
content_type='application/json',
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -325,20 +350,16 @@ def test_logout(app, user_1):
# user login
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
# valid token logout
response = client.get(
'/api/auth/logout',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -350,21 +371,17 @@ def test_logout_expired_token(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
# invalid token logout
time.sleep(4)
response = client.get(
'/api/auth/logout',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
@ -375,8 +392,8 @@ def test_logout_expired_token(app, user_1):
def test_logout_invalid(app):
client = app.test_client()
response = client.get(
'/api/auth/logout',
headers=dict(Authorization='Bearer invalid'))
'/api/auth/logout', headers=dict(Authorization='Bearer invalid')
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == 'Invalid token. Please log in again.'
@ -385,9 +402,7 @@ def test_logout_invalid(app):
def test_logout_invalid_headers(app):
client = app.test_client()
response = client.get(
'/api/auth/logout',
headers=dict())
response = client.get('/api/auth/logout', headers=dict())
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == 'Provide a valid auth token.'
@ -398,19 +413,15 @@ def test_user_profile_minimal(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/auth/profile',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -431,19 +442,15 @@ def test_user_profile_full(app, user_1_full):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/auth/profile',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -466,25 +473,25 @@ def test_user_profile_full(app, user_1_full):
def test_user_profile_with_activities(
app, user_1, sport_1_cycling, sport_2_running,
activity_cycling_user_1, activity_running_user_1
app,
user_1,
sport_1_cycling,
sport_2_running,
activity_cycling_user_1,
activity_running_user_1,
):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/auth/profile',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -504,8 +511,8 @@ def test_user_profile_with_activities(
def test_invalid_profile(app):
client = app.test_client()
response = client.get(
'/api/auth/profile',
headers=dict(Authorization='Bearer invalid'))
'/api/auth/profile', headers=dict(Authorization='Bearer invalid')
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == 'Invalid token. Please log in again.'
@ -516,16 +523,14 @@ def test_user_profile_valid_update(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
first_name='John',
last_name='Doe',
location='Somewhere',
@ -533,13 +538,13 @@ def test_user_profile_valid_update(app, user_1):
birth_date='1980-01-01',
password='87654321',
password_conf='87654321',
timezone='America/New_York'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
timezone='America/New_York',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -551,27 +556,25 @@ def test_user_profile_valid_update_without_password(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
first_name='John',
last_name='Doe',
location='Somewhere',
bio='just a random guy',
birth_date='1980-01-01'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
birth_date='1980-01-01',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -583,23 +586,17 @@ def test_user_profile_valid_update_with_one_field(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict(
first_name='John'
)),
data=json.dumps(dict(first_name='John')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -611,21 +608,17 @@ def test_user_profile_update_invalid_json(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert response.status_code == 400
@ -637,16 +630,14 @@ def test_user_profile_invalid_password(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
first_name='John',
last_name='Doe',
location='Somewhere',
@ -654,17 +645,19 @@ def test_user_profile_invalid_password(app, user_1):
birth_date='1980-01-01',
password='87654321',
password_conf='876543210',
timezone='America/New_York'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
timezone='America/New_York',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == 'Password and password confirmation don\'t match.\n' # noqa
assert (
data['message'] == 'Password and password confirmation don\'t match.\n'
) # noqa
assert response.status_code == 400
@ -672,33 +665,33 @@ def test_user_profile_missing_password_conf(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/profile/edit',
content_type='application/json',
data=json.dumps(dict(
data=json.dumps(
dict(
first_name='John',
last_name='Doe',
location='Somewhere',
bio='just a random guy',
birth_date='1980-01-01',
password='87654321',
timezone='America/New_York'
)),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
timezone='America/New_York',
)
),
headers=dict(
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'error'
assert data['message'] == 'Password and password confirmation don\'t match.\n' # noqa
assert (
data['message'] == 'Password and password confirmation don\'t match.\n'
) # noqa
assert response.status_code == 400
@ -706,22 +699,17 @@ def test_update_user_picture(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/picture',
data=dict(
file=(BytesIO(b'avatar'), 'avatar.png')
),
data=dict(file=(BytesIO(b'avatar'), 'avatar.png')),
headers=dict(
content_type='multipart/form-data',
authorization='Bearer ' +
json.loads(resp_login.data.decode())['auth_token']
)
authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -731,14 +719,12 @@ def test_update_user_picture(app, user_1):
response = client.post(
'/api/auth/picture',
data=dict(
file=(BytesIO(b'avatar2'), 'avatar2.png')
),
data=dict(file=(BytesIO(b'avatar2'), 'avatar2.png')),
headers=dict(
content_type='multipart/form-data',
authorization='Bearer ' +
json.loads(resp_login.data.decode())['auth_token']
)
authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'success'
@ -752,19 +738,16 @@ def test_update_user_no_picture(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/picture',
headers=dict(
content_type='multipart/form-data',
authorization='Bearer ' +
json.loads(resp_login.data.decode())['auth_token']
)
authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'fail'
@ -776,22 +759,17 @@ def test_update_user_invalid_picture(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/auth/picture',
data=dict(
file=(BytesIO(b'avatar'), 'avatar.bmp')
),
data=dict(file=(BytesIO(b'avatar'), 'avatar.bmp')),
headers=dict(
content_type='multipart/form-data',
authorization='Bearer ' +
json.loads(resp_login.data.decode())['auth_token']
)
authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token'],
),
)
data = json.loads(response.data.decode())
assert data['status'] == 'fail'

View File

@ -6,7 +6,8 @@ def test_development_config(app):
assert app.config['DEBUG']
assert not app.config['TESTING']
assert app.config['SQLALCHEMY_DATABASE_URI'] == os.environ.get(
'DATABASE_URL')
'DATABASE_URL'
)
def test_testing_config(app):
@ -15,4 +16,5 @@ def test_testing_config(app):
assert app.config['TESTING']
assert not app.config['PRESERVE_CONTEXT_ON_EXCEPTION']
assert app.config['SQLALCHEMY_DATABASE_URI'] == os.environ.get(
'DATABASE_TEST_URL')
'DATABASE_TEST_URL'
)

File diff suppressed because it is too large Load Diff

View File

@ -3,9 +3,7 @@ import datetime
from fittrackee_api.activities.models import Record
def test_record_model(
app, user_1, sport_1_cycling, activity_cycling_user_1
):
def test_record_model(app, user_1, sport_1_cycling, activity_cycling_user_1):
record_ld = Record.query.filter_by(
user_id=activity_cycling_user_1.user_id,
sport_id=activity_cycling_user_1.sport_id,
@ -49,9 +47,7 @@ def test_record_model_none_value(
assert record_serialize['value'] is None
def test_add_as_records(
app, user_1, sport_1_cycling, activity_cycling_user_1
):
def test_add_as_records(app, user_1, sport_1_cycling, activity_cycling_user_1):
record_as = Record.query.filter_by(
user_id=activity_cycling_user_1.user_id,
sport_id=activity_cycling_user_1.sport_id,
@ -67,9 +63,7 @@ def test_add_as_records(
assert isinstance(record_serialize.get('value'), float)
def test_add_fd_records(
app, user_1, sport_1_cycling, activity_cycling_user_1
):
def test_add_fd_records(app, user_1, sport_1_cycling, activity_cycling_user_1):
record_fd = Record.query.filter_by(
user_id=activity_cycling_user_1.user_id,
sport_id=activity_cycling_user_1.sport_id,
@ -85,9 +79,7 @@ def test_add_fd_records(
assert isinstance(record_serialize.get('value'), float)
def test_add_ld_records(
app, user_1, sport_1_cycling, activity_cycling_user_1
):
def test_add_ld_records(app, user_1, sport_1_cycling, activity_cycling_user_1):
record_ld = Record.query.filter_by(
user_id=activity_cycling_user_1.user_id,
sport_id=activity_cycling_user_1.sport_id,

View File

@ -4,14 +4,14 @@ expected_sport_1_cycling_result = {
'id': 1,
'label': 'Cycling',
'img': None,
'_can_be_deleted': True
'_can_be_deleted': True,
}
expected_sport_2_running_result = {
'id': 2,
'label': 'Running',
'img': None,
'_can_be_deleted': True
'_can_be_deleted': True,
}
@ -19,19 +19,15 @@ def test_get_all_sports(app, user_1, sport_1_cycling, sport_2_running):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/sports',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -47,19 +43,15 @@ def test_get_a_sport(app, user_1, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/sports/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -74,19 +66,15 @@ def test_get_a_sport_invalid(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/sports/1',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -99,23 +87,17 @@ def test_add_a_sport(app, user_1_admin):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/sports',
content_type='application/json',
data=json.dumps(dict(
label='Cycling'
)),
data=json.dumps(dict(label='Cycling')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -130,23 +112,17 @@ def test_add_a_sport_not_admin(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/sports',
content_type='application/json',
data=json.dumps(dict(
label='surfing'
)),
data=json.dumps(dict(label='surfing')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -160,21 +136,17 @@ def test_add_a_sport_invalid_payload(app, user_1_admin):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.post(
'/api/sports',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -187,23 +159,17 @@ def test_update_a_sport(app, user_1_admin, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/sports/1',
content_type='application/json',
data=json.dumps(dict(
label='cycling updated'
)),
data=json.dumps(dict(label='cycling updated')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -218,23 +184,17 @@ def test_update_a_sport_not_admin(app, user_1, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/sports/1',
content_type='application/json',
data=json.dumps(dict(
label='cycling updated'
)),
data=json.dumps(dict(label='cycling updated')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -248,21 +208,17 @@ def test_update_a_sport_invalid_payload(app, user_1_admin):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/sports/1',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -275,23 +231,17 @@ def test_update_a_sport_invalid_id(app, user_1_admin):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.patch(
'/api/sports/1',
content_type='application/json',
data=json.dumps(dict(
label='cycling updated'
)),
data=json.dumps(dict(label='cycling updated')),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -304,20 +254,16 @@ def test_delete_a_sport(app, user_1_admin, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/sports/1',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
assert response.status_code == 204
@ -326,20 +272,16 @@ def test_delete_a_sport_not_admin(app, user_1, sport_1_cycling):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/sports/1',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -352,21 +294,17 @@ def test_delete_a_sport_invalid_id(app, user_1_admin):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/sports/1',
content_type='application/json',
data=json.dumps(dict()),
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -381,20 +319,16 @@ def test_delete_a_sport_with_an_activity(
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='admin@example.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='admin@example.com', password='12345678')),
content_type='application/json',
)
response = client.delete(
'/api/sports/1',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())

File diff suppressed because it is too large Load Diff

View File

@ -18,20 +18,16 @@ def test_single_user(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
f'/api/users/{user_1.id}',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -55,27 +51,27 @@ def test_single_user(app, user_1):
def test_single_user_with_activities(
app, user_1, sport_1_cycling, sport_2_running,
activity_cycling_user_1, activity_running_user_1
app,
user_1,
sport_1_cycling,
sport_2_running,
activity_cycling_user_1,
activity_running_user_1,
):
"""=> Get single user details"""
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
f'/api/users/{user_1.id}',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -103,20 +99,16 @@ def test_single_user_no_id(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/users/blah',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -130,20 +122,16 @@ def test_single_user_wrong_id(app, user_1):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/users/99999999999',
content_type='application/json',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())
@ -158,19 +146,15 @@ def test_users_list(app, user_1, user_2):
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(
email='test@test.com',
password='12345678'
)),
content_type='application/json'
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
response = client.get(
'/api/users',
headers=dict(
Authorization='Bearer ' + json.loads(
resp_login.data.decode()
)['auth_token']
)
Authorization='Bearer '
+ json.loads(resp_login.data.decode())['auth_token']
),
)
data = json.loads(response.data.decode())

View File

@ -80,14 +80,14 @@ def register_user():
return jsonify(response_object), 403
# get post data
post_data = request.get_json()
if not post_data or post_data.get('username') is None \
or post_data.get('email') is None \
or post_data.get('password') is None \
or post_data.get('password_conf') is None:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
if (
not post_data
or post_data.get('username') is None
or post_data.get('email') is None
or post_data.get('password') is None
or post_data.get('password_conf') is None
):
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
username = post_data.get('username')
email = post_data.get('email')
@ -102,27 +102,21 @@ def register_user():
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
return jsonify(response_object), 500
if ret != '':
response_object = {
'status': 'error',
'message': 'Errors: ' + ret
}
response_object = {'status': 'error', 'message': 'Errors: ' + ret}
return jsonify(response_object), 400
try:
# check for existing user
user = User.query.filter(
or_(User.username == username, User.email == email)).first()
or_(User.username == username, User.email == email)
).first()
if not user:
# add new user to db
new_user = User(
username=username,
email=email,
password=password
)
new_user = User(username=username, email=email, password=password)
new_user.timezone = 'Europe/Paris'
db.session.add(new_user)
db.session.commit()
@ -131,13 +125,13 @@ def register_user():
response_object = {
'status': 'success',
'message': 'Successfully registered.',
'auth_token': auth_token.decode()
'auth_token': auth_token.decode(),
}
return jsonify(response_object), 201
else:
response_object = {
'status': 'error',
'message': 'Sorry. That user already exists.'
'message': 'Sorry. That user already exists.',
}
return jsonify(response_object), 400
# handler errors
@ -147,7 +141,7 @@ def register_user():
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
return jsonify(response_object), 500
@ -202,10 +196,7 @@ def login_user():
# get post data
post_data = request.get_json()
if not post_data:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
email = post_data.get('email')
password = post_data.get('password')
@ -218,13 +209,13 @@ def login_user():
response_object = {
'status': 'success',
'message': 'Successfully logged in.',
'auth_token': auth_token.decode()
'auth_token': auth_token.decode(),
}
return jsonify(response_object), 200
else:
response_object = {
'status': 'error',
'message': 'Invalid credentials.'
'message': 'Invalid credentials.',
}
return jsonify(response_object), 404
# handler errors
@ -233,7 +224,7 @@ def login_user():
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
return jsonify(response_object), 500
@ -291,19 +282,16 @@ def logout_user(user_id):
if not isinstance(user_id, str):
response_object = {
'status': 'success',
'message': 'Successfully logged out.'
'message': 'Successfully logged out.',
}
return jsonify(response_object), 200
else:
response_object = {
'status': 'error',
'message': resp
}
response_object = {'status': 'error', 'message': resp}
return jsonify(response_object), 401
else:
response_object = {
'status': 'error',
'message': 'Provide a valid auth token.'
'message': 'Provide a valid auth token.',
}
return jsonify(response_object), 401
@ -360,10 +348,7 @@ def get_user_status(user_id):
"""
user = User.query.filter_by(id=user_id).first()
response_object = {
'status': 'success',
'data': user.serialize()
}
response_object = {'status': 'success', 'data': user.serialize()}
return jsonify(response_object), 200
@ -434,10 +419,7 @@ def edit_user(user_id):
# get post data
post_data = request.get_json()
if not post_data:
response_object = {
'status': 'error',
'message': 'Invalid payload.'
}
response_object = {'status': 'error', 'message': 'Invalid payload.'}
return jsonify(response_object), 400
first_name = post_data.get('first_name')
last_name = post_data.get('last_name')
@ -452,7 +434,7 @@ def edit_user(user_id):
if password_conf != password:
response_object = {
'status': 'error',
'message': 'Password and password confirmation don\'t match.\n'
'message': 'Password and password confirmation don\'t match.\n',
}
return jsonify(response_object), 400
else:
@ -478,7 +460,7 @@ def edit_user(user_id):
response_object = {
'status': 'success',
'message': 'User profile updated.'
'message': 'User profile updated.',
}
return jsonify(response_object), 200
@ -488,7 +470,7 @@ def edit_user(user_id):
appLog.error(e)
response_object = {
'status': 'error',
'message': 'Error. Please try again or contact the administrator.'
'message': 'Error. Please try again or contact the administrator.',
}
return jsonify(response_object), 500
@ -560,18 +542,12 @@ def edit_picture(user_id):
file = request.files['file']
filename = secure_filename(file.filename)
dirpath = os.path.join(
current_app.config['UPLOAD_FOLDER'],
'pictures',
str(user_id)
current_app.config['UPLOAD_FOLDER'], 'pictures', str(user_id)
)
if not os.path.exists(dirpath):
os.makedirs(dirpath)
absolute_picture_path = os.path.join(dirpath, filename)
relative_picture_path = os.path.join(
'pictures',
str(user_id),
filename
)
relative_picture_path = os.path.join('pictures', str(user_id), filename)
try:
user = User.query.filter_by(id=user_id).first()
@ -585,7 +561,7 @@ def edit_picture(user_id):
response_object = {
'status': 'success',
'message': 'User picture updated.'
'message': 'User picture updated.',
}
return jsonify(response_object), 200
@ -594,7 +570,7 @@ def edit_picture(user_id):
appLog.error(e)
response_object = {
'status': 'fail',
'message': 'Error during picture update.'
'message': 'Error during picture update.',
}
return jsonify(response_object), 500
@ -637,9 +613,7 @@ def del_picture(user_id):
user.picture = None
db.session.commit()
response_object = {
'status': 'no content'
}
response_object = {'status': 'no content'}
return jsonify(response_object), 204
except (exc.IntegrityError, ValueError) as e:
@ -647,6 +621,6 @@ def del_picture(user_id):
appLog.error(e)
response_object = {
'status': 'fail',
'message': 'Error during picture deletion.'
'message': 'Error during picture deletion.',
}
return jsonify(response_object), 500

View File

@ -23,19 +23,19 @@ class User(db.Model):
bio = db.Column(db.String(200), nullable=True)
picture = db.Column(db.String(255), nullable=True)
timezone = db.Column(db.String(50), nullable=True)
activities = db.relationship('Activity',
lazy=True,
backref=db.backref('users', lazy='joined'))
records = db.relationship('Record',
lazy=True,
backref=db.backref('users', lazy='joined'))
activities = db.relationship(
'Activity', lazy=True, backref=db.backref('users', lazy='joined')
)
records = db.relationship(
'Record', lazy=True, backref=db.backref('users', lazy='joined')
)
def __repr__(self):
return f'<User {self.username!r}>'
def __init__(
self, username, email, password,
created_at=datetime.datetime.utcnow()):
self, username, email, password, created_at=datetime.datetime.utcnow()
):
self.username = username
self.email = email
self.password = bcrypt.generate_password_hash(
@ -52,17 +52,18 @@ class User(db.Model):
"""
try:
payload = {
'exp': datetime.datetime.utcnow() + datetime.timedelta(
'exp': datetime.datetime.utcnow()
+ datetime.timedelta(
days=current_app.config.get('TOKEN_EXPIRATION_DAYS'),
seconds=current_app.config.get('TOKEN_EXPIRATION_SECONDS')
seconds=current_app.config.get('TOKEN_EXPIRATION_SECONDS'),
),
'iat': datetime.datetime.utcnow(),
'sub': user_id
'sub': user_id,
}
return jwt.encode(
payload,
current_app.config.get('SECRET_KEY'),
algorithm='HS256'
algorithm='HS256',
)
except Exception as e:
return e
@ -76,8 +77,8 @@ class User(db.Model):
"""
try:
payload = jwt.decode(
auth_token,
current_app.config.get('SECRET_KEY'))
auth_token, current_app.config.get('SECRET_KEY')
)
return payload['sub']
except jwt.ExpiredSignatureError:
return 'Signature expired. Please log in again.'
@ -91,14 +92,13 @@ class User(db.Model):
sports = []
total = (None, None)
if nb_activity > 0:
sports = db.session.query(
func.count(Activity.sport_id)
).group_by(
Activity.sport_id
).all()
sports = (
db.session.query(func.count(Activity.sport_id))
.group_by(Activity.sport_id)
.all()
)
total = db.session.query(
func.sum(Activity.distance),
func.sum(Activity.duration)
func.sum(Activity.distance), func.sum(Activity.duration)
).first()
return {
'id': self.id,

View File

@ -85,9 +85,7 @@ def get_users(auth_user_id):
users = User.query.all()
response_object = {
'status': 'success',
'data': {
'users': [user.serialize() for user in users]
}
'data': {'users': [user.serialize() for user in users]},
}
return jsonify(response_object), 200
@ -148,19 +146,13 @@ def get_single_user(auth_user_id, user_id):
- User does not exist
"""
response_object = {
'status': 'fail',
'message': 'User does not exist.'
}
response_object = {'status': 'fail', 'message': 'User does not exist.'}
try:
user = User.query.filter_by(id=int(user_id)).first()
if not user:
return jsonify(response_object), 404
else:
response_object = {
'status': 'success',
'data': user.serialize()
}
response_object = {'status': 'success', 'data': user.serialize()}
return jsonify(response_object), 200
except ValueError:
return jsonify(response_object), 404
@ -192,16 +184,13 @@ def get_picture(user_id):
- No picture.
"""
response_object = {
'status': 'not found',
'message': 'No picture.'
}
response_object = {'status': 'not found', 'message': 'No picture.'}
try:
user = User.query.filter_by(id=int(user_id)).first()
if not user:
response_object = {
'status': 'fail',
'message': 'User does not exist.'
'message': 'User does not exist.',
}
return jsonify(response_object), 404
if user.picture is not None:
@ -238,7 +227,4 @@ def ping_pong():
:statuscode 200: success
"""
return jsonify({
'status': 'success',
'message': 'pong!'
})
return jsonify({'status': 'success', 'message': 'pong!'})

View File

@ -47,12 +47,14 @@ def verify_extension(file_type, req):
else 'PICTURE_ALLOWED_EXTENSIONS'
)
if not ('.' in file.filename and
file.filename.rsplit('.', 1)[1].lower() in
current_app.config.get(allowed_extensions)):
if not (
'.' in file.filename
and file.filename.rsplit('.', 1)[1].lower()
in current_app.config.get(allowed_extensions)
):
response_object = {
'status': 'fail',
'message': 'File extension not allowed.'
'message': 'File extension not allowed.',
}
return response_object
@ -61,7 +63,7 @@ def verify_extension(file_type, req):
def verify_user(current_request, verify_admin):
response_object = {
'status': 'error',
'message': 'Something went wrong. Please contact us.'
'message': 'Something went wrong. Please contact us.',
}
code = 401
auth_header = current_request.headers.get('Authorization')
@ -110,7 +112,7 @@ def can_view_activity(auth_user_id, activity_user_id):
if auth_user_id != activity_user_id:
response_object = {
'status': 'error',
'message': 'You do not have permissions.'
'message': 'You do not have permissions.',
}
return response_object, 403
return None, None

View File

@ -24,9 +24,8 @@ def drop_db():
def initdata():
"""Init the database."""
admin = User(
username='admin',
email='admin@example.com',
password='mpwoadmin')
username='admin', email='admin@example.com', password='mpwoadmin'
)
admin.admin = True
admin.timezone = 'Europe/Paris'
db.session.add(admin)
@ -61,9 +60,11 @@ def initdata():
@app.cli.command()
def recalculate():
print("Starting activities data refresh")
activities = Activity.query.filter(Activity.gpx != None).order_by( # noqa
Activity.activity_date.asc()
).all()
activities = (
Activity.query.filter(Activity.gpx != None)
.order_by(Activity.activity_date.asc()) # noqa
.all()
)
if len(activities) == 0:
print('➡️ no activities to upgrade.')
return None