def save_arguments(g, script_path, keys):
d = g.copy()
target_folder = d.pop('target_folder')
if not target_folder:
target_folder = make_enumerated_folder_for(script_path)
arguments_folder = make_folder(join(target_folder, 'arguments'))
# Migrate paths
path_by_key = OrderedDict()
for k, v in d.items():
if not k.endswith('_path'):
continue
file_name = _get_argument_file_name(k, v)
# Save a copy of each file
shutil.copy(v, join(arguments_folder, file_name))
# Make the reference point to the local copy
path_by_key[k] = file_name
d = sort_dictionary(d, keys)
d.update(path_by_key)
# Save global arguments as JSON
target_file = open(join(arguments_folder, 'arguments.json'), 'w')
json.dump(d, target_file, indent=4, separators=(',', ': '))
# Save global arguments as CSV
target_file = open(join(arguments_folder, 'arguments.csv'), 'w')
csv_writer = csv.writer(target_file)
for x in d.items():
csv_writer.writerow(x)
target_path = join(target_folder,
'selected_nyc_traffic_injury_summary_by_month.jpg')
fig.savefig(target_path)
return target_path
if __name__ == '__main__':
argument_parser = ArgumentParser()
argument_parser.add_argument('--target_folder',
metavar='FOLDER',
type=make_folder)
argument_parser.add_argument('--target_date',
metavar='DATE',
type=parse_date)
argument_parser.add_argument('--from_date',
metavar='DATE',
type=parse_date)
argument_parser.add_argument('--to_date', metavar='DATE', type=parse_date)
argument_parser.add_argument('--search_address', metavar='ADDRESS')
argument_parser.add_argument('--search_radius_in_meters',
metavar='RADIUS',
type=float)
args = argument_parser.parse_args()
d = run(args.target_folder or make_enumerated_folder_for(__file__),
args.target_date, args.from_date, args.to_date,
args.search_address, args.search_radius_in_meters)
print(format_summary(d))
metavar='FOLDER', type=make_folder)
argument_parser.add_argument(
'--demographic_by_year_table_path',
metavar='PATH', required=True)
argument_parser.add_argument(
'--demographic_by_year_table_population_column',
metavar='COLUMN', required=True)
argument_parser.add_argument(
'--electricity_consumption_by_population_table_path',
metavar='PATH', required=True)
argument_parser.add_argument(
'--electricity_consumption_by_population_table_population_column',
metavar='COLUMN', required=True)
argument_parser.add_argument(
'--electricity_consumption_by_population_table_consumption_column',
metavar='COLUMN', required=True)
args = argument_parser.parse_args()
d = run(
args.target_folder or make_enumerated_folder_for(__file__),
read_csv(args.demographic_by_year_table_path),
args.demographic_by_year_table_population_column,
read_csv(args.electricity_consumption_by_population_table_path),
args.electricity_consumption_by_population_table_population_column,
args.electricity_consumption_by_population_table_consumption_column)
print(format_summary(d))
def _get_alternate_country_name(country_name):
if not ALTERNATE_COUNTRY_NAME_BY_COUNTRY_NAME:
_prepare_country_names()
return ALTERNATE_COUNTRY_NAME_BY_COUNTRY_NAME[country_name]
def _format_label_for_region(region_name):
x = region_name.lower()
x = x.replace(" ", "-")
x = x.replace("&", "and")
return "region-%s" % x
def _format_label_for_income_group(income_group_name):
x = income_group_name.lower()
x = x.replace(" ", "-")
x = x.replace(":", "")
x = x.replace("non", "non-")
x = x.replace("-income", "")
return "income-group-%s" % x
if __name__ == "__main__":
argument_parser = ArgumentParser()
argument_parser.add_argument("--target_folder", metavar="FOLDER", type=make_folder)
argument_parser.add_argument("--target_year", metavar="YEAR", type=int, required=True)
args = argument_parser.parse_args()
d = run(args.target_folder or make_enumerated_folder_for(__file__), args.target_year)
print(format_summary(d))
