Implemented collection.create_with_erc20 and erc721

atbe · atbe · commit 86acd01d6e20 · 2021-10-21T17:47:53.000-07:00
diff --git a/nftlabs/modules/collection.py b/nftlabs/modules/collection.py
@@ -90,11 +90,17 @@ def create_and_mint_batch(self, meta_with_supply: List[CreateCollectionArg]) ->
         token_ids = result[0]['args']['tokenIds']
         return [self.get(i) for i in token_ids]
 
-    def create_with_erc20(self, token_contract: str, token_amount: int, metadata):
-        pass
+    def create_with_erc20(self, token_contract: str, token_amount: int, metadata: CreateCollectionArg):
+        uri = self.get_storage().upload(metadata.metadata, self.address, self.get_signer_address())
+        self.execute_tx(self.__abi_module.wrap_erc20.build_transaction(
+            token_contract, token_amount, metadata.supply, uri, self.get_transact_opts()
+        ))
 
     def create_with_erc721(self, token_contract: str, token_id: int, metadata):
-        pass
+        uri = self.get_storage().upload(metadata.metadata, self.address, self.get_signer_address())
+        self.execute_tx(self.__abi_module.wrap_erc721.build_transaction(
+            token_contract, token_id, uri, self.get_transact_opts()
+        ))
 
     def mint(self, args):
         pass
