Skip to content

Commit

Permalink
Extract all info for execution
Browse files Browse the repository at this point in the history
  • Loading branch information
nivcertora committed Jan 2, 2025
1 parent 5b2f399 commit 59e6ddf
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions Quorum/apis/governance/aave_governance.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,11 @@ class BGDProposalData(BaseModel):
events: List[EventData] = Field(default_factory=list)


class PayloadAddresses(BaseModel):
chain: str
addresses: List[str]


# ==============================
# Mapping for Chains
# ==============================
Expand Down Expand Up @@ -119,3 +124,14 @@ def get_payload_addresses(self, chain_id: str, controller: str, payload_id: int)
payload_data = resp.json()
# We only need the 'target' field from each action
return [a['target'] for a in payload_data['payload']['actions']]

def get_all_payload_addresses(self, proposal_id: int) -> List[PayloadAddresses]:
"""
Retrieves a list of payload addresses for each chain in the proposal.
"""
data = self.get_proposal_data(proposal_id)
results = []
for p in data.proposal.payloads:
addresses = self.get_payload_addresses(p.chain, p.payloads_controller, p.payload_id)
results.append(PayloadAddresses(chain=p.chain, addresses=addresses))
return results

0 comments on commit 59e6ddf

Please sign in to comment.