65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
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]
|