From bcb9b7e9b2b09400c0cb1f2702ed51df10f3d42c Mon Sep 17 00:00:00 2001
From: SebF <sebastien@forissier.eu>
Date: Sat, 9 Oct 2021 20:40:50 +0200
Subject: [PATCH] =?UTF-8?q?m=C3=A9thodes=20de=20sauvegarde=20d=C3=A9plac?=
 =?UTF-8?q?=C3=A9es=20dans=20une=20classe?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 osm_vc63/save.py              | 48 +++++++++++++++++++++++++++++++++++
 recup_donnees_OSM_Overpass.py | 46 +++------------------------------
 2 files changed, 51 insertions(+), 43 deletions(-)
 create mode 100644 osm_vc63/save.py

diff --git a/osm_vc63/save.py b/osm_vc63/save.py
new file mode 100644
index 0000000..b4c5426
--- /dev/null
+++ b/osm_vc63/save.py
@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+
+import json
+from pyexcel_ods3 import save_data
+from collections import OrderedDict
+
+
+class Save:
+    def as_ods(self, fields, data, dossier, nom_req):
+        """Sauvegarde de data dans un classeur ods"""
+
+        ODSdataSheet = OrderedDict()
+        ODSdata = []
+        ODSdata.append(fields.keys())
+        index_line = 2
+
+        for element in data["elements"]:
+            line = []
+            index_col = 0
+
+            for field in fields.keys():
+                if field in element["tags"]:
+                    if field == "capacity":
+                        val = element["tags"][field]
+                        line.append(int(val) if val.isdigit() else val)
+                    else:
+                        line.append(element["tags"][field])
+                else:
+                    line.append("")
+                index_col = index_col + 1
+
+            ODSdata.append(line)
+            index_line = index_line + 1
+
+        ODSdataSheet.update({"resultats": ODSdata})
+
+        save_data(dossier + nom_req + ".ods", ODSdataSheet)
+
+        print("Sauvegarde résultats format ODS pour " + nom_req)
+
+    def as_json(self, export_json, dossier, nom_req):
+        """Enregistrement du JSON"""
+
+        jsonFile = open(dossier + nom_req + ".json", "w")
+        jsonFile.write(json.dumps(export_json))
+        jsonFile.close()
+
+        print("Sauvegarde résultat format JSON/OSM " + nom_req)
diff --git a/recup_donnees_OSM_Overpass.py b/recup_donnees_OSM_Overpass.py
index 27a2a8d..a112edf 100644
--- a/recup_donnees_OSM_Overpass.py
+++ b/recup_donnees_OSM_Overpass.py
@@ -23,6 +23,7 @@ from collections import OrderedDict
 import os
 from osm_vc63 import errors
 from osm_vc63 import requetes
+from osm_vc63.save import Save
 
 overpass_url="http://overpass-api.de/api/interpreter"
 geo_api_url = "https://api-adresse.data.gouv.fr"
@@ -69,47 +70,6 @@ trad_bicycle_parking = {
     "handlebar_holder": "Accroche-guidons"}
 
 
-def sauvegarde_ods(fields, data, dossier, nom_req):
-    """Sauvegarde de la requête"""
-
-    ODSdataSheet = OrderedDict()
-    ODSdata = []
-    ODSdata.append(fields.keys())
-    index_line = 2
-
-    for element in data["elements"]:
-        line = []
-        index_col = 0
-
-        for field in fields.keys():
-            if field in element["tags"]:
-                if field == "capacity":
-                    val = element["tags"][field]
-                    line.append(int(val) if val.isdigit() else val)
-                else:
-                    line.append(element["tags"][field])
-            else:
-                line.append("")
-            index_col = index_col + 1
-        
-        ODSdata.append(line)
-        index_line = index_line + 1
-
-    ODSdataSheet.update({"resultats": ODSdata})
-
-    save_data(dossier + nom_req + ".ods", ODSdataSheet)
-
-    print("Sauvegarde résultats format ODS pour " + nom_req)
-
-
-def sauvegarde_json(export_json, dossier, nom_req):
-    jsonFile = open(dossier + nom_req + ".json", "w")
-    jsonFile.write(json.dumps(export_json))
-    jsonFile.close()
-
-    print("Sauvegarde résultat format JSON/OSM " + nom_req)
-
-
 def nettoyage_json_pour_umap(data, overpass_query_fields):
     export_json = {"version": data["version"],
             "generator" : data["generator"] + " and ETALAB API",
@@ -241,8 +201,8 @@ def executer_requete_et_exporter_resultats(nom_req, critere, aire_de_recherche,
     # Sauvegarde
     os.makedirs(dossier_sauvegarde, exist_ok = True)
 
-    sauvegarde_json(export_json, dossier_sauvegarde, nom_req)
-    sauvegarde_ods(overpass_query_fields, data, dossier_sauvegarde, nom_req)
+    Save().as_json(export_json, dossier_sauvegarde, nom_req)
+    Save().as_ods(overpass_query_fields, data, dossier_sauvegarde, nom_req)