55import factory .fuzzy
66from eth_account import Account
77from ethereum .utils import checksum_encode , mk_contract_address
8+ from factory .django import DjangoModelFactory
89from hexbytes import HexBytes
910from web3 import Web3
1011
2021logger = getLogger (__name__ )
2122
2223
23- class SafeContractFactory (factory . DjangoModelFactory ):
24+ class SafeContractFactory (DjangoModelFactory ):
2425 class Meta :
2526 model = SafeContract
2627
2728 address = factory .LazyFunction (lambda : Account .create ().address )
2829 master_copy = factory .LazyFunction (lambda : Account .create ().address )
2930
3031
31- class SafeCreationFactory (factory . DjangoModelFactory ):
32+ class SafeCreationFactory (DjangoModelFactory ):
3233 class Meta :
3334 model = SafeCreation
3435
@@ -53,7 +54,7 @@ class Meta:
5354 signed_tx = factory .Sequence (lambda n : HexBytes ('%x' % (n + 5000 )))
5455
5556
56- class SafeCreation2Factory (factory . DjangoModelFactory ):
57+ class SafeCreation2Factory (DjangoModelFactory ):
5758 class Meta :
5859 model = SafeCreation2
5960
@@ -76,14 +77,14 @@ class Meta:
7677 block_number = None
7778
7879
79- class SafeFundingFactory (factory . DjangoModelFactory ):
80+ class SafeFundingFactory (DjangoModelFactory ):
8081 class Meta :
8182 model = SafeFunding
8283
8384 safe = factory .SubFactory (SafeContractFactory )
8485
8586
86- class EthereumBlockFactory (factory . DjangoModelFactory ):
87+ class EthereumBlockFactory (DjangoModelFactory ):
8788 class Meta :
8889 model = EthereumBlock
8990
@@ -94,7 +95,7 @@ class Meta:
9495 block_hash = factory .Sequence (lambda n : Web3 .keccak (text = 'block%d' % n ))
9596
9697
97- class EthereumTxFactory (factory . DjangoModelFactory ):
98+ class EthereumTxFactory (DjangoModelFactory ):
9899 class Meta :
99100 model = EthereumTx
100101
@@ -112,7 +113,7 @@ class Meta:
112113 value = factory .fuzzy .FuzzyInteger (0 , 1000 )
113114
114115
115- class SafeMultisigTxFactory (factory . DjangoModelFactory ):
116+ class SafeMultisigTxFactory (DjangoModelFactory ):
116117 class Meta :
117118 model = SafeMultisigTx
118119
@@ -131,14 +132,14 @@ class Meta:
131132 safe_tx_hash = factory .Sequence (lambda n : Web3 .keccak (text = 'safe_tx_hash%d' % n ))
132133
133134
134- class SafeTxStatusFactory (factory . DjangoModelFactory ):
135+ class SafeTxStatusFactory (DjangoModelFactory ):
135136 class Meta :
136137 model = SafeTxStatus
137138
138139 safe = factory .SubFactory (SafeContractFactory )
139140
140141
141- class EthereumEventFactory (factory . DjangoModelFactory ):
142+ class EthereumEventFactory (DjangoModelFactory ):
142143 class Meta :
143144 model = EthereumEvent
144145
@@ -158,7 +159,7 @@ class Params:
158159 )
159160
160161
161- class BannedSignerFactory (factory . DjangoModelFactory ):
162+ class BannedSignerFactory (DjangoModelFactory ):
162163 class Meta :
163164 model = BannedSigner
164165
0 commit comments