55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
# standard imports
|
|
from typing import Dict, Optional
|
|
|
|
# external imports
|
|
import json
|
|
|
|
from cic_types.condiments import MetadataPointer
|
|
|
|
# local imports
|
|
from .base import UssdMetadataHandler
|
|
from cic_ussd.cache import cache_data
|
|
from cic_ussd.error import MetadataNotFoundError
|
|
|
|
|
|
class TokenMetadata(UssdMetadataHandler):
|
|
def __init__(self, identifier: bytes, **kwargs):
|
|
super(TokenMetadata, self).__init__(identifier=identifier, **kwargs)
|
|
|
|
|
|
def token_metadata_handler(metadata_client: TokenMetadata) -> Optional[Dict]:
|
|
"""
|
|
:param metadata_client:
|
|
:type metadata_client:
|
|
:return:
|
|
:rtype:
|
|
"""
|
|
result = metadata_client.query()
|
|
token_metadata = result.json()
|
|
if not token_metadata:
|
|
raise MetadataNotFoundError(f'No metadata found at: {metadata_client.metadata_pointer} for: {metadata_client.identifier.decode("utf-8")}')
|
|
cache_data(metadata_client.metadata_pointer, json.dumps(token_metadata))
|
|
return token_metadata
|
|
|
|
|
|
def query_token_metadata(identifier: bytes):
|
|
"""
|
|
:param identifier:
|
|
:type identifier:
|
|
:return:
|
|
:rtype:
|
|
"""
|
|
token_metadata_client = TokenMetadata(identifier=identifier, cic_type=MetadataPointer.TOKEN_META_SYMBOL)
|
|
return token_metadata_handler(token_metadata_client)
|
|
|
|
|
|
def query_token_info(identifier: bytes):
|
|
"""
|
|
:param identifier:
|
|
:type identifier:
|
|
:return:
|
|
:rtype:
|
|
"""
|
|
token_info_client = TokenMetadata(identifier=identifier, cic_type=MetadataPointer.TOKEN_PROOF_SYMBOL)
|
|
return token_metadata_handler(token_info_client)
|