cic-cli/cic/proof.py

96 lines
2.1 KiB
Python
Raw Normal View History

2021-10-09 19:56:29 +02:00
# standard imports
import os
import json
2021-10-09 20:37:54 +02:00
import logging
2021-10-10 21:30:18 +02:00
import tempfile
import cbor2
# external imports
from hexathon import strip_0x
2021-10-09 19:56:29 +02:00
# local imports
from .base import *
2021-10-09 20:37:54 +02:00
logg = logging.getLogger(__name__)
2021-10-09 19:56:29 +02:00
class Proof(Data):
2021-10-10 21:30:18 +02:00
def __init__(self, path='.', attachments=None):
2021-10-09 19:56:29 +02:00
super(Proof, self).__init__()
self.namespace = 'ge'
self.description = None
self.path = path
2021-10-10 21:30:18 +02:00
self.extra_attachments = attachments
self.attachments = {}
2021-10-09 19:56:29 +02:00
self.proof_path = os.path.join(self.path, 'proof.json')
2021-10-10 21:30:18 +02:00
self.temp_proof_path = tempfile.mkstemp()[1]
2021-10-09 19:56:29 +02:00
def load(self):
2021-10-09 20:37:54 +02:00
super(Proof, self).load()
2021-10-09 19:56:29 +02:00
2021-10-09 20:37:54 +02:00
f = open(self.proof_path, 'r')
o = json.load(f)
2021-10-09 19:56:29 +02:00
f.close()
2021-10-09 20:37:54 +02:00
self.set_version(o['version'])
2021-10-09 19:56:29 +02:00
self.description = o['description']
self.namespace = o['namespace']
self.inited = True
def start(self):
super(Proof, self).start()
proof_template_file_path = os.path.join(data_dir, 'proof_template_v{}.json'.format(self.version()))
f = open(proof_template_file_path)
o = json.load(f)
f.close()
f = open(self.proof_path, 'w')
json.dump(o, f)
f.close()
2021-10-10 21:30:18 +02:00
def asdict(self):
return {
'version': self.version(),
'namespace': self.namespace,
'description': self.description,
}
def __get_ordered_hashes(self):
ks = list(self.attachments.keys())
ks.sort()
return ks
def get(self):
v = self.asdict()
b = cbor2.dumps(v)
f = open(self.temp_proof_path, 'wb')
f.write(b)
f.close()
hsh = self.hash(b).hex()
self.attachments[hsh] = self.temp_proof_path
if self.extra_attachments != None:
a = self.extra_attachments.asdict()
for k in a.keys():
self.attachments[k] = a[k]
hshs = self.__get_ordered_hashes()
return hshs
2021-10-09 20:37:54 +02:00
def __str__(self):
2021-10-10 14:49:22 +02:00
return "description = {}\n".format(self.description)