Initial iol

iol/handler.py

@@ -0,0 +1,64 @@
+import logging
+import time
+from functools import lru_cache
+
+import requests as r
+from fastapi import APIRouter
+
+BASE_URL = "https://api.invertironline.com/"
+USER = open("/var/openfaas/secrets/iol-user").read().strip()
+PASS = open("/var/openfaas/secrets/iol-pass").read().strip()
+SECRET = open("/var/openfaas/secrets/iol-api-secret").read().strip()
+
+router = APIRouter()
+
+def get_ttl_hash(seconds=3600):
+    """Return the same value withing `seconds` time period"""
+    return round(time.time() / seconds)
+
+@lru_cache
+def get_token(ttl_hash=None):
+    logging.error("getting token")
+    url = BASE_URL + "token"
+    data = {"username": USER, "password": PASS, "grant_type": "password"}
+    response = r.post(url, data=data)
+    access_token = response.json()["access_token"]
+    refresh_token = response.json()["refresh_token"]
+    return access_token, refresh_token
+
+
+def refresh_token(refresh_token):
+    url = BASE_URL + "token"
+    data = {"refresh_token": refresh_token, "grant_type": "refresh_token"}
+    response = r.post(url, data=data)
+    access_token = response.json()["access_token"]
+    refresh_token = response.json()["refresh_token"]
+    return access_token, refresh_token
+
+
+paises = {
+    "US": "estados_Unidos",
+    "AR": "argentina",
+}
+
+@lru_cache
+def get(url, ttl_hash=None):
+    logging.error("getting data")
+    access, refresh = get_token(ttl_hash=get_ttl_hash())
+    response = r.get(url, headers={"Authorization": "Bearer " + access, "Accept": "application/json"})
+    return response.json()
+
+
+@router.get("/{secret}/accion/{pais}/{accion}")
+def get_accion(secret, pais, accion):
+    if secret != SECRET:
+        raise Exception("BAD")
+    pais = pais.upper()
+    accion = accion.upper()
+    access, refresh = get_token(ttl_hash=get_ttl_hash())
+    url = (
+        BASE_URL
+        + f"/api/v2/Cotizaciones/acciones/{pais}/Todos?cotizacionInstrumentoModel.instrumento=acciones&cotizacionInstrumentoModel.pais={paises[pais]}&api_key={access}"
+    )
+    data = get(url, ttl_hash=get_ttl_hash())
+    return [a for a in data["titulos"] if a["simbolo"] == accion][0]

iol/requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+requests
+uvicorn[standard]