2021-12-01 20:08:51 +00:00
|
|
|
from .db import yield_db_session, yield_db_session_ctx
|
|
|
|
from .models import (
|
|
|
|
EthereumBlock,
|
|
|
|
EthereumLabel,
|
|
|
|
EthereumTransaction,
|
|
|
|
PolygonBlock,
|
|
|
|
PolygonLabel,
|
|
|
|
PolygonTransaction,
|
2022-08-09 12:51:49 +00:00
|
|
|
MumbaiBlock,
|
|
|
|
MumbaiLabel,
|
|
|
|
MumbaiTransaction,
|
2022-05-26 13:23:12 +00:00
|
|
|
XDaiBlock,
|
|
|
|
XDaiLabel,
|
|
|
|
XDaiTransaction,
|
2021-12-01 20:08:51 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
from enum import Enum
|
|
|
|
|
|
|
|
from typing import Type, Union
|
|
|
|
|
|
|
|
|
|
|
|
class AvailableBlockchainType(Enum):
|
|
|
|
ETHEREUM = "ethereum"
|
|
|
|
POLYGON = "polygon"
|
2022-08-09 12:51:49 +00:00
|
|
|
MUMBAI = "mumbai"
|
2022-05-26 13:23:12 +00:00
|
|
|
XDAI = "xdai"
|
2021-12-01 20:08:51 +00:00
|
|
|
|
|
|
|
|
|
|
|
def get_block_model(
|
|
|
|
blockchain_type: AvailableBlockchainType,
|
2022-08-09 12:51:49 +00:00
|
|
|
) -> Type[Union[EthereumBlock, PolygonBlock, MumbaiBlock, XDaiBlock]]:
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-08-09 12:51:49 +00:00
|
|
|
Depends on provided blockchain type: Ethereum, Polygon, Mumbai or XDai,
|
|
|
|
set proper blocks model.
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-08-09 12:51:49 +00:00
|
|
|
block_model: Type[Union[EthereumBlock, PolygonBlock, MumbaiBlock, XDaiBlock]]
|
2021-12-01 20:08:51 +00:00
|
|
|
if blockchain_type == AvailableBlockchainType.ETHEREUM:
|
|
|
|
block_model = EthereumBlock
|
|
|
|
elif blockchain_type == AvailableBlockchainType.POLYGON:
|
|
|
|
block_model = PolygonBlock
|
2022-08-09 12:51:49 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.MUMBAI:
|
|
|
|
block_model = MumbaiBlock
|
2022-05-26 13:23:12 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.XDAI:
|
|
|
|
block_model = XDaiBlock
|
2021-12-01 20:08:51 +00:00
|
|
|
else:
|
|
|
|
raise Exception("Unsupported blockchain type provided")
|
|
|
|
|
|
|
|
return block_model
|
|
|
|
|
|
|
|
|
|
|
|
def get_label_model(
|
|
|
|
blockchain_type: AvailableBlockchainType,
|
2022-08-09 12:51:49 +00:00
|
|
|
) -> Type[Union[EthereumLabel, PolygonLabel, MumbaiLabel, XDaiLabel]]:
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-08-09 12:51:49 +00:00
|
|
|
Depends on provided blockchain type: Ethereum, Polygon, Mumbai or XDai,
|
|
|
|
set proper block label model.
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-08-09 12:51:49 +00:00
|
|
|
label_model: Type[Union[EthereumLabel, PolygonLabel, MumbaiLabel, XDaiLabel]]
|
2021-12-01 20:08:51 +00:00
|
|
|
if blockchain_type == AvailableBlockchainType.ETHEREUM:
|
|
|
|
label_model = EthereumLabel
|
|
|
|
elif blockchain_type == AvailableBlockchainType.POLYGON:
|
|
|
|
label_model = PolygonLabel
|
2022-08-09 12:51:49 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.MUMBAI:
|
|
|
|
label_model = MumbaiLabel
|
2022-05-26 13:23:12 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.XDAI:
|
|
|
|
label_model = XDaiLabel
|
2021-12-01 20:08:51 +00:00
|
|
|
else:
|
|
|
|
raise Exception("Unsupported blockchain type provided")
|
|
|
|
|
|
|
|
return label_model
|
|
|
|
|
|
|
|
|
|
|
|
def get_transaction_model(
|
|
|
|
blockchain_type: AvailableBlockchainType,
|
2022-08-09 12:51:49 +00:00
|
|
|
) -> Type[
|
|
|
|
Union[EthereumTransaction, PolygonTransaction, MumbaiTransaction, XDaiTransaction]
|
|
|
|
]:
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-08-09 12:51:49 +00:00
|
|
|
Depends on provided blockchain type: Ethereum, Polygon, Mumbai or XDai,
|
|
|
|
set proper block transactions model.
|
2021-12-01 20:08:51 +00:00
|
|
|
"""
|
2022-05-26 13:23:12 +00:00
|
|
|
transaction_model: Type[
|
2022-08-09 12:51:49 +00:00
|
|
|
Union[
|
|
|
|
EthereumTransaction, PolygonTransaction, MumbaiTransaction, XDaiTransaction
|
|
|
|
]
|
2022-05-26 13:23:12 +00:00
|
|
|
]
|
2021-12-01 20:08:51 +00:00
|
|
|
if blockchain_type == AvailableBlockchainType.ETHEREUM:
|
|
|
|
transaction_model = EthereumTransaction
|
|
|
|
elif blockchain_type == AvailableBlockchainType.POLYGON:
|
|
|
|
transaction_model = PolygonTransaction
|
2022-08-09 12:51:49 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.MUMBAI:
|
|
|
|
transaction_model = MumbaiTransaction
|
2022-05-26 13:23:12 +00:00
|
|
|
elif blockchain_type == AvailableBlockchainType.XDAI:
|
|
|
|
transaction_model = XDaiTransaction
|
2021-12-01 20:08:51 +00:00
|
|
|
else:
|
|
|
|
raise Exception("Unsupported blockchain type provided")
|
|
|
|
|
|
|
|
return transaction_model
|