34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
import datetime
|
|
import os
|
|
import redis
|
|
import pickle
|
|
from esy.client import ESIClient
|
|
from esy.auth import ESIAuthenticator
|
|
|
|
|
|
class ESICache(object):
|
|
def __init__(self, **kwargs):
|
|
self._r = redis.Redis(**kwargs)
|
|
# self._r = redis.StrictRedis(host=redis_url, port=redis_port, db=db)
|
|
|
|
def get(self, key):
|
|
# return pickle.loads(self._r[key])
|
|
return pickle.loads(self._r.get(key))
|
|
|
|
def set(self, key, data, cached_until: datetime.datetime):
|
|
self._r.set(key, pickle.dumps(data), ex=cached_until - datetime.datetime.now(datetime.timezone.utc))
|
|
|
|
def __contains__(self, item):
|
|
# return item in self._r
|
|
return self._r.exists(item)
|
|
|
|
|
|
esi_client_id = os.getenv('ESI_CLIENT_ID')
|
|
esi_secret_key = os.getenv('ESI_SECRET_KEY')
|
|
|
|
esi_cache = ESICache(host=os.getenv("REDIS_URL"), port=int(os.getenv("REDIS_PORT")), db="0",
|
|
password=os.getenv("REDIS_PASSWD"))
|
|
|
|
esi_client = ESIClient.get_client(user_agent=os.getenv('ESI_USER_AGENT'), cache=esi_cache)
|
|
esi_auth = ESIAuthenticator()
|