kopia lustrzana https://github.com/bugout-dev/moonstream
Merge pull request #1089 from moonstream-to/types-networks-support
Models map for networkspull/1090/head types/python/v0.0.3
Sergei Sumarokov
GitHub
commit
a0c4e29994
|
|
@ -60,7 +60,6 @@ from moonstreamdb.models import (
|
|||
ZkSyncEraTestnetTransaction,
|
||||
ZkSyncEraTransaction,
|
||||
)
|
||||
|
||||
from moonstreamdbv3.models import (
|
||||
EthereumLabel as EthereumLabelV3,
|
||||
SepoliaLabel as SepoliaLabelV3,
|
||||
|
|
|
|||
|
|
@ -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}")
|
||||
|
|
@ -1 +1 @@
|
|||
0.0.2
|
||||
0.0.3
|
||||
|
|
|
|||
Ładowanie…
Reference in New Issue