(wip) draft market module

ayshptk · ayshptk · commit 27e8ef7d905a · 2021-10-19T23:43:34.000+05:30
diff --git a/nftlabs/abi/market/__init__.py b/nftlabs/abi/market/__init__.py
diff --git a/nftlabs/modules/currency.py b/nftlabs/modules/currency.py
@@ -6,7 +6,7 @@
 from ..abi.erc20 import ERC20
 from web3 import Web3
 from typing import List, Dict
-
+ 
 
 class CurrencyModule(BaseModule):
     address: str
diff --git a/nftlabs/modules/market.py b/nftlabs/modules/market.py
@@ -0,0 +1,121 @@
+from web3 import Web3
+from typing import List, Dict
+from . import BaseModule
+from ..types import Role, Listing as ListingType
+from ..abi.market import Market
+#_types
+
+
+
+# from ..errors import NoSignerException
+# from ..abi.erc20 import ERC20
+
+
+class MarketModule(BaseModule):
+    address: str
+    __abi_module: Market
+    def __init__(self, client: Web3, address: str):
+        super().__init__()
+        self.address = address
+        self.__abi_module = Market(client, address)
+    
+    #todo: return types
+    def list_item(self, arg: ListArg):
+        if arg.properties is None:
+            arg.properties = {}
+        else:
+            final_properties = copy.copy(arg.properties)
+        tx = self.__abi_module.list.build_transaction(
+            arg.asset_contract, 
+            arg.token_id, 
+            arg.currency, 
+            arg.price_per_token,
+            arg.quantity, 
+            arg.tokens_per_buyer, 
+            arg.seconds_until_start, 
+            arg.seconds_until_end,
+            self.get_transact_opts()
+        )
+        self.execute_tx(tx)
+    
+    def unlist_item(self, listing_id, quantity):
+        tx = self.__abi_module.unlist.build_transaction(
+            listing_id, 
+            quantity,
+            self.get_transact_opts()
+            )
+        self.execute_tx(tx)
+    
+    def add_to_listing(self, listing_id, quantity):
+        tx = self.__abi_module.add_to_listing.build_transaction(
+            listing_id, 
+            quantity,
+            self.get_transact_opts()
+            )
+        self.execute_tx(tx)
+
+    def updateListingParams(self, args: ListUpdate):
+        tx = self.__abi_module.updateListingParams(
+            args.listing_id,
+            args.price_per_token,
+            args.currency,
+            args.tokens_per_buyer,
+            args.seconds_until_start,
+            args.seconds_until_end
+        )
+        self.execute_tx(tx)
+    
+    def buy(self, listing_id: int, quantity: int):
+        tx = self.__abi_module.buy.build_transaction(
+            listing_id, 
+            quantity, 
+            buyer,
+            self.get_transact_opts()
+            )
+        self.execute_tx(tx)
+
+    def set_market_fee_bps(self, amount: int):
+        tx = self.__abi_module.set_market_fee_bps.build_transaction(
+            amount, 
+            self.get_transact_opts())
+        self.execute_tx(tx)
+    
+    def get(self, listing_id) -> List:
+        self.__abi_module.get_listing.call(listing_id)
+    
+    def get_all(self) -> List[ListingType]:
+        self.__abi_module.get_all_listings.call()
+        
+    def get_all_by_seller(self, seller: str) -> List[ListingType]:
+        self.__abi_module.get_listings_by_seller.call(seller)
+    
+    def get_all_by_asset_contract(self, asset_contract: str) -> List[ListingType]:
+        self.__abi_module.get_listings_by_asset_contract.call(asset_contract)
+    
+    def get_all_by_asset(self, asset: str) -> List[ListingType]:
+        self.__abi_module.get_listings_by_asset.call(asset)
+
+
+    def grant_role(self, role: Role, address: str):
+        role_hash = role.get_hash()
+        tx = self.__abi_module.grant_role.build_transaction(
+            role_hash,
+            address,
+            self.get_transact_opts()
+        )
+
+        self.execute_tx(tx)
+
+    def revoke_role(self, role: Role, address: str):
+        role_hash = role.get_hash()
+        tx = self.__abi_module.revoke_role.build_transaction(
+            role_hash,
+            address,
+            self.get_transact_opts()
+        )
+        self.execute_tx(tx)
+
+    def total_supply(self) -> int:
+        self.__abi_module.total_listings.call()
+
+
diff --git a/nftlabs/modules/market_types.py b/nftlabs/modules/market_types.py
@@ -0,0 +1,23 @@
+from dataclasses import dataclass
+
+
+@dataclass.dataclass
+class ListArg:
+    asset_contract: str
+    token_id: int
+    currency: str
+    price_per_token: int
+    quantity: int
+    tokens_per_buyer: int
+    seconds_until_start: int
+    seconds_until_end: int
+
+
+@dataclass.dataclass
+class ListUpdate:
+    listing_id : int
+    price_per_token: int
+    currency: int
+    tokens_per_buyer: int
+    seconds_until_start: int
+    seconds_until_end: int
diff --git a/nftlabs/modules/nft.py b/nftlabs/modules/nft.py
@@ -10,7 +10,7 @@
 
 from ..types import NFT as NftType, Role
 from .nft_types import MintArg
