Skip to content

Commit

Permalink
get chain by name:
Browse files Browse the repository at this point in the history
  • Loading branch information
@Prometheo
Prometheo committed Nov 5, 2024
1 parent 251c52e commit 2ff3dd7
Show file tree
Hide file tree
Showing 2 changed files with 44 additions and 34 deletions.
14 changes: 7 additions & 7 deletions indexer_app/handler.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -172,35 +172,35 @@ async def handle_streamer_message(streamer_message: near_primitives.StreamerMess
if event_name == "create_round" or event_name == "round_updated":
create_or_update_round_async = sync_to_async(create_or_update_round)
await create_or_update_round_async(
parsed_log.get("data")[0], receiver_id, now_datetime, 1
parsed_log.get("data")[0], receiver_id, now_datetime, "NEAR"
)
if event_name == "deposit":
round_depo = sync_to_async(process_rounds_deposit_event)
await round_depo(parsed_log.get("data")[0], receipt.receipt_id, 1)
await round_depo(parsed_log.get("data")[0], receipt.receipt_id, "NEAR")
if event_name == "application_created":
create_round_application_async = sync_to_async(create_round_application)
await create_round_application_async(
parsed_log.get("data")[0], receipt.receipt_id, 1
parsed_log.get("data")[0], receipt.receipt_id, "NEAR"
)
if event_name == "application_updated":
update_application_async = sync_to_async(update_application)
await update_application_async(
parsed_log.get("data")[0], receipt.receipt_id, signer_id, 1
parsed_log.get("data")[0], receipt.receipt_id, signer_id, "NEAR"
)
if event_name == "vote":
process_vote_event_async = sync_to_async(process_vote_event)
await process_vote_event_async(
parsed_log.get("data")[0], receipt.receipt_id, 1
parsed_log.get("data")[0], receipt.receipt_id, "NEAR"
)
if event_name == "payouts_set":
create_round_payout_aync = sync_to_async(create_round_payout)
await create_round_payout_aync(
parsed_log.get("data")[0], receipt.receipt_id, 1
parsed_log.get("data")[0], receipt.receipt_id, "NEAR"
)
if event_name == "payout_processed":
update_round_payout_aync = sync_to_async(update_round_payout)
await update_round_payout_aync(
parsed_log.get("data")[0], receipt.receipt_id, 1
parsed_log.get("data")[0], receipt.receipt_id, "NEAR"
)
if event_name == "payouts_challenge_created":
pass
Expand Down
64 changes: 37 additions & 27 deletions indexer_app/utils.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1345,12 +1345,13 @@ def get_ledger_sequence() -> int:
return record.block_height


def update_approved_projects(event_data, chain_id=2):
def update_approved_projects(event_data, chain_id="stellar"):
round_id, project_ids = event_data[0], event_data[1]

with transaction.atomic():
try:
round_obj = Round.objects.get(on_chain_id=round_id, chain_id=chain_id)
chain = Chain.objects.get(name=chain_id)
round_obj = Round.objects.get(on_chain_id=round_id, chain=chain)
for ids in project_ids:
project = Project.objects.get(id=ids)
round_obj.approved_projects.add(project.owner)
Expand All @@ -1360,7 +1361,7 @@ def update_approved_projects(event_data, chain_id=2):
logger.error(f"Error updating application for Round {round_id}: {e}")
return False

def update_application(event_data, txhash, reviewer_id=None, chain_id=2):
def update_application(event_data, txhash, reviewer_id=None, chain_id="stellar"):
if type(event_data) == list:
round_id, application_data, reviewer_id = event_data[0], event_data[1], event_data[2]
else:
Expand All @@ -1369,10 +1370,11 @@ def update_application(event_data, txhash, reviewer_id=None, chain_id=2):

with transaction.atomic():
try:
round_obj = Round.objects.get(on_chain_id=round_id, chain_id=chain_id)
applicant = Account.objects.get(id=application_data['applicant_id'], chain_id=chain_id)
chain = Chain.objects.get(name=chain_id)
round_obj = Round.objects.get(on_chain_id=round_id, chain=chain)
applicant = Account.objects.get(id=application_data['applicant_id'], chain=chain)

reviewer = Account.objects.get(id=reviewer_id, chain_id=chain_id)
reviewer = Account.objects.get(id=reviewer_id, chain=chain)

status = PotApplicationStatus[application_data['status'].upper()]
submitted_at = datetime.fromtimestamp(application_data['submited_ms'] / 1000)
Expand Down Expand Up @@ -1408,7 +1410,7 @@ def update_application(event_data, txhash, reviewer_id=None, chain_id=2):



def process_vote_event(event_data, tx_hash, chain_id=2):
def process_vote_event(event_data, tx_hash, chain_id="stellar"):
try:
with transaction.atomic():
if type(event_data) == list:
Expand All @@ -1417,8 +1419,9 @@ def process_vote_event(event_data, tx_hash, chain_id=2):
event_data = event_data['vote']
round_id, vote_data = event_data.get("round_id", 2), event_data

round_obj = Round.objects.get(on_chain_id=round_id, chain_id=chain_id)
voter, _ = Account.objects.get_or_create(id=vote_data['voter'], chain_id=chain_id)
chain = Chain.objects.get(name=chain_id)
round_obj = Round.objects.get(on_chain_id=round_id, chain=chain)
voter, _ = Account.objects.get_or_create(id=vote_data['voter'], chain=chain)
voted_at = datetime.fromtimestamp(vote_data['voted_ms'] / 1000)

# Create or update the Vote
Expand Down Expand Up @@ -1458,18 +1461,20 @@ def process_vote_event(event_data, tx_hash, chain_id=2):
return False


def process_project_event(event_data, chain_id=2):
def process_project_event(event_data, chain_id="stellar"):
try:
# Extract project data
project_data = event_data

logger.info(f"process_project_event: {project_data}, {chain_id}")

chain = Chain.objects.get(name=chain_id)

# Create or get the owner Account
owner, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=project_data['owner'])
owner, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=project_data['owner'])

