def main() -> None:
"""Handle command line arguments and call other modules as needed."""
p = poster.Poster()
drawers = {
"grid": grid_drawer.GridDrawer(p),
"calendar": calendar_drawer.CalendarDrawer(p),
"heatmap": heatmap_drawer.HeatmapDrawer(p),
"circular": circular_drawer.CircularDrawer(p),
"github": github_drawer.GithubDrawer(p),
}
args_parser = argparse.ArgumentParser(prog=__app_name__)
args_parser.add_argument(
"--gpx-dir",
dest="gpx_dir",
metavar="DIR",
type=str,
default=".",
help="Directory containing GPX files (default: current directory).",
)
args_parser.add_argument(
"--output",
metavar="FILE",
type=str,
default="poster.svg",
help='Name of generated SVG image file (default: "poster.svg").',
)
args_parser.add_argument(
"--language",
metavar="LANGUAGE",
type=str,
default="",
help="Language (default: english).",
)
args_parser.add_argument(
"--localedir",
metavar="DIR",
type=str,
help=
"The directory where the translation files can be found (default: the system's locale directory).",
)
args_parser.add_argument(
"--year",
metavar="YEAR",
type=str,
default="all",
help=
'Filter tracks by year; "NUM", "NUM-NUM", "all" (default: all years)',
)
args_parser.add_argument("--title",
metavar="TITLE",
type=str,
help="Title to display.")
args_parser.add_argument(
"--athlete",
metavar="NAME",
type=str,
default="John Doe",
help='Athlete name to display (default: "John Doe").',
)
args_parser.add_argument(
"--special",
metavar="FILE",
action="append",
default=[],
help=
"Mark track file from the GPX directory as special; use multiple times to mark "
"multiple tracks.",
)
types = '", "'.join(drawers.keys())
args_parser.add_argument(
"--type",
metavar="TYPE",
default="grid",
choices=drawers.keys(),
help=
f'Type of poster to create (default: "grid", available: "{types}").',
)
args_parser.add_argument(
"--background-color",
dest="background_color",
metavar="COLOR",
type=str,
default="#222222",
help='Background color of poster (default: "#222222").',
)
args_parser.add_argument(
"--track-color",
dest="track_color",
metavar="COLOR",
type=str,
default="#4DD2FF",
help='Color of tracks (default: "#4DD2FF").',
)
args_parser.add_argument(
"--track-color2",
dest="track_color2",
metavar="COLOR",
type=str,
help="Secondary color of tracks (default: none).",
)
args_parser.add_argument(
"--text-color",
dest="text_color",
metavar="COLOR",
type=str,
default="#FFFFFF",
help='Color of text (default: "#FFFFFF").',
)
args_parser.add_argument(
"--special-color",
dest="special_color",
metavar="COLOR",
default="#FFFF00",
help='Special track color (default: "#FFFF00").',
)
args_parser.add_argument(
"--special-color2",
dest="special_color2",
metavar="COLOR",
help="Secondary color of special tracks (default: none).",
)
args_parser.add_argument(
"--units",
dest="units",
metavar="UNITS",
type=str,
choices=["metric", "imperial"],
default="metric",
help='Distance units; "metric", "imperial" (default: "metric").',
)
args_parser.add_argument(
"--clear-cache",
dest="clear_cache",
action="store_true",
help="Clear the track cache.",
)
args_parser.add_argument(
"--from-strava",
dest="from_strava",
metavar="FILE",
type=str,
help="strava config file",
)
args_parser.add_argument("--verbose",
dest="verbose",
action="store_true",
help="Verbose logging.")
args_parser.add_argument("--logfile",
dest="logfile",
metavar="FILE",
type=str)
args_parser.add_argument(
"--special-distance",
dest="special_distance",
metavar="DISTANCE",
type=float,
default=10.0,
help="Special Distance1 by km and color with the special_color",
)
args_parser.add_argument(
"--special-distance2",
dest="special_distance2",
metavar="DISTANCE",
type=float,
default=20.0,
help="Special Distance2 by km and corlor with the special_color2",
)
args_parser.add_argument(
"--min-distance",
dest="min_distance",
metavar="DISTANCE",
type=float,
default=1.0,
help="min distance by km for track filter",
)
for _, drawer in drawers.items():
drawer.create_args(args_parser)
args = args_parser.parse_args()
for _, drawer in drawers.items():
drawer.fetch_args(args)
log = logging.getLogger("gpxtrackposter")
log.setLevel(logging.INFO if args.verbose else logging.ERROR)
if args.logfile:
handler = logging.FileHandler(args.logfile)
log.addHandler(handler)
loader = track_loader.TrackLoader()
loader.set_cache_dir(
os.path.join(appdirs.user_cache_dir(__app_name__, __app_author__),
"tracks"))
if not loader.year_range.parse(args.year):
raise ParameterError(f"Bad year range: {args.year}.")
loader.special_file_names = args.special
loader.set_min_length(args.min_distance * Units().km)
if args.clear_cache:
print("Clearing cache...")
loader.clear_cache()
if args.from_strava:
tracks = loader.load_strava_tracks(args.from_strava)
else:
tracks = loader.load_tracks(args.gpx_dir)
if not tracks:
if not args.clear_cache:
print("No tracks found.")
return
print(
f"Creating poster of type {args.type} with {len(tracks)} tracks and storing it in file {args.output}..."
)
p.set_language(args.language, args.localedir)
p.set_athlete(args.athlete)
p.set_title(args.title if args.title else p.translate("MY TRACKS"))
p.special_distance = {
"special_distance": args.special_distance * Units().km,
"special_distance2": args.special_distance2 * Units().km,
}
p.colors = {
"background": args.background_color,
"track": args.track_color,
"track2": args.track_color2 or args.track_color,
"special": args.special_color,
"special2": args.special_color2 or args.special_color,
"text": args.text_color,
}
p.units = args.units
p.set_tracks(tracks)
if args.type == "github":
p.height = 55 + p.years.count() * 43
p.draw(drawers[args.type], args.output)
def main():
"""Handle command line arguments and call other modules as needed."""
p = poster.Poster()
drawers = {
"grid": grid_drawer.GridDrawer(p),
"circular": circular_drawer.CircularDrawer(p),
"github": github_drawer.GithubDrawer(p),
}
args_parser = argparse.ArgumentParser()
args_parser.add_argument(
"--gpx-dir",
dest="gpx_dir",
metavar="DIR",
type=str,
default=".",
help="Directory containing GPX files (default: current directory).",
)
args_parser.add_argument(
"--output",
metavar="FILE",
type=str,
default="poster.svg",
help='Name of generated SVG image file (default: "poster.svg").',
)
args_parser.add_argument(
"--language",
metavar="LANGUAGE",
type=str,
default="",
help="Language (default: english).",
)
args_parser.add_argument(
"--year",
metavar="YEAR",
type=str,
default="all",
help=
'Filter tracks by year; "NUM", "NUM-NUM", "all" (default: all years)',
)
args_parser.add_argument("--title",
metavar="TITLE",
type=str,
help="Title to display.")
args_parser.add_argument(
"--athlete",
metavar="NAME",
type=str,
default="John Doe",
help='Athlete name to display (default: "John Doe").',
)
args_parser.add_argument(
"--special",
metavar="FILE",
action="append",
default=[],
help=
"Mark track file from the GPX directory as special; use multiple times to mark "
"multiple tracks.",
)
types = '", "'.join(drawers.keys())
args_parser.add_argument(
"--type",
metavar="TYPE",
default="grid",
choices=drawers.keys(),
help=
f'Type of poster to create (default: "grid", available: "{types}").',
)
args_parser.add_argument(
"--background-color",
dest="background_color",
metavar="COLOR",
type=str,
default="#222222",
help='Background color of poster (default: "#222222").',
)
args_parser.add_argument(
"--track-color",
dest="track_color",
metavar="COLOR",
type=str,
default="#4DD2FF",
help='Color of tracks (default: "#4DD2FF").',
)
args_parser.add_argument(
"--track-color2",
dest="track_color2",
metavar="COLOR",
type=str,
help="Secondary color of tracks (default: none).",
)
args_parser.add_argument(
"--text-color",
dest="text_color",
metavar="COLOR",
type=str,
default="#FFFFFF",
help='Color of text (default: "#FFFFFF").',
)
args_parser.add_argument(
"--special-color",
dest="special_color",
metavar="COLOR",
default="#FFFF00",
help='Special track color (default: "#FFFF00").',
)
args_parser.add_argument(
"--special-color2",
dest="special_color2",
metavar="COLOR",
help="Secondary color of special tracks (default: none).",
)
args_parser.add_argument(
"--units",
dest="units",
metavar="UNITS",
type=str,
choices=["metric", "imperial"],
default="metric",
help='Distance units; "metric", "imperial" (default: "metric").',
)
args_parser.add_argument(
"--clear-cache",
dest="clear_cache",
action="store_true",
help="Clear the track cache.",
)
args_parser.add_argument("--verbose",
dest="verbose",
action="store_true",
help="Verbose logging.")
args_parser.add_argument("--logfile",
dest="logfile",
metavar="FILE",
type=str)
args_parser.add_argument(
"--special-distance",
dest="special_distance",
metavar="DISTANCE",
type=float,
default=10.0,
help="Special Distance1 by km and color with the special_color",
)
args_parser.add_argument(
"--special-distance2",
dest="special_distance2",
metavar="DISTANCE",
type=float,
default=20.0,
help="Special Distance2 by km and corlor with the special_color2",
)
args_parser.add_argument(
"--min-distance",
dest="min_distance",
metavar="DISTANCE",
type=float,
default=3.0,
help="min distance by km for track filter",
)
args_parser.add_argument(
"--use-localtime",
dest="use_localtime",
action="store_true",
help="Use utc time or local time",
)
args_parser.add_argument(
"--from-db",
dest="from_db",
action="store_true",
help="activities db file",
)
for _, drawer in drawers.items():
drawer.create_args(args_parser)
args = args_parser.parse_args()
for _, drawer in drawers.items():
drawer.fetch_args(args)
log = logging.getLogger("gpxtrackposter")
log.setLevel(logging.INFO if args.verbose else logging.ERROR)
if args.logfile:
handler = logging.FileHandler(args.logfile)
log.addHandler(handler)
loader = track_loader.TrackLoader()
if args.use_localtime:
loader.use_local_time = True
loader.cache_dir = os.path.join(
appdirs.user_cache_dir(__app_name__, __app_author__), "tracks")
if not loader.year_range.parse(args.year):
raise ParameterError(f"Bad year range: {args.year}.")
loader.special_file_names = args.special
loader.min_length = args.min_distance * 1000
if args.clear_cache:
print("Clearing cache...")
loader.clear_cache()
if args.from_db:
# for svg from db here if you want gpx please do not use --from-db
# args.type == "grid" means have polyline data or not
tracks = loader.load_tracks_from_db(SQL_FILE, args.type == "grid")
else:
tracks = loader.load_tracks(args.gpx_dir)
if not tracks:
if not args.clear_cache:
print("No tracks found.")
return
is_circular = args.type == "circular"
if not is_circular:
print(
f"Creating poster of type {args.type} with {len(tracks)} tracks and storing it in file {args.output}..."
)
p.set_language(args.language)
p.athlete = args.athlete
if args.title:
p.title = args.title
else:
p.title = p.trans("MY TRACKS")
p.special_distance = {
"special_distance": args.special_distance,
"special_distance2": args.special_distance2,
}
p.colors = {
"background": args.background_color,
"track": args.track_color,
"track2": args.track_color2 or args.track_color,
"special": args.special_color,
"special2": args.special_color2 or args.special_color,
"text": args.text_color,
}
p.units = args.units
p.set_tracks(tracks)
# circular not add footer and header
p.drawer_type = "plain" if is_circular else "title"
if args.type == "github":
p.height = 55 + p.years.count() * 43
# for special circular
if is_circular:
years = p.years.all()[:]
for y in years:
p.years.from_year, p.years.to_year = y, y
# may be refactor
p.set_tracks(tracks)
p.draw(drawers[args.type],
os.path.join("assets", f"year_{str(y)}.svg"))
else:
p.draw(drawers[args.type], args.output)