78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import plyvel
|
|
import json
|
|
import xcat.utils as utils
|
|
from xcat.trades import Trade
|
|
|
|
|
|
class DB():
|
|
|
|
def __init__(self):
|
|
self.db = plyvel.DB('/tmp/xcatDB', create_if_missing=True)
|
|
self.preimageDB = plyvel.DB('/tmp/preimageDB', create_if_missing=True)
|
|
|
|
#############################################
|
|
######## Trades stored by tradeid ###########
|
|
#############################################
|
|
|
|
# Takes dict or obj, saves json str as bytes
|
|
def create(self, trade, tradeid):
|
|
if isinstance(trade, dict):
|
|
trade = json.dumps(trade, sort_keys=True, indent=4)
|
|
elif isinstance(trade, Trade):
|
|
trade = trade.toJSON()
|
|
else:
|
|
raise ValueError('Expected dictionary or Trade object')
|
|
self.db.put(utils.b(tradeid), utils.b(trade))
|
|
|
|
# Uses the funding txid as the key to save trade
|
|
def createByFundtx(self, trade):
|
|
if isinstance(trade, dict):
|
|
txid = trade['sell']['fund_tx']
|
|
trade = json.dumps(trade, sort_keys=True, indent=4)
|
|
elif isinstance(trade, Trade):
|
|
txid = trade.sell.fund_tx
|
|
trade = trade.toJSON()
|
|
else:
|
|
raise ValueError('Expected dictionary or Trade object')
|
|
self.db.put(utils.b(txid), utils.b(trade))
|
|
|
|
def get(self, tradeid):
|
|
rawtrade = self.db.get(utils.b(tradeid))
|
|
tradestr = str(rawtrade, 'utf-8')
|
|
trade = Trade(fromJSON=tradestr)
|
|
return trade
|
|
|
|
#############################################
|
|
###### Preimages stored by tradeid ##########
|
|
#############################################
|
|
|
|
# Stores secret locally in key/value store by tradeid
|
|
def save_secret(self, tradeid, secret):
|
|
self.preimageDB.put(utils.b(tradeid), utils.b(secret))
|
|
|
|
def get_secret(self, tradeid):
|
|
secret = self.preimageDB.get(utils.b(tradeid))
|
|
secret = str(secret, 'utf-8')
|
|
return secret
|
|
|
|
#############################################
|
|
########## Dump or view db entries ##########
|
|
#############################################
|
|
|
|
def dump(self):
|
|
results = []
|
|
with self.db.iterator() as it:
|
|
for k, v in it:
|
|
j = json.loads(str(v, 'utf-8'))
|
|
results.append((str(k, 'utf-8'), j))
|
|
return results
|
|
|
|
def print_entries(self):
|
|
it = self.db.iterator()
|
|
with self.db.iterator() as it:
|
|
for k, v in it:
|
|
j = json.loads(utils.x2s(utils.b2x(v)))
|
|
print("Key:", k)
|
|
print('val: ', j)
|
|
# print('sell: ', j['sell'])
|