def main():
""" Main interface """
parser = _create_parser()
args = parser.parse_args()
if len(sys.argv) == 1:
parser.print_usage()
parser.exit(status=1)
elif args.countries:
print("\n".join(map(str, data.countries())))
sys.exit()
elif args.states:
print("\n".join(map(str, data.states())))
sys.exit()
region = data.read(args.region)
if not region:
print(f"Region '{args.region}' not found.")
sys.exit(1)
title = f"COVID-19 Infections ({region})"
with tempfile.NamedTemporaryFile(mode="w") as datfile:
for stats in region.daily_stats():
datfile.write(f"{stats.date}\t{stats.cases}\t{stats.avg}\n")
datfile.flush()
gnuplot = Gnuplot("./covid.gp", args.terminal, args.output)
gnuplot.set_var("datfile", datfile.name)
gnuplot.set_var("title", title)
gnuplot.run()
