mirror of
https://github.com/Mailu/Mailu.git
synced 2025-05-19 22:23:16 +02:00
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
""" Table lookup backends for podop
|
|
"""
|
|
|
|
import aiohttp
|
|
import logging
|
|
from urllib.parse import quote
|
|
|
|
class UrlTable(object):
|
|
""" Resolve an entry by querying a parametrized GET URL.
|
|
"""
|
|
|
|
def __init__(self, url_pattern):
|
|
""" url_pattern must contain a format ``{}`` so the key is injected in
|
|
the url before the query, the ``§`` character will be replaced with
|
|
``{}`` for easier setup.
|
|
"""
|
|
self.url_pattern = url_pattern.replace('§', '{}')
|
|
|
|
async def get(self, key, ns=None):
|
|
""" Get the given key in the provided namespace
|
|
"""
|
|
logging.debug("Table get {}".format(key))
|
|
if ns is not None:
|
|
key += "/" + ns
|
|
async with aiohttp.ClientSession() as session:
|
|
quoted_key = quote(key)
|
|
async with session.get(self.url_pattern.format(quoted_key)) as request:
|
|
if request.status == 200:
|
|
result = await request.json()
|
|
logging.debug("Table get {} is {}".format(key, result))
|
|
return result
|
|
elif request.status == 404:
|
|
raise KeyError()
|
|
else:
|
|
raise Exception(request.status)
|
|
|
|
async def set(self, key, value, ns=None):
|
|
""" Set a value for the given key in the provided namespace
|
|
"""
|
|
logging.debug("Table set {} to {}".format(key, value))
|
|
if ns is not None:
|
|
key += "/" + ns
|
|
async with aiohttp.ClientSession() as session:
|
|
quoted_key = quote(key)
|
|
await session.post(self.url_pattern.format(quoted_key), json=value)
|
|
|
|
async def iter(self, cat):
|
|
""" Iterate the given key (experimental)
|
|
"""
|
|
logging.debug("Table iter {}".format(cat))
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(self.url_pattern.format(cat)) as request:
|
|
if request.status == 200:
|
|
result = await request.json()
|
|
return result
|