def request(self, api, post_data=''): if is_offline(): offline_tx = {} offline_tx['api-type'] = 'POST' if post_data else 'GET' offline_tx['api-endpoint'] = api offline_tx['api-data'] = post_data return offline_tx headers = {} url = self.node_host + api if self.api_key: headers['api_key'] = self.api_key header_str = ' '.join( ['--header \'{}: {}\''.format(k, v) for k, v in headers.items()]) try: if post_data: headers['Content-Type'] = 'application/json' data_str = '-d {}'.format(post_data) logging.info("curl -X POST %s %s %s" % (header_str, data_str, url)) return requests.post(url, data=post_data, headers=headers).json() else: logging.info("curl -X GET %s %s" % (header_str, url)) return requests.get(url, headers=headers).json() except RequestException as ex: msg = 'Failed to get response: {}'.format(ex) raise NetworkException(msg)
def check_contend(self, slot_id, tx_fee): if tx_fee < DEFAULT_CONTEND_SLOT_FEE: msg = 'Transaction fee must be >= %d' % DEFAULT_CONTEND_SLOT_FEE pytvspos.throw_error(msg, InvalidParameterException) return False if slot_id >= 60 or slot_id < 0: msg = 'Slot id must be in 0 to 59' pytvspos.throw_error(msg, InvalidParameterException) return False if is_offline(): # if offline, skip other check return True balance_detail = self.get_info() min_effective_balance = MIN_CONTEND_SLOT_BALANCE + tx_fee if balance_detail["effective"] < min_effective_balance: msg = 'Insufficient TV balance. (The effective balance must be >= %d)' % min_effective_balance pytvspos.throw_error(msg, InvalidParameterException) return False slot_info = self.chain.slot_info(slot_id) if not slot_info or slot_info.get("mintingAverageBalance") is None: msg = 'Failed to get slot minting average balance' pytvspos.throw_error(msg, NetworkException) return False elif slot_info["mintingAverageBalance"] >= balance_detail[ "mintingAverage"]: msg = 'The minting average balance of slot %d is greater than or equals to yours. ' \ 'You will contend this slot failed.' % slot_id pytvspos.throw_error(msg, InsufficientBalanceException) return False return True
def height(self): if is_offline(): pytvspos.throw_error("Cannot check height in offline mode.", NetworkException) return 0 else: return self.api_wrapper.request('/blocks/height')['height']
def dbput(self, db_key, db_data, db_data_type="ByteArray", tx_fee=DEFAULT_DBPUT_FEE, fee_scale=DEFAULT_FEE_SCALE, timestamp=0): if not self.privateKey: msg = 'Private key required' pytvspos.throw_error(msg, MissingPrivateKeyException) elif tx_fee < DEFAULT_DBPUT_FEE: msg = 'Transaction fee must be >= %d' % DEFAULT_DBPUT_FEE pytvspos.throw_error(msg, InvalidParameterException) elif len(db_key) > MAX_DB_KEY_SIZE or len(db_key) < MIN_DB_KEY_SIZE: msg = 'DB key length must be greater than %d and smaller than %d' % ( MIN_DB_KEY_SIZE, MAX_ATTACHMENT_SIZE) pytvspos.throw_error(msg, InvalidParameterException) elif CHECK_FEE_SCALE and fee_scale != DEFAULT_FEE_SCALE: msg = 'Wrong fee scale (currently, fee scale must be %d).' % DEFAULT_FEE_SCALE pytvspos.throw_error(msg, InvalidParameterException) elif not is_offline() and self.balance() < tx_fee: msg = 'Insufficient TV balance' pytvspos.throw_error(msg, InsufficientBalanceException) else: if timestamp == 0: timestamp = int(time.time() * 1000000000) # "ByteArray" is the only supported type in first version if db_data_type == "ByteArray": data_type_id = b'\x01' # TODO: add more DB data type here else: msg = 'Unsupported data type: {}'.format(db_data_type) pytvspos.throw_error(msg, InvalidParameterException) return sData = struct.pack(">B", DBPUT_TX_TYPE) + \ struct.pack(">H", len(db_key)) + \ str2bytes(db_key) + \ struct.pack(">H", len(db_data)+1) + \ data_type_id + \ str2bytes(db_data) + \ struct.pack(">Q", tx_fee) + \ struct.pack(">H", fee_scale) + \ struct.pack(">Q", timestamp) signature = bytes2str(sign(self.privateKey, sData)) data = json.dumps({ "senderPublicKey": self.publicKey, "dbKey": db_key, "dataType": db_data_type, "data": db_data, "fee": tx_fee, "feeScale": fee_scale, "timestamp": timestamp, "signature": signature }) return self.wrapper.request('/database/broadcast/put', data)
def get_connected_peers(self): if is_offline(): pytvspos.throw_error("Cannot check peers in offline mode.", NetworkException) return [] response = self.api_wrapper.request('/peers/connected') if not response.get("peers"): return [] else: return [peer["address"] for peer in response.get("peers")]
def check_node(self, other_node_host=None): if is_offline(): pytvspos.throw_error("Cannot check node in offline mode.", NetworkException) return False if other_node_host: res = self.chain.check_with_other_node(other_node_host) else: res = self.chain.self_check() # add more check if need return res
def __str__(self): if not self.address: raise InvalidAddressException("No address") result = 'address = %s\npublicKey = %s\nprivateKey = %s\nseed = %s\nnonce = %d' % \ (self.address, self.publicKey, self.privateKey, self.seed, self.nonce) if not is_offline(): try: balance = self.balance() result += "\nbalance: {}".format(balance) except NetworkException: logging.error("Failed to get balance") return result
def send_payment(self, recipient, amount, attachment='', tx_fee=DEFAULT_PAYMENT_FEE, fee_scale=DEFAULT_FEE_SCALE, timestamp=0): if not self.privateKey: msg = 'Private key required' pytvspos.throw_error(msg, MissingPrivateKeyException) if not self.chain.validate_address(recipient.address): msg = 'Invalid recipient address' pytvspos.throw_error(msg, InvalidAddressException) elif amount <= 0: msg = 'Amount must be > 0' pytvspos.throw_error(msg, InvalidParameterException) elif tx_fee < DEFAULT_PAYMENT_FEE: msg = 'Transaction fee must be >= %d' % DEFAULT_PAYMENT_FEE pytvspos.throw_error(msg, InvalidParameterException) elif len(attachment) > MAX_ATTACHMENT_SIZE: msg = 'Attachment length must be <= %d' % MAX_ATTACHMENT_SIZE pytvspos.throw_error(msg, InvalidParameterException) elif CHECK_FEE_SCALE and fee_scale != DEFAULT_FEE_SCALE: msg = 'Wrong fee scale (currently, fee scale must be %d).' % DEFAULT_FEE_SCALE pytvspos.throw_error(msg, InvalidParameterException) elif not is_offline() and self.balance() < amount + tx_fee: msg = 'Insufficient TV balance' pytvspos.throw_error(msg, InsufficientBalanceException) else: if timestamp == 0: timestamp = int(time.time() * 1000000000) sData = struct.pack(">B", PAYMENT_TX_TYPE) + \ struct.pack(">Q", timestamp) + \ struct.pack(">Q", amount) + \ struct.pack(">Q", tx_fee) + \ struct.pack(">H", fee_scale) + \ base58.b58decode(recipient.address) + \ struct.pack(">H", len(attachment)) + \ str2bytes(attachment) signature = bytes2str(sign(self.privateKey, sData)) attachment_str = bytes2str(base58.b58encode(str2bytes(attachment))) data = json.dumps({ "senderPublicKey": self.publicKey, "recipient": recipient.address, "amount": amount, "fee": tx_fee, "feeScale": fee_scale, "timestamp": timestamp, "attachment": attachment_str, "signature": signature }) return self.wrapper.request('/vsys/broadcast/payment', data)
def balance(self, confirmations=0): if is_offline(): pytvspos.throw_error("Cannot check balance in offline mode.", NetworkException) return 0 try: confirmations_str = '' if confirmations == 0 else '/%d' % confirmations resp = self.wrapper.request('/addresses/balance/%s%s' % (self.address, confirmations_str)) logging.debug(resp) return resp['balance'] except Exception as ex: msg = "Failed to get balance. ({})".format(ex) pytvspos.throw_error(msg, NetworkException) return 0
def fetch_minting_average_balance(self): if is_offline(): pytvspos.throw_error("Cannot fetch minting_average_balance in offline mode.", NetworkException) return 0 try: resp = self.wrapper.request('/addresses/balance/details/%s' % self.address) logging.debug(resp) #print(resp) self.minting_average_balance = resp['mintingAverage'] self.available_balance = resp['available'] #print(self.minting_average_balance) return None except Exception as ex: msg = "Failed to get minting average balance. ({})".format(ex) pytvspos.throw_error(msg, NetworkException) return 0
def get_tx_history(self, limit=100, type_filter=PAYMENT_TX_TYPE): if is_offline(): pytvspos.throw_error("Cannot check history in offline mode.", NetworkException) return [] if not self.address: msg = 'Address required' pytvspos.throw_error(msg, MissingAddressException) elif limit > MAX_TX_HISTORY_LIMIT: msg = 'Too big sequences requested (Max limitation is %d).' % MAX_TX_HISTORY_LIMIT pytvspos.throw_error(msg, InvalidParameterException) else: url = '/transactions/address/{}/limit/{}'.format( self.address, limit) resp = self.wrapper.request(url) if isinstance(resp, list) and type_filter: resp = [tx for tx in resp[0] if tx['type'] == type_filter] return resp
def get_info(self): if not (self.address and self.publicKey): msg = 'Address required' pytvspos.throw_error(msg, MissingAddressException) return None if not self.publicKey: msg = 'Public key and address required' pytvspos.throw_error(msg, MissingPublicKeyException) return None if is_offline(): info = {"publicKey": self.publicKey, "address": self.address} return info info = self.balance_detail() if not info: msg = 'Failed to get balance detail' pytvspos.throw_error(msg, NetworkException) else: info["publicKey"] = self.publicKey return info
def check_with_other_node(self, node_host, super_node_num=DEFAULT_SUPER_NODE_NUM): if is_offline(): pytvspos.throw_error("Cannot check height in offline mode.", NetworkException) return False try: h1 = self.height() except NetworkException: logging.error("Fail to connect {}.".format(node_host)) return False try: other_api = Wrapper(node_host) h2 = other_api.request('/blocks/height')['height'] except NetworkException: logging.error("Fail to connect {}.".format(node_host)) return False # Add more check if need return h2 - h1 <= super_node_num
def lease_cancel(self, lease_id, tx_fee=DEFAULT_CANCEL_LEASE_FEE, fee_scale=DEFAULT_FEE_SCALE, timestamp=0): decode_lease_id = base58.b58decode(lease_id) if not self.privateKey: msg = 'Private key required' pytvspos.throw_error(msg, MissingPrivateKeyException) elif tx_fee < DEFAULT_CANCEL_LEASE_FEE: msg = 'Transaction fee must be > %d' % DEFAULT_CANCEL_LEASE_FEE pytvspos.throw_error(msg, InvalidParameterException) elif len(decode_lease_id) != LEASE_TX_ID_BYTES: msg = 'Invalid lease transaction id' pytvspos.throw_error(msg, InvalidParameterException) elif CHECK_FEE_SCALE and fee_scale != DEFAULT_FEE_SCALE: msg = 'Wrong fee scale (currently, fee scale must be %d).' % DEFAULT_FEE_SCALE pytvspos.throw_error(msg, InvalidParameterException) elif not is_offline() and self.balance() < tx_fee: msg = 'Insufficient TV balance' pytvspos.throw_error(msg, InsufficientBalanceException) else: if timestamp == 0: timestamp = int(time.time() * 1000000000) sData = struct.pack(">B", LEASE_CANCEL_TX_TYPE) + \ struct.pack(">Q", tx_fee) + \ struct.pack(">H", fee_scale) + \ struct.pack(">Q", timestamp) + \ decode_lease_id signature = bytes2str(sign(self.privateKey, sData)) data = json.dumps({ "senderPublicKey": self.publicKey, "txId": lease_id, "fee": tx_fee, "feeScale": fee_scale, "timestamp": timestamp, "signature": signature }) req = self.wrapper.request('/leasing/broadcast/cancel', data) return req
def release(self, slot_id, tx_fee=DEFAULT_RELEASE_SLOT_FEE, fee_scale=DEFAULT_FEE_SCALE, timestamp=0): if not self.privateKey: msg = 'Private key required' pytvspos.throw_error(msg, MissingPrivateKeyException) elif tx_fee < DEFAULT_RELEASE_SLOT_FEE: msg = 'Transaction fee must be >= %d' % DEFAULT_RELEASE_SLOT_FEE pytvspos.throw_error(msg, InvalidParameterException) elif slot_id >= 60 or slot_id < 0: msg = 'Slot id must be in 0 to 59' pytvspos.throw_error(msg, InvalidParameterException) elif CHECK_FEE_SCALE and fee_scale != DEFAULT_FEE_SCALE: msg = 'Wrong fee scale (currently, fee scale must be %d).' % DEFAULT_FEE_SCALE pytvspos.throw_error(msg, InvalidParameterException) elif not is_offline() and self.balance() < tx_fee: msg = 'Insufficient TV balance' pytvspos.throw_error(msg, InsufficientBalanceException) else: if timestamp == 0: timestamp = int(time.time() * 1000000000) sData = struct.pack(">B", RELEASE_SLOT_TX_TYPE) + \ struct.pack(">I", slot_id) + \ struct.pack(">Q", tx_fee) + \ struct.pack(">H", fee_scale) + \ struct.pack(">Q", timestamp) signature = bytes2str(sign(self.privateKey, sData)) data = json.dumps({ "senderPublicKey": self.publicKey, "fee": tx_fee, "feeScale": fee_scale, "slotId": slot_id, "timestamp": timestamp, "signature": signature }) return self.wrapper.request('/spos/broadcast/release', data)
def fetch_leases_to_pay(self, last_height): if is_offline(): pytvspos.throw_error("Cannot fetch leases_to_pay in offline mode.", NetworkException) try: resp = self.wrapper.request('/transactions/activeLeaseList/%s' % self.address) logging.debug(resp) #print(resp[0]) except Exception as ex: msg = "Failed to get activeLeaseList. ({})".format(ex) pytvspos.throw_error(msg, NetworkException) self.leases_to_pay = [] self.leases_total = 0 for leaseidx in range(len(resp[0])): lease = resp[0][leaseidx] if lease['recipient']==self.address and lease['type']==3 and lease['height']<last_height: lease_to_pay = {} address = self.chain.public_key_to_address(base58.b58decode(lease['proofs'][0]['publicKey'])) lease_to_pay['lease_id'] = lease['id'] lease_to_pay['address'] = address lease_to_pay['lease_height'] = lease['height'] lease_to_pay['amount'] = lease['amount'] self.leases_to_pay.append(lease_to_pay) self.leases_total += lease_to_pay['amount']
def check_tx(self, tx_id, confirmations=0): """Confirm tx on chain. Return True if Transaction is fully confirmed. Return False if Transaction is sent but not confirmed or failed. Return None if Transaction does not exist! """ if is_offline(): pytvspos.throw_error("Cannot check transaction in offline mode.", NetworkException) return None utx_res = self.chain.unconfirmed_tx(tx_id) if "id" in utx_res: logging.error( "Transaction {} is pending in UTX pool.".format(tx_id)) return False else: tx_res = self.chain.tx(tx_id) if tx_res.get("status") == "Success": tx_height = tx_res["height"] cur_height = self.chain.height() if cur_height >= tx_height + confirmations: logging.debug( "Transaction {} is fully confirmed.".format(tx_id)) return True else: logging.info( "Transaction {} is sent but not fully confirmed.". format(tx_id)) return False elif "id" not in tx_res: logging.error("Transaction does not exist!") logging.debug("Tx API response: {}".format(tx_res)) return None else: logging.error("Transaction failed to process!") logging.debug("Tx API response: {}".format(tx_res)) return False