Create module georequest

Author: pavloae

pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "georef-ar-py"
+version = "0.0.1"
+authors = [
+  { name="Pablo E. Andino", email="[email protected]" },
+]
+description = "Paquete para consultas y procesamiento de respuestas sobre la API de georef-ar"
+readme = "README.md"
+requires-python = ">=3.7"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+"Homepage" = "https://github.com/pavloae/georef-ar-py"
+"Bug Tracker" = "https://github.com/pavloae/georef-ar-py/issues"

src/georef_ar_py/__init__.py

@@ -0,0 +1,58 @@
+import urllib
+
+import requests as requests
+
+API_BASE_URL = "https://apis.datos.gob.ar/georef/api/"
+
+
+def get_similar(endpoint, nombre, **kwargs):
+    kwargs["nombre"] = nombre
+    url = "{}{}?{}".format(API_BASE_URL, endpoint, urllib.parse.urlencode(kwargs))
+    return requests.get(url).json()[endpoint]
+
+
+def get_similar_bulk(endpoint, nombres):
+    """Normaliza una lista de nombres de alguna de las entidades geográficas."""
+
+    # realiza consulta a la API
+    data = {
+        endpoint: [
+            {"nombre": nombre, "max": 1} for nombre in nombres
+        ]}
+    url = API_BASE_URL + endpoint
+    results = requests.post(
+        url, json=data, headers={"Content-Type": "application/json"}
+    ).json()
+
+    # convierte a una lista de "resultado más probable" o "vacío" cuando no hay
+    parsed_results = [
+        single_result[endpoint][0] if single_result[endpoint] else {}
+        for single_result in results["resultados"]
+    ]
+
+    return parsed_results
+
+
+def get_territorial_units(ubicaciones):
+    """Pide las unidades territoriales que contienen a c/punto de una lista de coordenadas."""
+
+    # realiza consulta a la API
+    endpoint = "ubicacion"
+    data = {
+        "ubicaciones": [
+            {"lat": ubicacion["lat"], "lon": ubicacion["lon"], "aplanar": True}
+            for ubicacion in ubicaciones
+        ]}
+    url = API_BASE_URL + endpoint
+
+    results = requests.post(
+        url, json=data, headers={"Content-Type": "application/json"}
+    ).json()
+
+    # convierte a una lista de "resultado más probable" o "vacío" cuando no hay
+    parsed_results = [
+        single_result[endpoint] if single_result[endpoint] else {}
+        for single_result in results["resultados"]
+    ]
+
+    return parsed_results

src/georef_ar_py/georequests.py

@@ -0,0 +1 @@
+

tests/__init__.py

@@ -0,0 +1,32 @@
+import unittest
+
+from src.georef_ar_py import georequests
+
+
+class MyTestCase(unittest.TestCase):
+    def test_get_similar(self):
+        response = georequests.get_similar('provincias', 'San Juan')
+        self.assertEqual(response[0]['id'], '70')
+        self.assertEqual(response[0]['nombre'], 'San Juan')
+
+    def test_similar_bulk(self):
+        response = georequests.get_similar_bulk("provincias", ["pxa", "sant fe"])
+        self.assertEqual(len(response), 2)
+        self.assertEqual(len(response[0]), 0)
+        self.assertEqual(response[1]['id'], '82')
+        self.assertEqual(response[1]['nombre'], 'Santa Fe')
+
+    def test_get_territorial_units(self):
+        location1, location2 = georequests.get_territorial_units([
+            {"lat": -32.9477132, "lon": -60.6304658},
+            {"lat": -34.6037389, "lon": -58.3815704}
+        ])
+        self.assertEqual('Santa Fe', location1['provincia_nombre'])
+        self.assertEqual('Rosario', location1['departamento_nombre'])
+
+        self.assertEqual(location2['provincia_nombre'], 'Ciudad Autónoma de Buenos Aires')
+        self.assertEqual(location2['departamento_nombre'], 'Comuna 1')
+
+
+if __name__ == '__main__':
+    unittest.main()