[lightapi/python] task: add NemConnector endpoints

lightapi/python/symbollightapi/connector/BasicConnector.py

@@ -15,15 +15,27 @@ def __init__(self, endpoint):
 		self.endpoint = endpoint
 		self.timeout_seconds = None
 
-	async def _dispatch(self, action, url_path, property_name, **kwargs):
+	async def _dispatch(self, action, url_path, property_name=None, response_type='json', **kwargs):
+		headers = kwargs.get('headers', {})
+
+		if response_type == 'binary':
+			headers['Accept'] = 'application/binary'
+
+		kwargs['headers'] = headers
+
 		try:
 			timeout = ClientTimeout(total=self.timeout_seconds)
 			async with ClientSession(timeout=timeout) as session:
 				async with getattr(session, action)(f'{self.endpoint}/{url_path}', **kwargs) as response:
-					try:
-						response_json = await response.json()
-					except (client_exceptions.ContentTypeError, json.decoder.JSONDecodeError) as ex:
-						raise NodeException from ex
+					if response_type == 'json':
+						try:
+							response_json = await response.json()
+						except (client_exceptions.ContentTypeError, json.decoder.JSONDecodeError) as ex:
+							raise NodeException from ex
+					elif response_type == 'binary':
+						response_json = await response.read()
+					else:
+						raise ValueError(f'Unsupported response type: {response_type}')
 
 					if response.status not in (200, 404):
 						error_message = f'HTTP request failed with code {response.status}'
@@ -33,7 +45,10 @@ async def _dispatch(self, action, url_path, property_name, **kwargs):
 
 						raise NodeException(error_message)
 
-					return response_json if property_name is None else response_json[property_name]
+					if response_type == 'json':
+						return response_json if property_name is None else response_json[property_name]
+					return response_json
+
 		except (asyncio.TimeoutError, client_exceptions.ClientConnectorError) as ex:
 			raise NodeException from ex
 
@@ -51,6 +66,12 @@ async def put(self, url_path, request_payload, property_name=None):
 		Raises NodeException on connection or content failure.
 		"""
 
-		return await self._dispatch('put', url_path, property_name, data=json.dumps(request_payload), headers={
-			'Content-Type': 'application/json'
-		})
+		return await self._dispatch('put', url_path, property_name, json=request_payload)
+
+	async def post(self, url_path, request_payload, property_name=None, response_type='json'):
+		"""
+		Initiates a POST to the specified path and returns the desired property.
+		Raises NodeException on connection or content failure.
+		"""
+
+		return await self._dispatch('post', url_path, property_name, json=request_payload, response_type=response_type)

lightapi/python/symbollightapi/connector/NemConnector.py

@@ -1,8 +1,12 @@
+import asyncio
+
 from symbolchain.CryptoTypes import PublicKey
 from symbolchain.nem.Network import Address
 
+from ..model.Block import Block
 from ..model.Endpoint import Endpoint
 from ..model.NodeInfo import NodeInfo
+from ..model.Transaction import TransactionFactory, TransactionMapperFactory
 from .BasicConnector import BasicConnector
 
 MICROXEM_PER_XEM = 1000000
@@ -91,3 +95,78 @@ def _map_to_node_info(node_dict):
 			node_dict['identity']['name'],
 			node_dict['metaData']['version'],
 			NodeInfo.API_ROLE_FLAG)
+
+	async def get_blocks_after(self, height):
+		""""Gets Blocks data"""
+
+		blocks = await self.post('local/chain/blocks-after', {'height': height})
+
+		block_heights = [block['block']['height'] for block in blocks['data']]
+		block_sizes = await asyncio.gather(*[self.get_block_size(height) for height in block_heights])
+
+		for block, size in zip(blocks['data'], block_sizes):
+			block['size'] = size
+
+		return [self._map_to_block(block) for block in blocks['data']]
+
+	async def get_block(self, height):
+		""""Gets Block data"""
+
+		block = await self.post('local/block/at', {'height': height})
+
+		block_sizes = await self.get_block_size(height)
+		block['size'] = block_sizes
+
+		return self._map_to_block(block)
+
+	async def get_block_size(self, height):
+		""""Gets Block size"""
+
+		block_size = await self.post('block/at/public', {'height': height}, response_type='binary')
+		return len(block_size)
+
+	def _map_to_block(self, block_dict):
+		block = block_dict['block']
+		difficulty = block_dict['difficulty']
+		block_hash = block_dict['hash']
+		transactions = block_dict['txes']
+		size = block_dict['size']
+
+		return Block(
+			block['height'],
+			block['timeStamp'],
+			[
+				self._map_to_transaction(transaction, block['height'])
+				for transaction in transactions
+			],
+			difficulty,
+			block_hash,
+			block['signer'],
+			block['signature'],
+			size
+		)
+
+	@staticmethod
+	def _map_to_transaction(transaction, block_height):
+		"""Maps a transaction dictionary to a transaction object."""
+
+		tx_dict = transaction['tx']
+		tx_type = tx_dict['type']
+
+		# Define common arguments for all transactions
+		common_args = {
+			'transaction_hash': transaction['hash'],
+			'height': block_height,
+			'sender': tx_dict['signer'],
+			'fee': tx_dict['fee'],
+			'timestamp': tx_dict['timeStamp'],
+			'deadline': tx_dict['deadline'],
+			'signature': tx_dict['signature'],
+		}
+
+		specific_args = {}
+
+		mapper = TransactionMapperFactory.get_mapper(tx_type)
+		specific_args = mapper.map_transaction(tx_dict)
+
+		return TransactionFactory.create_transaction(tx_type, common_args, specific_args)

lightapi/python/symbollightapi/model/Block.py

@@ -0,0 +1,28 @@
+class Block:
+	"""Block model."""
+
+	def __init__(self, height, timestamp, transactions, difficulty, block_hash, signer, signature, size):
+		"""Create a Block model."""
+
+		# pylint: disable=too-many-arguments
+
+		self.height = height
+		self.timestamp = timestamp
+		self.transactions = transactions
+		self.difficulty = difficulty
+		self.block_hash = block_hash
+		self.signer = signer
+		self.signature = signature
+		self.size = size
+
+	def __eq__(self, other):
+		return isinstance(other, Block) and all([
+			self.height == other.height,
+			self.timestamp == other.timestamp,
+			self.transactions == other.transactions,
+			self.difficulty == other.difficulty,
+			self.block_hash == other.block_hash,
+			self.signer == other.signer,
+			self.signature == other.signature,
+			self.size == other.size
+		])

lightapi/python/symbollightapi/model/Exceptions.py

@@ -4,3 +4,7 @@ class NodeException(Exception):
 
 class CorruptDataException(NodeException):
 	"""Exception raised when corrupt data is received from a node."""
+
+
+class UnknownTransactionType(Exception):
+	"""Exception raised when Unknown transaction type."""