Added originated_contract_tzips field to OperationData (#350)

Author: droserasprout

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## [unreleased]
+
+### Added
+
+* tzkt: Added `originated_contract_tzips` field to `OperationData`.
+
 ## 5.1.1 - 2022-05-13
 
 ### Fixed

src/dipdup/datasources/tzkt/datasource.py

@@ -1059,6 +1059,7 @@ def convert_operation(cls, operation_json: Dict[str, Any], type_: Optional[str]
             originated_contract_address=originated_contract_json.get('address'),
             originated_contract_type_hash=originated_contract_json.get('typeHash'),
             originated_contract_code_hash=originated_contract_json.get('codeHash'),
+            originated_contract_tzips=originated_contract_json.get('tzips'),
             storage=operation_json.get('storage'),
             diffs=operation_json.get('diffs') or (),
             delegate_address=delegate_json.get('address'),

src/dipdup/hasura.py

@@ -377,7 +377,7 @@ async def _get_fields(self, name: str = 'query_root') -> List[Field]:
         for field_json in fields_json:
             # NOTE: Exclude autogenerated aggregate and pk fields
             ignore_postfixes = ('_aggregate', '_by_pk', 'Aggregate', 'ByPk')
-            if any(map(lambda postfix: field_json['name'].endswith(postfix), ignore_postfixes)):
+            if any(map(lambda postfix: field_json['name'].endswith(postfix), ignore_postfixes)):  # noqa: C417
                 continue
 
             # NOTE: Exclude relations. Not reliable enough but ok for now.
@@ -438,7 +438,7 @@ def _format_rest_query(self, name: str, table: str, filter: str, fields: Iterabl
             name = humps.camelize(name)
             filter = humps.camelize(filter)
             table = humps.camelize(table)
-            map(lambda f: f.camelize(), fields)
+            map(lambda f: f.camelize(), fields)  # noqa: C417
 
         try:
             filter_field = next(f for f in fields if f.name == filter)

src/dipdup/models.py

@@ -52,6 +52,7 @@ class OperationData:
     originated_contract_alias: Optional[str] = None
     originated_contract_type_hash: Optional[int] = None
     originated_contract_code_hash: Optional[int] = None
+    originated_contract_tzips: Optional[Tuple[str, ...]] = None
     delegate_address: Optional[str] = None
     delegate_alias: Optional[str] = None
 

tests/test_dipdup/test_datasources/test_tzkt.py

@@ -16,9 +16,9 @@
 
 
 @asynccontextmanager
-async def with_tzkt(batch_size: int) -> AsyncIterator[TzktDatasource]:
+async def with_tzkt(batch_size: int, url: str | None = None) -> AsyncIterator[TzktDatasource]:
     config = HTTPConfig(batch_size=batch_size)
-    datasource = TzktDatasource('https://api.tzkt.io', config)
+    datasource = TzktDatasource(url or 'https://api.tzkt.io', config)
     async with datasource:
         yield datasource
 
@@ -200,6 +200,12 @@ async def test_iter_migration_originations(self) -> None:
             self.assertEqual(67955553, originations[0].id)
             self.assertEqual(67955554, originations[1].id)
 
+    async def test_get_originations(self) -> None:
+        async with with_tzkt(1, 'https://tzkt-mainnet.dipdup.net') as tzkt:
+            originations = await tzkt.get_originations({'KT1PWx2mnDueood7fEmfbBDKx1D9BAnnXitn'}, 889027, 889027)
+            self.assertEqual(23812803, originations[0].id)
+            self.assertEqual(('fa12',), originations[0].originated_contract_tzips)
+
     async def test_on_operation_message_data(self) -> None:
         with open(join(dirname(__file__), '..', 'ftzfun.json')) as f:
             operations_json = json.load(f)