Fix OperationData.amount, add delegate fields (#197)

Author: droserasprout

CHANGELOG.md

@@ -2,6 +2,16 @@
 
 Please use [this](https://docs.gitlab.com/ee/development/changelog.html) document as guidelines to keep a changelog.
 
+## [unreleased]
+
+## Added
+
+* tzkt: Added optional `delegate_address` and `delegate_alias` fields to `OperationData`.
+
+## Fixed
+
+* tzkt: Fixed incorrect parsing of `OperationData.amount` field.
+
 ## 4.0.1 - 2021-12-30
 
 ## Fixed

src/dipdup/datasources/tzkt/datasource.py

@@ -680,6 +680,7 @@ def convert_operation(cls, operation_json: Dict[str, Any], type_: Optional[str]
         sender_json = operation_json.get('sender') or {}
         target_json = operation_json.get('target') or {}
         initiator_json = operation_json.get('initiator') or {}
+        delegate_json = operation_json.get('delegate') or {}
         parameter_json = operation_json.get('parameter') or {}
         originated_contract_json = operation_json.get('originatedContract') or {}
 
@@ -703,7 +704,7 @@ def convert_operation(cls, operation_json: Dict[str, Any], type_: Optional[str]
             sender_address=sender_json.get('address'),
             target_address=target_json.get('address'),
             initiator_address=initiator_json.get('address'),
-            amount=operation_json.get('amount') or operation_json.get('contractBalance'),
+            amount=operation_json.get('amount', operation_json.get('contractBalance')),
             status=operation_json['status'],
             has_internals=operation_json.get('hasInternals'),
             sender_alias=operation_json['sender'].get('alias'),
@@ -717,6 +718,8 @@ def convert_operation(cls, operation_json: Dict[str, Any], type_: Optional[str]
             originated_contract_code_hash=originated_contract_json.get('codeHash'),
             storage=operation_json.get('storage'),
             diffs=operation_json.get('diffs') or (),
+            delegate_address=delegate_json.get('address'),
+            delegate_alias=delegate_json.get('alias'),
         )
 
     @classmethod

src/dipdup/datasources/tzkt/enums.py

@@ -23,6 +23,7 @@ class TzktMessageType(Enum):
     "status",
     "hasInternals",
     "diffs",
+    "delegate",
 )
 ORIGINATION_MIGRATION_FIELDS = (
     "id",

src/dipdup/models.py

@@ -52,6 +52,8 @@ class OperationData:
     originated_contract_alias: Optional[str] = None
     originated_contract_type_hash: Optional[int] = None
     originated_contract_code_hash: Optional[int] = None
+    delegate_address: Optional[str] = None
+    delegate_alias: Optional[str] = None
 
 
 @dataclass