From 51aa5d3505f8fd7d82647cabce06efd9543c36c6 Mon Sep 17 00:00:00 2001 From: kompotkot Date: Wed, 12 Jun 2024 11:09:39 +0000 Subject: [PATCH] Models map for networks --- types/python/moonstreamtypes/blockchain.py | 1 - types/python/moonstreamtypes/networks.py | 375 +++++++++++++++++++++ types/python/moonstreamtypes/version.txt | 2 +- 3 files changed, 376 insertions(+), 2 deletions(-) create mode 100644 types/python/moonstreamtypes/networks.py diff --git a/types/python/moonstreamtypes/blockchain.py b/types/python/moonstreamtypes/blockchain.py index feb0cef2..e8abf3d0 100644 --- a/types/python/moonstreamtypes/blockchain.py +++ b/types/python/moonstreamtypes/blockchain.py @@ -60,7 +60,6 @@ ZkSyncEraTestnetTransaction, ZkSyncEraTransaction, ) - from moonstreamdbv3.models import ( EthereumLabel as EthereumLabelV3, SepoliaLabel as SepoliaLabelV3, diff --git a/types/python/moonstreamtypes/networks.py b/types/python/moonstreamtypes/networks.py new file mode 100644 index 00000000..9b4ebeba --- /dev/null +++ b/types/python/moonstreamtypes/networks.py @@ -0,0 +1,375 @@ +from enum import Enum +from typing import Dict, Union + +from moonstreamdb.models import ( + AmoyBlock, + AmoyLabel, + AmoyTransaction, + ArbitrumOneBlock, + ArbitrumOneLabel, + ArbitrumOneTransaction, + ArbitrumNovaBlock, + ArbitrumNovaLabel, + ArbitrumNovaTransaction, + ArbitrumSepoliaBlock, + ArbitrumSepoliaLabel, + ArbitrumSepoliaTransaction, + AvalancheBlock, + AvalancheFujiBlock, + AvalancheFujiLabel, + AvalancheFujiTransaction, + AvalancheLabel, + AvalancheTransaction, + Base, + BlastBlock, + BlastLabel, + BlastSepoliaBlock, + BlastSepoliaLabel, + BlastSepoliaTransaction, + BlastTransaction, + EthereumBlock, + EthereumLabel, + EthereumTransaction, + MumbaiBlock, + MumbaiLabel, + MumbaiTransaction, + PolygonBlock, + PolygonLabel, + PolygonTransaction, + ProofOfPlayApexBlock, + ProofOfPlayApexLabel, + ProofOfPlayApexTransaction, + WyrmBlock, + WyrmLabel, + WyrmTransaction, + XaiBlock, + XaiLabel, + XaiSepoliaBlock, + XaiSepoliaLabel, + XaiSepoliaTransaction, + XaiTransaction, + XDaiBlock, + XDaiLabel, + XDaiTransaction, + ZkSyncEraBlock, + ZkSyncEraLabel, + ZkSyncEraSepoliaBlock, + ZkSyncEraSepoliaLabel, + ZkSyncEraSepoliaTransaction, + ZkSyncEraTestnetBlock, + ZkSyncEraTestnetLabel, + ZkSyncEraTestnetTransaction, + ZkSyncEraTransaction, +) +from moonstreamdbv3.models import ( + EthereumLabel as EthereumLabelV3, + SepoliaLabel as SepoliaLabelV3, + PolygonLabel as PolygonLabelV3, + MumbaiLabel as MumbaiLabelV3, + AmoyLabel as AmoyLabelV3, + XDaiLabel as XDaiLabelV3, + ZkSyncEraLabel as ZkSyncEraLabelV3, + ZkSyncEraSepoliaLabel as ZkSyncEraSepoliaLabelV3, + BaseLabel as BaseLabelV3, + ArbitrumNovaLabel as ArbitrumNovaLabelV3, + ArbitrumOneLabel as ArbitrumOneLabelV3, + ArbitrumSepoliaLabel as ArbitrumSepoliaLabelV3, + Game7OrbitArbitrumSepoliaLabel as Game7OrbitArbitrumSepoliaLabelV3, + XaiLabel as XaiLabelV3, + XaiSepoliaLabel as XaiSepoliaLabelV3, + AvalancheLabel as AvalancheLabelV3, + AvalancheFujiLabel as AvalancheFujiLabelV3, + BlastLabel as BlastLabelV3, + BlastSepoliaLabel as BlastSepoliaLabelV3, + ProofOfPlayApexLabel as ProofOfPlayApexLabelV3, + StarknetLabel as StarknetLabelV3, + StarknetSepoliaLabel as StarknetSepoliaLabelV3, + MantleLabel as MantleLabelV3, + MantleSepoliaLabel as MantleSepoliaLabelV3, +) + +from .blockchain import AvailableBlockchainType + + +class Network(Enum): + ethereum = "ethereum" + sepolia = "sepolia" + polygon = "polygon" + mumbai = "mumbai" + amoy = "amoy" + xdai = "xdai" + wyrm = "wyrm" + zksync_era_testnet = "zksync_era_testnet" + zksync_era = "zksync_era" + zksync_era_sepolia = "zksync_era_sepolia" + base = "base" + arbitrum_one = "arbitrum_one" + arbitrum_nova = "arbitrum_nova" + arbitrum_sepolia = "arbitrum_sepolia" + game7_orbit_arbitrum_sepolia = "game7_orbit_arbitrum_sepolia" + xai = "xai" + xai_sepolia = "xai_sepolia" + avalanche = "avalanche" + avalanche_fuji = "avalanche_fuji" + blast = "blast" + blast_sepolia = "blast_sepolia" + proofofplay_apex = "proofofplay_apex" + starknet = "starknet" + starknet_sepolia = "starknet_sepolia" + mantle = "mantle" + mantle_sepolia = "mantle_sepolia" + + +tx_raw_types = Union[ + EthereumTransaction, + MumbaiTransaction, + AmoyTransaction, + PolygonTransaction, + WyrmTransaction, + XDaiTransaction, + ZkSyncEraTestnetTransaction, + ZkSyncEraTransaction, + ZkSyncEraSepoliaTransaction, + ArbitrumOneTransaction, + ArbitrumNovaTransaction, + ArbitrumSepoliaTransaction, + XaiTransaction, + XaiSepoliaTransaction, + AvalancheTransaction, + AvalancheFujiTransaction, + BlastTransaction, + BlastSepoliaTransaction, + ProofOfPlayApexTransaction, +] + +MODELS: Dict[Network, Dict[str, Base]] = { + Network.ethereum: { + "blocks": EthereumBlock, + "labels": EthereumLabel, + "transactions": EthereumTransaction, + }, + Network.mumbai: { + "blocks": MumbaiBlock, + "labels": MumbaiLabel, + "transactions": MumbaiTransaction, + }, + Network.amoy: { + "blocks": AmoyBlock, + "labels": AmoyLabel, + "transactions": AmoyTransaction, + }, + Network.polygon: { + "blocks": PolygonBlock, + "labels": PolygonLabel, + "transactions": PolygonTransaction, + }, + Network.xdai: { + "blocks": XDaiBlock, + "labels": XDaiLabel, + "transactions": XDaiTransaction, + }, + Network.wyrm: { + "blocks": WyrmBlock, + "labels": WyrmLabel, + "transactions": WyrmTransaction, + }, + Network.zksync_era_testnet: { + "blocks": ZkSyncEraTestnetBlock, + "labels": ZkSyncEraTestnetLabel, + "transactions": ZkSyncEraTestnetTransaction, + }, + Network.zksync_era: { + "blocks": ZkSyncEraSepoliaBlock, + "labels": ZkSyncEraSepoliaLabel, + "transactions": ZkSyncEraSepoliaTransaction, + }, + Network.zksync_era_sepolia: { + "blocks": ZkSyncEraBlock, + "labels": ZkSyncEraLabel, + "transactions": ZkSyncEraTransaction, + }, + Network.arbitrum_one: { + "blocks": ArbitrumOneBlock, + "labels": ArbitrumOneLabel, + "transactions": ArbitrumOneTransaction, + }, + Network.arbitrum_nova: { + "blocks": ArbitrumNovaBlock, + "labels": ArbitrumNovaLabel, + "transactions": ArbitrumNovaTransaction, + }, + Network.arbitrum_sepolia: { + "blocks": ArbitrumSepoliaBlock, + "labels": ArbitrumSepoliaLabel, + "transactions": ArbitrumSepoliaTransaction, + }, + Network.xai: { + "blocks": XaiBlock, + "labels": XaiLabel, + "transactions": XaiTransaction, + }, + Network.xai_sepolia: { + "blocks": XaiSepoliaBlock, + "labels": XaiSepoliaLabel, + "transactions": XaiSepoliaTransaction, + }, + Network.avalanche: { + "blocks": AvalancheBlock, + "labels": AvalancheLabel, + "transactions": AvalancheTransaction, + }, + Network.avalanche_fuji: { + "blocks": AvalancheFujiBlock, + "labels": AvalancheFujiLabel, + "transactions": AvalancheFujiTransaction, + }, + Network.blast: { + "blocks": BlastBlock, + "labels": BlastLabel, + "transactions": BlastTransaction, + }, + Network.blast_sepolia: { + "blocks": BlastSepoliaBlock, + "labels": BlastSepoliaLabel, + "transactions": BlastSepoliaTransaction, + }, + Network.proofofplay_apex: { + "blocks": ProofOfPlayApexBlock, + "labels": ProofOfPlayApexLabel, + "transactions": ProofOfPlayApexTransaction, + }, +} + +MODELS_V3: Dict[Network, Dict[str, Base]] = { + Network.ethereum: { + "labels": EthereumLabelV3, + }, + Network.sepolia: { + "labels": SepoliaLabelV3, + }, + Network.polygon: { + "labels": PolygonLabelV3, + }, + Network.mumbai: { + "labels": MumbaiLabelV3, + }, + Network.amoy: { + "labels": AmoyLabelV3, + }, + Network.xdai: { + "labels": XDaiLabelV3, + }, + Network.zksync_era: { + "labels": ZkSyncEraLabelV3, + }, + Network.zksync_era_sepolia: { + "labels": ZkSyncEraSepoliaLabelV3, + }, + Network.base: { + "labels": BaseLabelV3, + }, + Network.arbitrum_nova: { + "labels": ArbitrumNovaLabelV3, + }, + Network.arbitrum_sepolia: { + "labels": ArbitrumSepoliaLabelV3, + }, + Network.arbitrum_one: { + "labels": ArbitrumOneLabelV3, + }, + Network.game7_orbit_arbitrum_sepolia: { + "labels": Game7OrbitArbitrumSepoliaLabelV3, + }, + Network.xai: { + "labels": XaiLabelV3, + }, + Network.xai_sepolia: { + "labels": XaiSepoliaLabelV3, + }, + Network.avalanche: { + "labels": AvalancheLabelV3, + }, + Network.avalanche_fuji: { + "labels": AvalancheFujiLabelV3, + }, + Network.blast: { + "labels": BlastLabelV3, + }, + Network.blast_sepolia: { + "labels": BlastSepoliaLabelV3, + }, + Network.proofofplay_apex: { + "labels": ProofOfPlayApexLabelV3, + }, + Network.starknet: { + "labels": StarknetLabelV3, + }, + Network.starknet_sepolia: { + "labels": StarknetSepoliaLabelV3, + }, + Network.mantle: { + "labels": MantleLabelV3, + }, + Network.mantle_sepolia: { + "labels": MantleSepoliaLabelV3, + }, +} + + +def blockchain_type_to_network_type( + blockchain_type: AvailableBlockchainType, +) -> Network: + if blockchain_type == AvailableBlockchainType.ETHEREUM: + return Network.ethereum + elif blockchain_type == AvailableBlockchainType.SEPOLIA: + return Network.sepolia + elif blockchain_type == AvailableBlockchainType.POLYGON: + return Network.polygon + elif blockchain_type == AvailableBlockchainType.MUMBAI: + return Network.mumbai + elif blockchain_type == AvailableBlockchainType.AMOY: + return Network.amoy + elif blockchain_type == AvailableBlockchainType.XDAI: + return Network.xdai + elif blockchain_type == AvailableBlockchainType.WYRM: + return Network.wyrm + elif blockchain_type == AvailableBlockchainType.ZKSYNC_ERA_TESTNET: + return Network.zksync_era_testnet + elif blockchain_type == AvailableBlockchainType.ZKSYNC_ERA: + return Network.zksync_era + elif blockchain_type == AvailableBlockchainType.ZKSYNC_ERA_SEPOLIA: + return Network.zksync_era_sepolia + elif blockchain_type == AvailableBlockchainType.BASE: + return Network.base + elif blockchain_type == AvailableBlockchainType.ARBITRUM_ONE: + return Network.arbitrum_one + elif blockchain_type == AvailableBlockchainType.ARBITRUM_NOVA: + return Network.arbitrum_nova + elif blockchain_type == AvailableBlockchainType.ARBITRUM_SEPOLIA: + return Network.arbitrum_sepolia + elif blockchain_type == AvailableBlockchainType.GAME7_ORBIT_ARBITRUM_SEPOLIA: + return Network.game7_orbit_arbitrum_sepolia + elif blockchain_type == AvailableBlockchainType.XAI: + return Network.xai + elif blockchain_type == AvailableBlockchainType.XAI_SEPOLIA: + return Network.xai_sepolia + elif blockchain_type == AvailableBlockchainType.AVALANCHE: + return Network.avalanche + elif blockchain_type == AvailableBlockchainType.AVALANCHE_FUJI: + return Network.avalanche_fuji + elif blockchain_type == AvailableBlockchainType.BLAST: + return Network.blast + elif blockchain_type == AvailableBlockchainType.BLAST_SEPOLIA: + return Network.blast_sepolia + elif blockchain_type == AvailableBlockchainType.PROOFOFPLAY_APEX: + return Network.proofofplay_apex + elif blockchain_type == AvailableBlockchainType.STARKNET: + return Network.starknet + elif blockchain_type == AvailableBlockchainType.STARKNET_SEPOLIA: + return Network.starknet_sepolia + elif blockchain_type == AvailableBlockchainType.MANTLE: + return Network.mantle + elif blockchain_type == AvailableBlockchainType.MANTLE_SEPOLIA: + return Network.mantle_sepolia + else: + raise ValueError(f"Unknown blockchain type: {blockchain_type}") diff --git a/types/python/moonstreamtypes/version.txt b/types/python/moonstreamtypes/version.txt index 4e379d2b..bcab45af 100644 --- a/types/python/moonstreamtypes/version.txt +++ b/types/python/moonstreamtypes/version.txt @@ -1 +1 @@ -0.0.2 +0.0.3