lint utils.py

This commit is contained in:
James Prestwich 2017-09-04 18:00:55 -06:00
parent 5198a2b765
commit 5fc8cf0d13
No known key found for this signature in database
GPG Key ID: 519E010A79028CCC
1 changed files with 34 additions and 9 deletions

View File

@ -1,80 +1,102 @@
import hashlib, json, random, binascii
import xcat.trades as trades
import hashlib
import json
import random
import binascii
import os
import xcat.trades as trades
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
############################################
########### Data conversion utils ##########
############################################
def b(string):
"""Convert a string to bytes"""
return str.encode(string)
def x(h):
"""Convert a hex string to bytes"""
return binascii.unhexlify(h.encode('utf8'))
def b2x(b):
"""Convert bytes to a hex string"""
return binascii.hexlify(b).decode('utf8')
def x2s(hexstring):
"""Convert hex to a utf-8 string"""
return binascii.unhexlify(hexstring).decode('utf-8')
def s2x(string):
"""Convert a utf-8 string to hex"""
return b2x(b(string))
def hex2dict(hexstr):
jsonstr = x2s(hexstr)
print(hexstr['fund_tx'])
print(jsonstr)
return json.loads(jsonstr)
def jsonformat(trade):
return {
'sell': trade.sell.__dict__,
'buy': trade.buyContract.__dict__
'sell': trade.sell.__dict__,
'buy': trade.buyContract.__dict__
}
############################################
########### Preimage utils #################
############################################
def generate_password():
s = "1234567890abcdefghijklmnopqrstuvwxyz01234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ"
s = ("1234567890abcdefghijklmnopqrstuvwxyz"
"01234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ")
passlen = 32
p = "".join(random.sample(s,passlen))
p = "".join(random.sample(s, passlen))
return p
def sha256(secret):
preimage = secret.encode('utf8')
h = hashlib.sha256(preimage).digest()
return h
############################################
######## Error handling for CLI ############
############################################
def throw(err):
print(err)
exit()
#############################################
######### xcat.json temp file #############
#############################################
tmp_dir = os.path.join(ROOT_DIR, '.tmp')
if not os.path.exists(tmp_dir):
os.makedirs(tmp_dir)
xcatjson = os.path.join(tmp_dir, 'xcat.json')
def save_trade(trade):
with open(xcatjson, 'w+') as outfile:
json.dump(trade, outfile)
def get_trade():
with open(xcatjson) as data_file:
xcatdb = json.load(data_file)
@ -83,6 +105,7 @@ def get_trade():
trade = trades.Trade(sell, buy, commitment=xcatdb['commitment'])
return trade
def erase_trade():
try:
with open(xcatjson, 'w') as outfile:
@ -90,15 +113,17 @@ def erase_trade():
except:
pass
def save(trade):
# print("Saving trade")
trade = {
'sell': trade.sell.__dict__,
'buy': trade.buy.__dict__,
'commitment': trade.commitment
'sell': trade.sell.__dict__,
'buy': trade.buy.__dict__,
'commitment': trade.commitment
}
save_trade(trade)
# Remove tmp files when trade is complete
def cleanup(tradeid):
try: