[P] Add bump remove function
This commit is contained in:
@@ -14,11 +14,24 @@ line_layer_colors = [
|
||||
[1, 0, 1, 1],
|
||||
]
|
||||
|
||||
|
||||
def get_lat_lon(point: dict[str, float] | list[float] | tuple[float, float]) -> tuple[float, float] | None:
|
||||
if isinstance(point, dict):
|
||||
lat = point.get("lat")
|
||||
lon = point.get("lon")
|
||||
else:
|
||||
lat, lon = point
|
||||
|
||||
if lat is None or lon is None:
|
||||
return None
|
||||
return lat, lon
|
||||
|
||||
|
||||
class MapViewApp(App):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
self.mapview = None
|
||||
self.mapview: MapView | None = None
|
||||
self.datasource = Datasource(user_id=1)
|
||||
self.line_layers = dict()
|
||||
self.car_markers = dict()
|
||||
@@ -110,24 +123,34 @@ class MapViewApp(App):
|
||||
self.pothole_markers.append(marker)
|
||||
|
||||
def set_bump_marker(self, point):
|
||||
if isinstance(point, dict):
|
||||
lat = point.get("lat")
|
||||
lon = point.get("lon")
|
||||
else:
|
||||
lat, lon = point
|
||||
|
||||
lat, lon = get_lat_lon(point)
|
||||
if lat is None or lon is None:
|
||||
return
|
||||
|
||||
|
||||
marker = MapMarker(
|
||||
lat=lat,
|
||||
lon=lon,
|
||||
source="images/bump.png"
|
||||
source="images/bump.png"
|
||||
)
|
||||
|
||||
self.mapview.add_marker(marker)
|
||||
self.bump_markers.append(marker)
|
||||
|
||||
def delete_bump_marker(self, point):
|
||||
lat, lon = get_lat_lon(point)
|
||||
if lat is None or lon is None:
|
||||
return
|
||||
|
||||
marker = MapMarker(
|
||||
lat=lat,
|
||||
lon=lon,
|
||||
source="images/bump.png"
|
||||
)
|
||||
|
||||
if marker in self.bump_markers:
|
||||
self.mapview.remove_marker(marker)
|
||||
self.bump_markers.pop(self.bump_markers.index(marker))
|
||||
|
||||
|
||||
def build(self):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user