diff options
| -rwxr-xr-x | osm_proposals/proposals.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/osm_proposals/proposals.py b/osm_proposals/proposals.py index edcb453..e265fcc 100755 --- a/osm_proposals/proposals.py +++ b/osm_proposals/proposals.py @@ -28,6 +28,15 @@ def run(): arg_parser.add_argument("out_file") args = arg_parser.parse_args() + proposals = find_proposals() + + with open(args.out_file, 'w') as f: + json.dump(proposals, f) + + logger.info(f"updated {args.out_file}") + + +def find_proposals(): res = requests.get( OSMWIKI_ENDPOINT, params=dict( @@ -65,11 +74,7 @@ def run(): proposals.append(proposal) proposals.sort(key=sort_key, reverse=True) - - with open(args.out_file, 'w') as f: - json.dump([{k: v for k, v in p.items() if v is not None} for p in proposals], f) - - logger.info(f"updated {args.out_file}") + return [{k: v for k, v in p.items() if v is not None} for p in proposals] def get_template_val(tpl, name): |