-
+ 
 
 class NftModule(BaseModule):
     address: str
@@ -54,9 +54,6 @@ def total_supply(self) -> int:
         return self.__abi_module.total_supply.call()
 
     def get(self, nft_id: int) -> NftType:
-        return self.__get_metadata(nft_id)
-
-    def __get_metadata(self, nft_id: int) -> NftType:
         uri = self.__get_metadata_uri(nft_id)
         meta = self.get_storage().get(uri)
         meta_obj: NftType = NftType.from_json(meta)
@@ -107,6 +104,7 @@ def transfer_from(self, from_address: str, to_address: str, token_id: int):
     """
     Transfers NFT from the current signers wallet to another wallet
     """
+    
     def transfer(self, to_address: str, token_id: int):
         tx = self.__abi_module.safe_transfer_from1.build_transaction(
             self.get_signer_address(),
@@ -138,6 +136,10 @@ def get_owned(self, address: str = "") -> List[NftType]:
 
     def __token_of_owner_by_index(self, address: str, token_id: int) -> int:
         return self.__abi_module.token_of_owner_by_index.call(address, token_id)
+    
+    def get_creator(self, token_id: int) -> str:
+        return self.__abi_module.get_creator.call(token_id)
+
 
     """
     Returns balance of the current signers wallet
@@ -161,11 +163,13 @@ def set_approval(self, operator: str, approved: bool = True):
 
     def grant_role(self, role: Role, address: str):
         role_hash = role.get_hash()
-        self.execute_tx(self.__abi_module.grant_role.build_transaction(
-            role_hash, address, self.get_transact_opts()
-        ))
+        tx = self.__abi_module.grant_role.build_transaction(
+                role_hash, address, 
+                self.get_transact_opts()
+        )
+        self.execute_tx(tx)
 
-    def revoke_role(self, role, address: str):
+    def revoke_role(self, role: Role, address: str):
         role_hash = role.get_hash()
         self.execute_tx(self.__abi_module.revoke_role.build_transaction(
             role_hash, address, self.get_transact_opts()
diff --git a/nftlabs/types/listing.py b/nftlabs/types/listing.py
@@ -0,0 +1,15 @@
+from dataclasses import dataclass
+from typing import Optional, Union
+
+from dataclasses_json import dataclass_json
+
+
+@dataclass_json
+@dataclass
+class Listing:
+    name: str
+    description: str
+    image: str
+    properties: Optional[Union[str, dict]] = None
+    token_id: int
+    uri: str
diff --git a/nftlabs/types/role.py b/nftlabs/types/role.py
@@ -7,6 +7,7 @@ class Role(enum.Enum):
     minter = "MINTER_ROLE"
     transfer = "TRANSFER_ROLE"
     pauser = "PAUSER_ROLE"
+    lister = "LISTER_ROLE"
 
     def get_hash(self) -> bytes:
         if self.name == "admin":
