support polygon ledger (#233)

credmark · Jun 11, 2023 · ed4ce2e · ed4ce2e
1 parent 8ddff26
commit ed4ce2e
Show file tree

Hide file tree

Showing 4 changed files with 117 additions and 4 deletions.
diff --git a/HISTORY.rst b/HISTORY.rst
@@ -2,6 +2,9 @@
 
 History
 -------
+0.8.116 [2023-06-11]
+* feat: support polygon ledger
+
 0.8.115 [2023-06-03]
 * fix: disable BSC ledger
 

diff --git a/credmark/cmf/types/ledger.py b/credmark/cmf/types/ledger.py
@@ -455,8 +455,8 @@ class ContractTable(LedgerTable):
     """"""
     BYTECODE = ColumnField('bytecode')
     """"""
-    FUNCTION_SIGHASHES = ColumnField('function_sighashes')
-    """"""
+    # FUNCTION_SIGHASHES = ColumnField('function_sighashes')
+    # """"""
     IS_ERC20 = ColumnField('is_erc20')
     """"""
     IS_ERC721 = ColumnField('is_erc721')

diff --git a/tests/test_ledger.py b/tests/test_ledger.py
@@ -45,6 +45,14 @@ def no_test_ledger_contract_events(self):
                                limit=10).to_dataframe()
                 self.assertEqual(df.shape[0], 10)
 
+    def test_ledger_contract_events(self):
+        context = credmark.cmf.model.ModelContext.current_context()
+        block_number = context.block_number
+        cc = Contract('0xC36442b4a4522E871399CD717aBDD847Ab11FE88')
+        with cc.ledger.events.Collect as q:
+            df = q.select(q.columns, where=q.BLOCK_NUMBER.between_(block_number-1000, block_number)).to_dataframe()
+            self.assertGreaterEqual(df.shape[0], 0)
+
     def test_transaction(self):
         with self.context.ledger.Transaction.as_('tx') as tx:
             df = tx.select([tx.BLOCK_NUMBER, tx.FROM_ADDRESS, tx.TO_ADDRESS, tx.VALUE],
@@ -117,7 +125,7 @@ def _test_not_between1():
                 self.assertTrue(6949017 not in df.block_number.unique())
                 self.assertTrue(13949017 not in df.block_number.unique())
 
-            print(f'Between {(13919017, 13949018)}')
+            # print(f'Between {(13919017, 13949018)}')
             df = oo.select(
                 where=oo.BLOCK_NUMBER.between_(13919017, 13949018),
                 group_by=[oo.HASH, oo.BLOCK_NUMBER],
@@ -127,7 +135,7 @@ def _test_not_between1():
             self.assertTrue(df.block_number.max() <= 13949017)
 
             # between with number with str_lower True/False: ineffective
-            print(f'Between {(6949017, 6949018)}')
+            # print(f'Between {(6949017, 6949018)}')
             df = oo.select(
                 where=oo.BLOCK_NUMBER.between_(
                     6949017, 6949018, case_sensitive=True),

diff --git a/tests/test_polygon.py b/tests/test_polygon.py
@@ -0,0 +1,102 @@
+# pylint: disable=line-too-long, pointless-string-statement
+
+import credmark.cmf.model
+from credmark.cmf.engine.model_unittest import ModelTestCase
+from credmark.cmf.types import BlockNumber, Contract
+
+"""
+with events
+"0x0297e37f1873d2dab4487aa67cd56b58e2f27875"
+"0x22126f95b1cde15182fe6a816765beb6a3096dc3"
+"0x775d4d8c4751bae18ebdba4ebaa46a1ff54d0c9d"
+"0x0bffc5692960bb043d3216839bdd6e5e64ff1b4e"
+"0xde1df189259dd9ea1686ed939bdfc03a97621423"
+"0x6286a9e6f7e745a6d884561d88f94542d6715698"
+"0x8b397084699cc64e429f610f81fac13bf061ef55"
+"0x3dec619dc529363767dee9e71d8dd1a5bc270d76"
+"0x371b97c779e8c5197426215225de0eeac7dd13af"
+"0x55b1a124c04a54eefdefe5fa2ef5f852fb5f2f26"
+"0x5b45b414c6cd2a3341be70ba22be786b0124003f"
+"0x5cf21a9899e679448284427c0baca807fad6bb8f"
+"0xe8af04ad759ad790aa5592f587d3cfb3ecc6a9da"
+"0x3b7335f3f1771122cd0107416b1da1b2fb7e94dd"
+"0x4e5e55baeef3bc747d22123ce4ade3661c916a3e"
+"0xf17a03a7f6db32441614ac0022e57d4e185530bb"
+"0x7c621229fb0293ef8a4f5caa79a8bb4d60bf5ca4"
+"0x58c1bbb508e96cfec1787acf6afe1c7008a5b064"
+"0x1c57a5ee9c5a90c9a5e31b5265175e0642b943b1"
+"0x33c6eec1723b12c46732f7ab41398de45641fa42"
+"0x0000000000004946c0e9f43f4dee607b0ef1fa1c"
+"""
+
+
+class TestPolygon(ModelTestCase):
+    def test_metadata(self):
+        context = credmark.cmf.model.ModelContext.current_context()
+        with context.fork(chain_id=137):
+            cc = Contract("0xefe9c015cff1e9bd1d7cc1782b13571ea77cf63b")
+            _ = cc.abi
+            self.assertTrue(cc.abi is not None)
+
+            # "0x104b9b1c41c6764e88edf1207f498902d840fe2a"	14951280	"0xcb9cc9d0f8f4ac451a523d3c064553dd33ea39e3"
+            # "0xb6f53428b2e9a7875844299b2d3f80bfd696678e"	35695159	"0x469f7cbcd9e3b82287bed461d2dc8965cc8e4e66"
+            cc = Contract("0xb6f53428b2e9a7875844299b2d3f80bfd696678e")
+            _ = cc.abi
+            # self.assertTrue(cc.proxy_for is not None)
+
+    def test_contract(self):
+        context = credmark.cmf.model.ModelContext.current_context()
+        with context.fork(chain_id=137):
+            block_number = BlockNumber.from_ymd(2022, 1, 1, 12, 3)
+            block_number_ts = block_number.timestamp
+            block_number2 = BlockNumber.from_timestamp(block_number_ts)
+            self.assertEqual(block_number, block_number2)
+
+    def test_ledger(self):
+        context = credmark.cmf.model.ModelContext.current_context()
+        with context.fork(chain_id=137, block_number=43464962) as context_poly:
+            context_poly.__dict__['original_input'] = {}
+            context_poly.__dict__['slug'] = 'test'
+
+            block_number = context_poly.block_number
+            with context_poly.ledger.Block as q:
+                df = q.select(q.columns, where=q.NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] == 1)
+
+            with context_poly.ledger.Contract as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.between_(block_number-1000, block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.Token as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.between_(block_number-1000, block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.Trace as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.Log as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(38683710)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.Transaction as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.Receipt as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.TokenTransfer as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            with context_poly.ledger.TokenBalance as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.eq(block_number)).to_dataframe()
+                self.assertTrue(df.shape[0] > 0)
+
+            cc = Contract('0xde1df189259dd9ea1686ed939bdfc03a97621423')
+
+            with cc.ledger.events.Transfer as q:
+                df = q.select(q.columns, where=q.BLOCK_NUMBER.between_(43451034, 43464962)).to_dataframe()
+                self.assertTrue(df.shape[0] == 2)