PTV/v6d1tokens
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

request_token query_mode+tail

Browse Source
This commit is contained in:
AF 2021-12-22 19:58:41 +03:00
parent 59e9ba94ad
commit 8b525ee05b
2 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
v6d1tokens/client.py
View File

@ -3,21 +3,30 @@ from typing import Optional
import aiohttp import aiohttp
from v6d0auth import certs from v6d0auth import certs
from v6d0auth.client import mycert from v6d0auth.client import mycert, with_role
from v6d1tokens.config import taurl from v6d1tokens.config import taurl
__all__ = ('request_token',) __all__ = ('request_token',)
async def request_token(token_id: str, role: Optional[str] = None) -> str: async def request_token(query: str, query_mode: Optional[str] = None, tail: str = '') -> str:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
if role is None: if query_mode is None:
if tail:
raise ValueError('tail is used only for non-None query modes')
request = session.post( request = session.post(
f'{taurl}/get', f'{taurl}/get',
data=json.dumps([token_id, (await mycert()).hex()]).encode() data=json.dumps([query, (await mycert()).hex()]).encode()
) )
else: else:
if query_mode == 'token':
token_id, role = query, f'token:{query}::{tail}'
elif query_mode == 'role':
token_id, role = f'role:{query}::{tail}', query
else:
raise ValueError('unknown query mode')
await with_role(role)
request = session.post( request = session.post(
f'{taurl}/get', f'{taurl}/get',
data=json.dumps([token_id, certs.vkey.encode().hex()]).encode(), data=json.dumps([token_id, certs.vkey.encode().hex()]).encode(),

8
v6d1tokens/test-request.py
View File

@ -1,16 +1,12 @@
import asyncio import asyncio
from v6d0auth.client import with_role
from v6d1tokens.client import request_token from v6d1tokens.client import request_token
async def main(): async def main():
print(await request_token('test')) print(await request_token('test'))
await with_role('t3st') print(await request_token('t3st', 'role', '7est'))
await with_role('token:te5t::tes7') print(await request_token('te5t', 'token', 'tes7'))
print(await request_token('role:t3st::7est', 't3st'))
print(await request_token('te5t', 'token:te5t::tes7'))
if __name__ == '__main__': if __name__ == '__main__':