add todos, stubs, prototypes

Author: Kilo59

app/redis_cache.py

@@ -0,0 +1,45 @@
+"""app.redis_cache.py"""
+# from walrus import Database
+import asyncio
+import functools
+
+import aiocache
+
+import config
+
+SETTINGS = config.get_settings()
+
+if SETTINGS.rediscloud_url:
+    REDIS_URL = SETTINGS.rediscloud_url
+else:
+    REDIS_URL = SETTINGS.local_redis_url
+
+
+@functools.lru_cache()
+def get_cache(namespace):
+    return aiocache.Cache(
+        aiocache.Cache.REDIS,
+        endpoint=REDIS_URL.host,
+        port=REDIS_URL.port,
+        password=REDIS_URL.password,
+        namespace=namespace,
+        create_connection_timeout=5,
+    )
+
+
+REDIS_CACHE = get_cache("test")
+
+
+async def cach_test():
+    # CACHE = REDIS_DB.cache()
+    await REDIS_CACHE.set("foo", "bar", ttl=30)
+    print(await REDIS_CACHE.get("foo"))
+    await REDIS_CACHE.close()
+
+
+if __name__ == "__main__":
+    # print(REDIS_DB)
+    # h = REDIS_DB.Hash("Test Hash")
+    # h["foo"] = "bar"
+    # print(h)
+    asyncio.get_event_loop().run_until_complete(cach_test())

app/services/location/jhu.py

@@ -57,6 +57,8 @@ async def get_category(category):
     category = category.lower()
     data_id = f"jhu.{category}"
 
+    # TODO: check cache
+
     # URL to request data from.
     url = BASE_URL + "time_series_covid19_%s_global.csv" % category
 
@@ -115,6 +117,8 @@ async def get_category(category):
         "source": "https://github.com/ExpDev07/coronavirus-tracker-api",
     }
     LOGGER.info(f"{data_id} results:\n{pf(results, depth=1)}")
+    # save the results to distributed cache
+    # TODO: async
     return results
 
 

pyproject.toml

@@ -1,5 +1,5 @@
 [tool.black]
-line-length = 120
+line-length = 100
 target-version = ['py36', 'py37', 'py38']
 include = '\.pyi?$'
 exclude = '''
@@ -23,4 +23,4 @@ multi_line_output = 3
 include_trailing_comma = "True"
 force_grid_wrap = 0
 use_parentheses = "True"
-line_length = 120
+line_length = 100

tasks.py

@@ -46,6 +46,7 @@ def check(ctx, fmt=False, sort=False, diff=False):  # pylint: disable=redefined-
         fmt_args.append("--diff")
         sort_args.append("--diff")
 
+    # FIXME: run each command and check return code
     cmd_args = []
     if fmt:
         cmd_args.extend(fmt_args)