FitTrackee/fittrackee/workouts/utils/maps.py

48 lines
1.5 KiB
Python
Raw Normal View History

2022-02-16 17:46:22 +01:00
import hashlib
import random
from typing import Dict, List
2022-02-16 17:46:22 +01:00
from flask import current_app
from staticmap import Line, StaticMap
from fittrackee import VERSION
2022-02-16 17:46:22 +01:00
from fittrackee.files import get_absolute_file_path
def get_static_map_tile_server_url(tile_server_config: Dict) -> str:
if tile_server_config['STATICMAP_SUBDOMAINS']:
subdomains = tile_server_config['STATICMAP_SUBDOMAINS'].split(',')
subdomain = f'{random.choice(subdomains)}.' # nosec
else:
subdomain = ''
return tile_server_config['URL'].replace('{s}.', subdomain)
2022-02-16 17:46:22 +01:00
def generate_map(map_filepath: str, map_data: List) -> None:
"""
Generate and save map image from map data
"""
m = StaticMap(400, 225, 10)
m.headers = {'User-Agent': f'FitTrackee v{VERSION}'}
2022-02-16 17:46:22 +01:00
if not current_app.config['TILE_SERVER']['DEFAULT_STATICMAP']:
m.url_template = get_static_map_tile_server_url(
current_app.config['TILE_SERVER']
2022-02-16 17:46:22 +01:00
)
line = Line(map_data, '#3388FF', 4)
m.add_line(line)
image = m.render()
image.save(map_filepath)
def get_map_hash(map_filepath: str) -> str:
"""
Generate a md5 hash used as id instead of workout id, to retrieve map
image (maps are sensitive data)
"""
2022-05-28 20:01:14 +02:00
md5 = hashlib.md5() # nosec # need 3.9+ to use 'usedforsecurity' flag
2022-02-16 17:46:22 +01:00
absolute_map_filepath = get_absolute_file_path(map_filepath)
with open(absolute_map_filepath, 'rb') as f:
for chunk in iter(lambda: f.read(128 * md5.block_size), b''):
md5.update(chunk)
return md5.hexdigest()