diff --git a/whotracksme/website/build/companies.py b/whotracksme/website/build/companies.py index 3bb4d37b23..6c6262703d 100644 --- a/whotracksme/website/build/companies.py +++ b/whotracksme/website/build/companies.py @@ -96,7 +96,7 @@ def company_page(template, company_data, data): company_id = company_data['overview']['id'] company_name = get_company_name(company_data) - write_json(f'_site/{data.url_for("company", company_id)}', + write_json(f'_site/api/v2/organizations/{data.url_for("company", company_id)}.json', demographics=company_data, initials=company_name[:2] ) @@ -122,8 +122,9 @@ def build_company_reach_chart_page(data): top100 = company_reach(data.companies, n=100) chart = Markup(overview_bars(top100, highlight=10, height=3000)) template = get_template(data, name='reach-chart-page.html', path_to_root='..') - write_json('_site/api/v2/companies.json', - top100=top100 + write_json('_site/api/v2/organizations.json', + top100=top100, + organizations=data.companies ) with open('_site/companies/reach-chart.html', 'w') as output: output.write(render_template( diff --git a/whotracksme/website/utils.py b/whotracksme/website/utils.py index d8e54a4ad0..cc45aa7673 100644 --- a/whotracksme/website/utils.py +++ b/whotracksme/website/utils.py @@ -3,16 +3,15 @@ import pathlib import os -def myconverter(o): - if isinstance(o, datetime.datetime): - return o.isoformat() +def write_json(path, **data): + def myconverter(o): + if isinstance(o, datetime.datetime): + return o.isoformat() -def write_json(aPath, **aDict): - pathlib.Path(os.path.dirname(aPath)).mkdir(parents=True, exist_ok=True) - jsonString = simplejson.dumps(aDict, indent=2, default = myconverter) - jsonFile = open(aPath, "w") - jsonFile.write(jsonString) - jsonFile.close() + pathlib.Path(os.path.dirname(path)).mkdir(parents=True, exist_ok=True) + json = simplejson.dumps(data, indent=2, default = myconverter) + with open(path, "w") as file: + file.write(json) def without_keys(d, keys): return {k: d[k] for k in d.keys() - keys}