# standard imports import os import json import logging import base64 # external imports from cic_types import MetadataPointer from cic_types.processor import generate_metadata_pointer from cic_types.ext.metadata import MetadataRequestsHandler from hexathon import strip_0x # local imports from .base import ( Data, data_dir, ) from cic.output import OutputWriter logg = logging.getLogger(__name__) class Meta(Data): def __init__(self, path='.', writer=None): super(Meta, self).__init__() self.name = None self.contact = {} self.path = path self.writer = writer self.meta_path = os.path.join(self.path, 'meta.json') def load(self): super(Meta, self).load() f = open(self.meta_path, 'r') o = json.load(f) f.close() self.name = o['name'] self.contact = o['contact'] self.inited = True def start(self): super(Meta, self).start() meta_template_file_path = os.path.join(data_dir, 'meta_template_v{}.json'.format(self.version())) f = open(meta_template_file_path) o = json.load(f) f.close() f = open(self.meta_path, 'w') json.dump(o, f, sort_keys=True, indent="\t") f.close() def reference(self, token_address): token_address_bytes = bytes.fromhex(strip_0x(token_address)) return generate_metadata_pointer(token_address_bytes, MetadataPointer.TOKEN_META) def asdict(self): return { 'name': self.name, 'contact': self.contact, } def process(self, token_address=None, token_symbol=None, writer=None): if writer == None: writer = self.writer v = json.dumps(self.asdict()) token_address_bytes = bytes.fromhex(strip_0x(token_address)) k = generate_metadata_pointer(token_address_bytes, MetadataPointer.TOKEN_META) writer.write(k, v.encode('utf-8')) token_symbol_bytes = token_symbol.encode('utf-8') k = generate_metadata_pointer(token_symbol_bytes, MetadataPointer.TOKEN_META_SYMBOL) writer.write(k, v.encode('utf-8')) return (k, v) def __str__(self): s = "contact.name = {}\n".format(self.name) for k in self.contact.keys(): if self.contact[k] == '': continue s += "contact.{} = {}\n".format(k.lower(), self.contact[k]) return s class MetadataWriter(OutputWriter): def write(self, k, v): rq = MetadataRequestsHandler(MetadataPointer.NONE, bytes.fromhex(k)) try: v = v.decode('utf-8') v = json.loads(v) logg.debug('metadatawriter bindecode {} {}'.format(k, v)) except UnicodeDecodeError: v = base64.b64encode(v).decode('utf-8') v = json.loads(json.dumps(v)) logg.debug('metadatawriter b64encode {} {}'.format(k, v)) r = rq.create(v) logg.info('metadata submitted at {}'.format(k)) return r