# Create or get the payout Account
payout_address, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=project_data['payout_address'])
payout_address, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=project_data['payout_address'])

# Create the Project
project, created = Project.objects.update_or_create(
Expand Down Expand Up @@ -1541,22 +1546,23 @@ def process_project_event(event_data, chain_id=2):



def create_or_update_round(event_data, contract_id, timestamp, chain_id=2):
def create_or_update_round(event_data, contract_id, timestamp, chain_id="stellar"):
try:
logger.info(f"create_or_update_round: {event_data}, {contract_id}, {chain_id}")
# Create Round
# event_data = event_data.get('round_detail')
round_id = event_data.get('id')
owner_address = event_data.get('owner')
owner, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=owner_address)
factory_contract, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=contract_id)
chain = Chain.objects.get(name=chain_id)
owner, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=owner_address)
factory_contract, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=contract_id)
remaining_dist_address = event_data.get('remaining_dist_address', event_data.get('remaining_funds_redistribution_recipient'))
if remaining_dist_address:
remaining_dist_address_obj, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=remaining_dist_address)
remaining_dist_address_obj, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=remaining_dist_address)

remaining_dist_by = event_data.get('remaining_dist_by', event_data.get('remaining_funds_redistributed_by'))
if remaining_dist_by:
remaining_dist_by_obj, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=remaining_dist_by)
remaining_dist_by_obj, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=remaining_dist_by)

near_round_cp = event_data.get('round_complete')
if near_round_cp:
Expand Down Expand Up @@ -1653,16 +1659,17 @@ def process_application_to_round(event_data, tx_hash):



def create_round_application(event_data, tx_hash, chain_id=2):
def create_round_application(event_data, tx_hash, chain_id="stellar"):
try:
logger.info(f"create_round_application: {event_data}, {tx_hash}, {chain_id}")
if type(event_data) == list:
round_id, application_data = event_data[0], event_data[1]
else:
event_data = event_data['application']
round_id, application_data = event_data["round_id"], event_data
applicant, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=application_data["applicant_id"])
round_obj = Round.objects.get(on_chain_id=round_id, chain_id=chain_id)
chain = Chain.objects.get(name=chain_id)
applicant, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=application_data["applicant_id"])
round_obj = Round.objects.get(on_chain_id=round_id, chain=chain)
status = PotApplicationStatus[application_data['status'].upper()]
logger.info(f"Creating application for round: {round_id}")

Expand All @@ -1687,7 +1694,7 @@ def create_round_application(event_data, tx_hash, chain_id=2):
return False


def process_rounds_deposit_event(event_data, tx_hash, chain_id=2):
def process_rounds_deposit_event(event_data, tx_hash, chain_id="stellar"):
try:
logger.info(f"process_rounds_deposit_event: {event_data}, {tx_hash}, {chain_id}")
# Process deposit event
Expand All @@ -1696,9 +1703,10 @@ def process_rounds_deposit_event(event_data, tx_hash, chain_id=2):
else:
event_data = event_data['deposit']
round_id, deposit_data = event_data["round_id"], event_data
round_obj = Round.objects.get(on_chain_id=round_id, chain_id=chain_id)
chain = Chain.objects.get(name=chain_id)
round_obj = Round.objects.get(on_chain_id=round_id, chain=chain)
amount = deposit_data["total_amount"]
depositor, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=deposit_data["depositor_id"])
depositor, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=deposit_data["depositor_id"])

# Create or update a RoundDeposit object
deposit, created = RoundDeposit.objects.update_or_create(
Expand Down Expand Up @@ -1728,15 +1736,16 @@ def process_rounds_deposit_event(event_data, tx_hash, chain_id=2):



def create_round_payout(event_data, tx_hash, chain_id=2):
def create_round_payout(event_data, tx_hash, chain_id="stellar"):
try:
logger.info(f"create_round_payout: {event_data}, {tx_hash}, {chain_id}")
round_id, payout_data = event_data
amount = payout_data["amount"]
recipient_id = payout_data["recipient_id"]
memo = payout_data.get("memo")

stellar_token_acct, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id},id="stellar")
chain = Chain.objects.get(name=chain_id)
stellar_token_acct, _ = Account.objects.get_or_create(defaults={"chain":chain},id="stellar")
stellar_token, _ = Token.objects.get_or_create(
account=stellar_token_acct
)
Expand All @@ -1759,14 +1768,15 @@ def create_round_payout(event_data, tx_hash, chain_id=2):
return False


def update_round_payout(event_data, tx_hash, chain_id=2):
def update_round_payout(event_data, tx_hash, chain_id="stellar"):
try:
logger.info(f"update_round_payout: {event_data}, {tx_hash}, {chain_id}")
_, payout_data = event_data
payout_on_chain_id = payout_data["id"]
amount = payout_data["amount"]
paid_at_ms = payout_data.get("paid_at_ms")
recipient_id, _ = Account.objects.get_or_create(defaults={"chain_id":chain_id}, id=payout_data["recipient_id"])
chain = Chain.objects.get(name=chain_id)
recipient_id, _ = Account.objects.get_or_create(defaults={"chain":chain}, id=payout_data["recipient_id"])
memo = payout_data.get("memo")
payout = PotPayout.objects.get(on_chain_id=payout_on_chain_id)
payout.amount = amount
Expand Down

0 comments on commit 2ff3dd7

Please sign in to comment.