refactor: extract find_proposals function

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):