|
6 | 6 | from thirdweb.abi.i_token_erc1155 import ITokenERC1155 |
7 | 7 | from thirdweb.abi.i_token_erc20 import ITokenERC20 |
8 | 8 | from thirdweb.abi.i_token_erc721 import ITokenERC721 |
| 9 | +from thirdweb.abi.token_erc1155 import TokenERC1155 |
| 10 | +from thirdweb.abi.token_erc20 import TokenERC20 |
9 | 11 | from thirdweb.abi.token_erc721 import TokenERC721 |
10 | 12 | from thirdweb.common.error import NoSignerException |
11 | 13 | from thirdweb.common.feature_detection import matches_interface |
@@ -97,9 +99,9 @@ def __init__( |
97 | 99 | self.sales = self._detect_primary_sales() |
98 | 100 | self.platform_fee = self._detect_platform_fee() |
99 | 101 |
|
100 | | - self.token = self._detect_erc_20() |
101 | | - self.nft = self._detect_erc_721() |
102 | | - self.edition = self._detect_erc_1155() |
| 102 | + self.erc20 = self._detect_erc_20() |
| 103 | + self.erc721 = self._detect_erc_721() |
| 104 | + self.erc1155 = self._detect_erc_1155() |
103 | 105 |
|
104 | 106 | def call(self, fn: str, *args) -> Any: |
105 | 107 | func = cast(ContractFunction, getattr(self.functions, fn, None)) |
@@ -190,26 +192,26 @@ def _detect_platform_fee(self): |
190 | 192 | return None |
191 | 193 |
|
192 | 194 | def _detect_erc_20(self): |
193 | | - interface_to_match = self._get_interface_functions(ITokenERC20.abi()) |
| 195 | + interface_to_match = self._get_interface_functions(TokenERC20.abi()) |
194 | 196 |
|
195 | 197 | if matches_interface(self.functions, interface_to_match): |
196 | | - contract_wrapper = self._get_contract_wrapper(ITokenERC20) |
| 198 | + contract_wrapper = self._get_contract_wrapper(TokenERC20) |
197 | 199 | return ERC20(contract_wrapper, self._storage) |
198 | 200 | return None |
199 | 201 |
|
200 | 202 | def _detect_erc_721(self): |
201 | | - interface_to_match = self._get_interface_functions(ITokenERC721.abi()) |
| 203 | + interface_to_match = self._get_interface_functions(TokenERC721.abi()) |
202 | 204 |
|
203 | 205 | if matches_interface(self.functions, interface_to_match): |
204 | | - contract_wrapper = self._get_contract_wrapper(ITokenERC721) |
| 206 | + contract_wrapper = self._get_contract_wrapper(TokenERC721) |
205 | 207 | return ERC721(contract_wrapper, self._storage) |
206 | 208 | return None |
207 | 209 |
|
208 | 210 | def _detect_erc_1155(self): |
209 | | - interface_to_match = self._get_interface_functions(ITokenERC1155.abi()) |
| 211 | + interface_to_match = self._get_interface_functions(TokenERC1155.abi()) |
210 | 212 |
|
211 | 213 | if matches_interface(self.functions, interface_to_match): |
212 | | - contract_wrapper = self._get_contract_wrapper(ITokenERC1155) |
| 214 | + contract_wrapper = self._get_contract_wrapper(TokenERC1155) |
213 | 215 | return ERC1155(contract_wrapper, self._storage) |
214 | 216 | return None |
215 | 217 |
|
|
0 commit comments