vpc_user/user_api/call2user_api.py

57 lines
1.9 KiB
Python

from aiohttp import ClientSession
from json import dumps, loads
import asyncio
from db import read, write
global url_prefix
url_prefix = 'http://127.0.0.1:8010/'
# async def call(api_url, data, pre=True, fix=True):
# url = (url_prefix + api_url) if pre else api_url
# async with ClientSession() as session:
# async with session.post(url, json=data) as response:
# text = await response.text()
# try:
# json = loads(text)
# if 'detail' in json:
# return json['detail']
# else:
# return json
# except:
# if fix:
# return text.replace('"', '')
# else:
# return text
from call2api import call
#------------------------------------------------------------
# UserAPI
#------------------------------------------------------------
async def get_user_token_info(username, user_token):
data = {'username': username, 'user_token': user_token}
return await call('api/get_user_token_info/', data)
async def user_in_db(username, user_token):
data = {'username': username, 'user_token': user_token}
return await call('api/user_in_db/', data)
async def transfer_coins(username, user_token,
dst_username, amount):
data = {'username': username, 'user_token': user_token,
'dst_username': dst_username, 'amount': amount}
return await call('api/transfer_coins/', data)
#------------------------- END ------------------------------
#------------------------------------------------------------
# SystemAPI usage only
#------------------------------------------------------------
async def register_user_token(token, user_token):
data = {'token': token, 'user_token': user_token}
return await call('api/register_user_token/', data)
async def unregister_user_token(token, user_token):
data = {'token': token, 'user_token': user_token}
return await call('api/unregister_user_token/', data)
#------------------------- END ------------------------------