diff --git a/api/.env.example b/api/.env.example index 041a74a3..66cfbc8d 100644 --- a/api/.env.example +++ b/api/.env.example @@ -5,4 +5,6 @@ AGENT_MNEMONIC= KAFKA_ENABLED=true METADATA_BASE_URL='https://metadata.drep.id/api' DB_SYNC_BASE_URL= -KAFKA_PREFIX= \ No newline at end of file +KAFKA_PREFIX= +#preview , mainnet , prepod ... defaults to mainnet +NETWORK='' \ No newline at end of file diff --git a/api/backend/app/services/agent_service.py b/api/backend/app/services/agent_service.py index d483659e..89689a0e 100644 --- a/api/backend/app/services/agent_service.py +++ b/api/backend/app/services/agent_service.py @@ -280,10 +280,14 @@ async def return_agent_with_wallet_details(self, agent: AgentResponse): async with aiohttp.ClientSession() as session: async with asyncio.TaskGroup() as group: agent_configurations = group.create_task(self.trigger_service.list_triggers_by_agent_id(agent.id)) - wallet_balance = group.create_task(self.fetch_balance(wallet.stake_key_hash, session)) - drep_details = group.create_task(self.fetch_drep_details(wallet.stake_key_hash, session)) + wallet_balance = group.create_task( + self.fetch_balance(convert_stake_key_hash_to_address(wallet.stake_key_hash), session) + ) + drep_details = group.create_task( + self.fetch_drep_details(convert_stake_key_hash_to_address(wallet.stake_key_hash), session) + ) delegation_details = group.create_task( - self.fetch_delegation_details(wallet.stake_key_hash, session) + self.fetch_delegation_details(convert_stake_key_hash_to_address(wallet.stake_key_hash), session) ) stake_address_details = group.create_task( self.fetch_stake_address_details(wallet.stake_key_hash, session) @@ -303,3 +307,13 @@ async def return_agent_with_wallet_details(self, agent: AgentResponse): is_stake_registered=stake_address_details.result().get("is_stake_registered"), stake_last_registered=stake_address_details.result().get("last_registered"), ) + + +def convert_stake_key_hash_to_address(stake_key_hash): + if stake_key_hash.startswith("e0") or stake_key_hash.startswith("e1"): + return stake_key_hash + else: + if api_settings.APP_ENV == "mainnet": + return "e1" + stake_key_hash + else: + return "e0" + stake_key_hash diff --git a/api/backend/config/application.py b/api/backend/config/application.py index 1fd1cba3..41836cc8 100644 --- a/api/backend/config/application.py +++ b/api/backend/config/application.py @@ -36,6 +36,7 @@ class Application(BaseSettings): KAFKA_BROKERS: str = "localhost:9092" KAFKA_ENABLED: bool = False KUBER_URL: str = "" + NETWORK: str = "mainnet" # All your additional application configuration should go either here or in # separate file in this submodule.