mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-18 14:44:21 +01:00
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import requests
|
|
import re
|
|
import os
|
|
import json
|
|
from getopt import getopt
|
|
from sys import argv
|
|
import logging
|
|
from conf import CLIENT_LOG_FILE
|
|
from hashlib import sha256
|
|
from binascii import unhexlify
|
|
from apps.blob import Blob
|
|
from requests import ConnectTimeout
|
|
from apps import DEFAULT_PISA_API_SERVER, DEFAULT_PISA_API_PORT
|
|
|
|
|
|
commands = ['add_appointment']
|
|
|
|
|
|
def build_appointment(tx, tx_id, start_block, end_block, dispute_delta, debug, logging):
|
|
locator = sha256(unhexlify(tx_id)).hexdigest()
|
|
|
|
cipher = "AES-GCM-128"
|
|
hash_function = "SHA256"
|
|
|
|
# FIXME: The blob data should contain more things that just the transaction. Leaving like this for now.
|
|
blob = Blob(tx, cipher, hash_function)
|
|
|
|
# FIXME: tx_id should not be necessary (can be derived from tx SegWit-like). Passing it for now
|
|
encrypted_blob = blob.encrypt(tx_id, debug, logging)
|
|
|
|
appointment = {"locator": locator, "start_time": start_block, "end_time": end_block,
|
|
"dispute_delta": dispute_delta, "encrypted_blob": encrypted_blob, "cipher": cipher, "hash_function":
|
|
hash_function}
|
|
|
|
return appointment
|
|
|
|
|
|
def check_txid_format(txid):
|
|
if len(txid) != 64:
|
|
raise Exception("txid does not matches the expected size (32-byte / 64 hex chars).")
|
|
|
|
return re.search(r'^[0-9A-Fa-f]+$', txid) is not None
|
|
|
|
|
|
def show_usage():
|
|
print("usage: python pisa-cli.py argument [additional_arguments]."
|
|
"\nArguments:"
|
|
"\nadd_appointment appointment: \tregisters a json formatted appointment "
|
|
"\nhelp: \t\tshows this message.")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
debug = False
|
|
pisa_api_server = DEFAULT_PISA_API_SERVER
|
|
pisa_api_port = DEFAULT_PISA_API_PORT
|
|
command = None
|
|
|
|
opts, args = getopt(argv[1:], 'a:dh:p:', ['add_appointment, debug, host, port'])
|
|
for opt, arg in opts:
|
|
if opt in ['-a', '--add_appointment']:
|
|
if arg:
|
|
if not os.path.isfile(arg):
|
|
raise Exception("Can't find file " + arg)
|
|
else:
|
|
command = 'add_appointment'
|
|
json_file = arg
|
|
else:
|
|
raise Exception("Path to appointment_data.json missing.")
|
|
if opt in ['-d', '--debug']:
|
|
debug = True
|
|
|
|
if opt in ['-h', 'host']:
|
|
if arg:
|
|
pisa_api_server = arg
|
|
|
|
if opt in ['-p', '--port']:
|
|
if arg:
|
|
pisa_api_port = int(arg)
|
|
|
|
# Configure logging
|
|
logging.basicConfig(format='%(asctime)s %(message)s', level=logging.INFO, handlers=[
|
|
logging.FileHandler(CLIENT_LOG_FILE),
|
|
logging.StreamHandler()
|
|
])
|
|
|
|
if command in commands:
|
|
if command == 'add_appointment':
|
|
appointment_data = json.load(open(json_file))
|
|
valid_locator = check_txid_format(appointment_data.get('tx_id'))
|
|
|
|
if valid_locator:
|
|
pisa_url = "http://{}:{}".format(pisa_api_server, pisa_api_port)
|
|
appointment = build_appointment(appointment_data.get('tx'), appointment_data.get('tx_id'),
|
|
appointment_data.get('start_time'), appointment_data.get('end_time'),
|
|
appointment_data.get('dispute_delta'), debug, logging)
|
|
|
|
if debug:
|
|
logging.info("[Client] sending appointment to PISA")
|
|
|
|
try:
|
|
r = requests.post(url=pisa_url, json=json.dumps(appointment), timeout=5)
|
|
|
|
if debug:
|
|
logging.info("[Client] {} (code: {})".format(r.text, r.status_code))
|
|
|
|
except ConnectTimeout:
|
|
if debug:
|
|
logging.info("[Client] can't connect to pisa API. Connection timeout")
|
|
|
|
else:
|
|
raise ValueError("The provided locator is not valid.")
|
|
|
|
else:
|
|
show_usage()
|
|
|