Stahovani elevace

This commit is contained in:
Michal Kváček
2024-07-29 06:23:06 +02:00
parent 10bc0847b6
commit 9e6b0b574c
5 changed files with 58 additions and 98 deletions
-28
View File
@@ -3,7 +3,6 @@ from datetime import datetime, timedelta
from typing import List, Dict, Any
from aiocache import cached
from lxml import etree
from lxml.etree import _ElementTree
class GPXParser:
@@ -136,30 +135,3 @@ class GPXParser:
async def get_avg_altitude(self):
altitudes = await self.get_altitude()
return round(sum(altitudes) / len(altitudes), 2)
def add_terrain_elevation(self, points_with_elevation: List[Dict[str, float]]):
track_points = self.run_xpath("//gpx:trkpt")
# TODO: open elevation API umi jen presnost na 6 desetinnych mist!
track_points_index = {(float(n.attrib['lat']), float(n.attrib['lon'])): n for n in track_points}
for point_with_elevation in points_with_elevation:
lat = point_with_elevation['lat']
lng = point_with_elevation['lng']
elevation = point_with_elevation['elevation']
target_node = track_points_index.get((lat, lng))
if target_node is None:
continue
extensions = target_node.find("./extensions", self.gpx.getroot().nsmap)
if extensions is not None:
extensions.append(etree.XML(f"<terrain_elevation>{elevation}</terrain_elevation>"))
else:
target_node.append(
etree.XML(f"<extensions><terrain_elevation>{elevation}</terrain_elevation></extensions>")
)
return self.gpx
def write(self, tree: _ElementTree, output: str):
tree.write(output)