Calvin Ng
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
import argparse
|
|
import re
|
|
|
|
"""Endless ATC custom airport file section renumbering utility."""
|
|
|
|
|
|
def process(args, file=None):
|
|
"""Rewrites an Endless ATC airport file with renumbered sections.
|
|
|
|
Sections are renumbered from 1. Sections do not need to be numbered
|
|
in the input. Routes, however, need to be numbered in the input.
|
|
Returns the renumbered contents of the file as a list of lines as a
|
|
list of `str`.
|
|
|
|
Args:
|
|
`args`: An `argparse.Namespace`. The command line args from the invoking module.
|
|
`file`: The file to process. Defaults to `input_file` in `args`."""
|
|
|
|
header_re = re.compile(r"^(?:\[(?P<header>(?:approach)|(?:transition)|(?:(?:common)?departure)|(?:area)|(?:airport))\d*\])|^(?:route\d? *= *)")
|
|
|
|
def number_approach(match, indexes={'approach': 0, 'transition': 0, 'departure': 0, 'area': 0, 'route': 0, 'airport': 0, 'commondeparture': 0}):
|
|
header = match.group("header")
|
|
if header:
|
|
indexes['route'] = 0
|
|
else:
|
|
header = 'route'
|
|
indexes[header] += 1
|
|
return header == 'route' and f"{header}{indexes[header]} = " or f"[{header}{indexes[header]}]"
|
|
|
|
result = []
|
|
|
|
with open(file, 'r', newline='') as input_file:
|
|
for line in input_file:
|
|
result.append(header_re.sub(number_approach, line))
|
|
|
|
return result
|
|
|
|
|
|
def process_to_file(args, file=None):
|
|
"""Same as `process()`, but writes the renumbered result to a file.
|
|
|
|
Args:
|
|
`args`: An `argparse.Namespace`. The command line args from the invoking module.
|
|
`file`: The file to process. Defaults to `input_file` in `args`."""
|
|
|
|
result = process(args)
|
|
|
|
if file is None:
|
|
file = args.input_file
|
|
|
|
with open(file, 'w', newline='') as output_file:
|
|
output_file.writelines(result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description='Re-number [approach] sections for Endless ATC airport files.')
|
|
parser.add_argument('input_file')
|
|
|
|
process(parser.parse_args())
|