mirror of
https://github.com/enzet/map-machine.git
synced 2025-04-30 18:57:49 +02:00
110 lines
2.9 KiB
Python
110 lines
2.9 KiB
Python
"""
|
|
OSM address tag processing.
|
|
"""
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from colour import Color
|
|
|
|
__author__ = "Sergey Vartanov"
|
|
__email__ = "me@enzet.ru"
|
|
|
|
DEFAULT_FONT_SIZE: float = 10.0
|
|
DEFAULT_COLOR: Color = Color("#444444")
|
|
|
|
|
|
@dataclass
|
|
class Label:
|
|
"""
|
|
Text label.
|
|
"""
|
|
|
|
text: str
|
|
fill: Color = DEFAULT_COLOR
|
|
size: float = DEFAULT_FONT_SIZE
|
|
|
|
|
|
def get_address(
|
|
tags: dict[str, Any], draw_captions_mode: str, processed: set[str]
|
|
) -> list[str]:
|
|
"""
|
|
Construct address text list from the tags.
|
|
|
|
:param tags: OSM node, way or relation tags
|
|
:param draw_captions_mode: captions mode ("all", "main", or "no")
|
|
:param processed: set of processed tag keys
|
|
"""
|
|
address: list[str] = []
|
|
|
|
if draw_captions_mode == "address":
|
|
if "addr:postcode" in tags:
|
|
address.append(tags["addr:postcode"])
|
|
processed.add("addr:postcode")
|
|
if "addr:country" in tags:
|
|
address.append(tags["addr:country"])
|
|
processed.add("addr:country")
|
|
if "addr:city" in tags:
|
|
address.append(tags["addr:city"])
|
|
processed.add("addr:city")
|
|
if "addr:street" in tags:
|
|
street = tags["addr:street"]
|
|
if street.startswith("улица "):
|
|
street = "ул. " + street[len("улица ") :]
|
|
address.append(street)
|
|
processed.add("addr:street")
|
|
|
|
if "addr:housenumber" in tags:
|
|
address.append(tags["addr:housenumber"])
|
|
processed.add("addr:housenumber")
|
|
|
|
return address
|
|
|
|
|
|
def format_voltage(value: str) -> str:
|
|
"""
|
|
Format voltage value to more human-readable form.
|
|
|
|
:param value: presumably string representation of integer, in Volts
|
|
"""
|
|
try:
|
|
int_value: int = int(value)
|
|
if int_value % 1000 == 0:
|
|
return f"{int(int_value / 1000)} kV"
|
|
return f"{value} V"
|
|
except ValueError:
|
|
return value
|
|
|
|
|
|
def format_frequency(value: str) -> str:
|
|
"""Format frequency value to more human-readable form."""
|
|
return f"{value} "
|
|
|
|
|
|
def get_text(tags: dict[str, Any], processed: set[str]) -> list[Label]:
|
|
"""Get text representation of writable tags."""
|
|
texts: list[Label] = []
|
|
values: list[str] = []
|
|
|
|
if "voltage:primary" in tags:
|
|
values.append(tags["voltage:primary"])
|
|
processed.add("voltage:primary")
|
|
|
|
if "voltage:secondary" in tags:
|
|
values.append(tags["voltage:secondary"])
|
|
processed.add("voltage:secondary")
|
|
|
|
if "voltage" in tags:
|
|
values = tags["voltage"].split(";")
|
|
processed.add("voltage")
|
|
|
|
if values:
|
|
texts.append(Label(", ".join(map(format_voltage, values))))
|
|
|
|
if "frequency" in tags:
|
|
text: str = ", ".join(
|
|
map(format_frequency, tags["frequency"].split(";"))
|
|
)
|
|
texts.append(Label(text))
|
|
processed.add("frequency")
|
|
|
|
return texts
|