first commit

Author: ngmisl

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "cSpell.words": ["csvfile", "standardabi"]
+}

main.py

@@ -0,0 +1,58 @@
+import sys
+import csv
+import os
+from web3 import Web3
+from standardabi import erc721_abi
+import polars as pl
+
+erc721_abi = erc721_abi
+
+# Make sure to add the correct RPC provider
+w3 = Web3(Web3.HTTPProvider("https://canto.slingshot.finance/"))
+
+
+def main():
+    if len(sys.argv) != 2:
+        print("Error: You can only enter the Contract Address")
+    else:
+        for arg in sys.argv[1:]:
+            token_owners = total_supply(arg)
+            snapshot_file = "snapshot.csv"
+            if not os.path.exists(snapshot_file):
+                write_to_csv(token_owners, snapshot_file)
+            create_clean_snapshot(snapshot_file, "snapshot_clean.csv")
+
+
+def total_supply(contract):
+    erc721_contract = w3.eth.contract(
+        address=w3.to_checksum_address(contract), abi=erc721_abi
+    )
+    total_supply = erc721_contract.functions.totalSupply().call()
+
+    token_owners = {}
+    for i in range(1, total_supply + 1):
+        owner = erc721_contract.functions.ownerOf(i).call()
+        token_owners[i] = owner
+
+    return token_owners
+
+
+def write_to_csv(token_owners, file_name):
+    with open(file_name, "w", newline="") as csvfile:
+        fieldnames = ["id", "wallet"]
+        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+
+        writer.writeheader()
+        for token_id, wallet in token_owners.items():
+            writer.writerow({"id": token_id, "wallet": wallet})
+
+
+def create_clean_snapshot(input_file, output_file):
+    if not os.path.exists(output_file):
+        df = pl.read_csv(input_file)
+        df_clean = df.groupby("wallet").agg(pl.col("id").alias("holdings").count())
+        df_clean.write_csv(output_file)
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,43 @@
+aiohttp==3.8.4
+aiosignal==1.3.1
+async-timeout==4.0.2
+attrs==22.2.0
+bitarray==2.7.3
+black==23.3.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+click==8.1.3
+cytoolz==0.12.1
+eth-abi==4.0.0
+eth-account==0.8.0
+eth-hash==0.5.1
+eth-keyfile==0.6.1
+eth-keys==0.4.0
+eth-rlp==0.3.0
+eth-typing==3.3.0
+eth-utils==2.1.0
+frozenlist==1.3.3
+hexbytes==0.3.0
+idna==3.4
+jsonschema==4.17.3
+lru-dict==1.1.8
+multidict==6.0.4
+mypy==1.2.0
+mypy-extensions==1.0.0
+packaging==23.1
+parsimonious==0.9.0
+pathspec==0.11.1
+platformdirs==3.2.0
+polars==0.17.2
+protobuf==4.22.3
+pycryptodome==3.17
+pyrsistent==0.19.3
+regex==2023.3.23
+requests==2.28.2
+rlp==3.0.0
+toolz==0.12.0
+typing_extensions==4.5.0
+urllib3==1.26.15
+web3==6.2.0
+websockets==11.0.1
+yarl==1.8.2

standardabi.py

@@ -0,0 +1,114 @@
+erc721_abi = [
+    {
+        "inputs": [{"internalType": "address", "name": "owner", "type": "address"}],
+        "name": "balanceOf",
+        "outputs": [{"internalType": "uint256", "name": "", "type": "uint256"}],
+        "payable": False,
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [],
+        "name": "name",
+        "outputs": [{"internalType": "string", "name": "", "type": "string"}],
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [{"internalType": "uint256", "name": "tokenId", "type": "uint256"}],
+        "name": "ownerOf",
+        "outputs": [{"internalType": "address", "name": "", "type": "address"}],
+        "payable": False,
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [{"internalType": "uint256", "name": "tokenId", "type": "uint256"}],
+        "name": "tokenURI",
+        "outputs": [{"internalType": "string", "name": "", "type": "string"}],
+        "payable": False,
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [{"internalType": "uint256", "name": "tokenId", "type": "uint256"}],
+        "name": "uri",
+        "outputs": [{"internalType": "string", "name": "", "type": "string"}],
+        "payable": False,
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [],
+        "name": "symbol",
+        "outputs": [{"internalType": "string", "name": "", "type": "string"}],
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [],
+        "name": "totalSupply",
+        "outputs": [{"internalType": "uint256", "name": "", "type": "uint256"}],
+        "stateMutability": "view",
+        "type": "function",
+        "constant": True,
+    },
+    {
+        "inputs": [
+            {"internalType": "address", "name": "owner", "type": "address"},
+            {"internalType": "uint256", "name": "index", "type": "uint256"},
+        ],
+        "name": "tokenOfOwnerByIndex",
+        "outputs": [{"internalType": "uint256", "name": "", "type": "uint256"}],
+        "stateMutability": "view",
+        "type": "function",
+    },
+    {
+        "inputs": [{"internalType": "address", "name": "_owner", "type": "address"}],
+        "name": "walletOfOwner",
+        "outputs": [{"internalType": "uint256[]", "name": "", "type": "uint256[]"}],
+        "stateMutability": "view",
+        "type": "function",
+    },
+    {
+        "inputs": [
+            {"internalType": "address", "name": "owner", "type": "address"},
+            {"internalType": "address", "name": "nft", "type": "address"},
+            {"internalType": "uint256", "name": "start", "type": "uint256"},
+            {"internalType": "uint256", "name": "end", "type": "uint256"},
+        ],
+        "name": "getOwnerNFTs",
+        "outputs": [{"internalType": "uint256[]", "name": "", "type": "uint256[]"}],
+        "stateMutability": "view",
+        "type": "function",
+    },
+    {
+        "inputs": [{"internalType": "address", "name": "addr", "type": "address"}],
+        "name": "checkAccess",
+        "outputs": [{"internalType": "bool", "name": "val", "type": "bool"}],
+        "stateMutability": "view",
+        "type": "function",
+    },
+    {
+        "inputs": [{"internalType": "uint256", "name": "_tokenId", "type": "uint256"}],
+        "name": "accessAddressPerTokenId",
+        "outputs": [
+            {"internalType": "address[]", "name": "addresses", "type": "address[]"}
+        ],
+        "stateMutability": "view",
+        "type": "function",
+    },
+    {
+        "inputs": [{"internalType": "uint256", "name": "", "type": "uint256"}],
+        "name": "balances",
+        "outputs": [{"internalType": "uint256", "name": "", "type": "uint256"}],
+        "stateMutability": "view",
+        "type": "function",
+    },
+]