mirror of
https://github.com/enzet/map-machine.git
synced 2025-05-14 01:26:51 +02:00
126 lines
3.5 KiB
Python
126 lines
3.5 KiB
Python
"""
|
|
Author: Sergey Vartanov (me@enzet.ru).
|
|
"""
|
|
import argparse
|
|
import sys
|
|
|
|
from typing import List, Optional
|
|
|
|
BOXES: List[str] = [" ", "▏", "▎", "▍", "▌", "▋", "▊", "▉"]
|
|
BOXES_LENGTH: int = len(BOXES)
|
|
|
|
|
|
def parse_options(args) -> argparse.Namespace:
|
|
"""
|
|
Parse Röntgen command-line options.
|
|
"""
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
"-i", "--input",
|
|
dest="input_file_name",
|
|
metavar="<path>",
|
|
nargs="*",
|
|
help="input XML file name (if not specified, file will be downloaded "
|
|
"using OpenStreetMap API)")
|
|
parser.add_argument(
|
|
"-o", "--output",
|
|
dest="output_file_name",
|
|
metavar="<path>",
|
|
default="map.svg",
|
|
help="output SVG file name (map.svg by default)")
|
|
parser.add_argument(
|
|
"-b", "--boundary-box",
|
|
dest="boundary_box",
|
|
metavar="<lon1>,<lat1>,<lon2>,<lat2>",
|
|
help="geo boundary box, use space before \"-\" for negative values",
|
|
required=True)
|
|
parser.add_argument(
|
|
"-s", "--scale",
|
|
metavar="<float>",
|
|
help="OSM zoom level (may not be integer, default is 18)",
|
|
default=18,
|
|
dest="scale",
|
|
type=float,
|
|
required=True)
|
|
parser.add_argument(
|
|
"-nn", "--no-draw-nodes",
|
|
dest="draw_nodes",
|
|
action="store_false",
|
|
default=True)
|
|
parser.add_argument(
|
|
"-nw", "--no-draw-ways",
|
|
dest="draw_ways",
|
|
action="store_false",
|
|
default=True)
|
|
parser.add_argument(
|
|
"--captions", "--no-draw-captions",
|
|
dest="draw_captions",
|
|
default="main")
|
|
parser.add_argument(
|
|
"--show-missing-tags",
|
|
dest="show_missing_tags",
|
|
action="store_true")
|
|
parser.add_argument(
|
|
"--no-show-missing-tags",
|
|
dest="show_missing_tags",
|
|
action="store_false")
|
|
parser.add_argument(
|
|
"--overlap",
|
|
dest="overlap",
|
|
default=12,
|
|
type=int)
|
|
parser.add_argument(
|
|
"--mode",
|
|
default="normal")
|
|
parser.add_argument(
|
|
"--seed",
|
|
default="")
|
|
parser.add_argument(
|
|
"--level",
|
|
default=None)
|
|
|
|
arguments: argparse.Namespace = parser.parse_args(args[1:])
|
|
|
|
if arguments.boundary_box:
|
|
arguments.boundary_box = arguments.boundary_box.replace(" ", "")
|
|
|
|
return arguments
|
|
|
|
|
|
def progress_bar(
|
|
number: int, total: int, length: int = 20, step: int = 1000,
|
|
text: str = "") -> None:
|
|
"""
|
|
Draw progress bar using Unicode symbols.
|
|
|
|
:param number: current value
|
|
:param total: maximum value
|
|
:param length: progress bar length.
|
|
:param step: frequency of progress bar updating (assuming that numbers go
|
|
subsequently)
|
|
:param text: short description
|
|
"""
|
|
if number == -1:
|
|
print(f"100 % {length * '█'}▏{text}")
|
|
elif number % step == 0:
|
|
ratio: float = number / total
|
|
parts: int = int(ratio * length * BOXES_LENGTH)
|
|
fill_length: int = int(parts / BOXES_LENGTH)
|
|
box: str = BOXES[int(parts - fill_length * BOXES_LENGTH)]
|
|
print(
|
|
f"{str(int(int(ratio * 1000) / 10)):>3} % {fill_length * '█'}{box}"
|
|
f"{int(length - fill_length - 1) * ' '}▏{text}")
|
|
sys.stdout.write("\033[F")
|
|
|
|
|
|
def error(message: Optional[str] = None):
|
|
"""
|
|
Print error message.
|
|
|
|
:param message: message to print.
|
|
"""
|
|
if message:
|
|
print(f"Error: {message}.")
|
|
else:
|
|
print("Error.")
|