API: generate a static map for activities w/ gpx

mpwo_api/migrations/versions/5a42db64e872_.py

@@ -0,0 +1,30 @@
+"""empty message
+
+Revision ID: 5a42db64e872
+Revises: 92adde6ac0d0
+Create Date: 2018-05-30 10:52:33.433687
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '5a42db64e872'
+down_revision = '92adde6ac0d0'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('activities', sa.Column('map', sa.String(length=255), nullable=True))
+    op.create_unique_constraint(None, 'sports', ['img'])
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, 'sports', type_='unique')
+    op.drop_column('activities', 'map')
+    # ### end Alembic commands ###

mpwo_api/mpwo_api/activities/activities.py

@@ -280,13 +280,8 @@ def delete_activity(auth_user_id, activity_id):
     try:
         activity = Activity.query.filter_by(id=activity_id).first()
         if activity:
-            gpx_filepath = activity.gpx
             db.session.delete(activity)
             db.session.commit()
-
-            if gpx_filepath:
-                os.remove(gpx_filepath)
-
             response_object = {
                 'status': 'no content'
             }

mpwo_api/mpwo_api/activities/models.py

@@ -1,4 +1,5 @@
 import datetime
+import os
 
 from mpwo_api import db
 from sqlalchemy.dialects import postgresql
@@ -133,6 +134,7 @@ class Activity(db.Model):
     max_speed = db.Column(db.Numeric(6, 2), nullable=True)   # km/h
     ave_speed = db.Column(db.Numeric(6, 2), nullable=True)   # km/h
     bounds = db.Column(postgresql.ARRAY(db.Float), nullable=True)
+    map = db.Column(db.String(255), nullable=True)
     segments = db.relationship('ActivitySegment',
                                lazy=True,
                                cascade='all, delete',
@@ -196,7 +198,8 @@ class Activity(db.Model):
             "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]
+            "records": [record.serialize() for record in self.records],
+            "with_map": self.map is not None
         }
 
     @classmethod
@@ -248,6 +251,17 @@ def on_activity_update(mapper, connection, activity):
                 update_records(activity.user_id, sport_id, connection, session)
 
 
+@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:
+            os.remove(old_record.map)
+        if old_record.gpx:
+            os.remove(old_record.gpx)
+
+
 class ActivitySegment(db.Model):
     __tablename__ = "activity_segments"
     activity_id = db.Column(

mpwo_api/mpwo_api/activities/utils.py

@@ -7,6 +7,7 @@ import gpxpy.gpx
 from flask import current_app
 from mpwo_api import db
 from sqlalchemy import exc
+from staticmap import Line, StaticMap
 from werkzeug.utils import secure_filename
 
 from .models import Activity, ActivitySegment, Sport
@@ -150,12 +151,16 @@ def get_gpx_info(gpx_file):
     }
     max_speed = 0
     start = 0
+    map_data = []
 
     for segment_idx, segment in enumerate(gpx.tracks[0].segments):
         segment_start = 0
         for point_idx, point in enumerate(segment.points):
             if point_idx == 0 and start == 0:
                 start = point.time
+            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)
@@ -179,7 +184,7 @@ def get_gpx_info(gpx_file):
         bounds.max_longitude
     ]
 
-    return gpx_data
+    return gpx_data, map_data
 
 
 def get_chart_data(gpx_file):
@@ -229,7 +234,10 @@ def get_file_path(auth_user_id, dir_path, filename):
     return file_path
 
 
-def get_new_file_path(auth_user_id, activity_date, old_filename, sport):
+def get_new_file_path(
+        auth_user_id, activity_date, sport, old_filename=None, extension=None
+):
+    if not extension:
         extension = f".{old_filename.rsplit('.', 1)[1].lower()}"
     _, new_filename = tempfile.mkstemp(
         prefix=f'{activity_date}_{sport}_',
@@ -244,9 +252,17 @@ def get_new_file_path(auth_user_id, activity_date, old_filename, sport):
     return file_path
 
 
+def generate_map(map_filepath, map_data):
+    m = StaticMap(400, 225, 10)
+    line = Line(map_data, '#3388FF', 4)
+    m.add_line(line)
+    image = m.render()
+    image.save(map_filepath)
+
+
 def process_one_gpx_file(auth_user_id, activity_data, file_path, filename):
     try:
-        gpx_data = get_gpx_info(file_path)
+        gpx_data, map_data = get_gpx_info(file_path)
 
         sport = Sport.query.filter_by(id=activity_data.get('sport_id')).first()
         new_filepath = get_new_file_path(
@@ -257,6 +273,14 @@ def process_one_gpx_file(auth_user_id, activity_data, file_path, filename):
         )
         os.rename(file_path, new_filepath)
         gpx_data['filename'] = new_filepath
+
+        map_filepath = get_new_file_path(
+            auth_user_id=auth_user_id,
+            activity_date=gpx_data['start'],
+            extension='.png',
+            sport=sport.label
+        )
+        generate_map(map_filepath, map_data)
     except (gpxpy.gpx.GPXXMLSyntaxException, TypeError) as e:
         raise ActivityException('error', 'Error during gpx file parsing.', e)
     except Exception as e:
@@ -265,6 +289,7 @@ def process_one_gpx_file(auth_user_id, activity_data, file_path, filename):
     try:
         new_activity = create_activity(
             auth_user_id, activity_data, gpx_data)
+        new_activity.map = map_filepath
         db.session.add(new_activity)
         db.session.flush()
 

mpwo_api/mpwo_api/tests/test_activities_api_1_post.py

@@ -19,6 +19,7 @@ def assert_activity_data_with_gpx(data):
     assert data['data']['activities'][0]['moving'] == '0:04:10'
     assert data['data']['activities'][0]['pauses'] is None
     assert data['data']['activities'][0]['with_gpx'] is True
+    assert data['data']['activities'][0]['with_map'] is True
     assert len(data['data']['activities'][0]['segments']) == 1
 
     segment = data['data']['activities'][0]['segments'][0]
@@ -76,6 +77,7 @@ def assert_activity_data_wo_gpx(data):
     assert data['data']['activities'][0]['moving'] == '1:00:00'
     assert data['data']['activities'][0]['pauses'] is None
     assert data['data']['activities'][0]['with_gpx'] is False
+    assert data['data']['activities'][0]['with_map'] is False
 
     assert len(data['data']['activities'][0]['segments']) == 0
 

mpwo_api/mpwo_api/tests/test_records_api.py

@@ -762,8 +762,6 @@ def test_get_records_after_sport_change(
     )
     data = json.loads(response.data.decode())
 
-    print(data['data']['records'])
-
     assert response.status_code == 200
     assert 'success' in data['status']
     assert len(data['data']['records']) == 8

mpwo_api/requirements.txt

@@ -27,6 +27,7 @@ MarkupSafe==1.0
 mccabe==0.6.1
 more-itertools==4.2.0
 mysqlclient==1.3.12
+Pillow==5.1.0
 pluggy==0.6.0
 psycopg2==2.7.3.2
 py==1.5.3
@@ -45,6 +46,7 @@ python-editor==1.0.3
 requests==2.18.4
 six==1.11.0
 SQLAlchemy==1.2.7
+staticmap==0.5.3
 testfixtures==5.3.1
 urllib3==1.22
 Werkzeug==0.14.1