Skip to content

Commit 5df26bc

Browse files
committed
Rename trusted_for_delegate
1 parent bd8d2b1 commit 5df26bc

File tree

5 files changed

+15
-20
lines changed

5 files changed

+15
-20
lines changed

app/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class Settings(BaseSettings):
4343
CONTRACT_MAX_DOWNLOAD_RETRIES: int = (
4444
90 # Task running once per day, means 3 months trying.
4545
)
46-
CONTRACTS_LOGO_BASE_URL: str = (
46+
CONTRACT_LOGO_BASE_URL: str = (
4747
"https://safe-transaction-assets.safe.global/contracts/logos"
4848
)
4949
CONTRACTS_TRUSTED_FOR_DELEGATE_CALL: list[str] = [

app/datasources/db/models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ class Contract(SqlQueryBase, TimeStampedSQLModel, table=True):
217217
name: str | None = None
218218
display_name: str | None = None
219219
description: str | None = None
220-
trusted_for_delegate: bool = Field(nullable=False, default=False)
220+
trusted_for_delegate_call: bool = Field(nullable=False, default=False)
221221
implementation: bytes | None = None
222222
fetch_retries: int = Field(nullable=False, default=0)
223223
abi_id: int | None = Field(nullable=True, default=None, foreign_key="abi.id")
@@ -352,15 +352,15 @@ async def update_contract_info(
352352
address: bytes,
353353
name: str,
354354
display_name: str,
355-
trusted_for_delegate: bool | None = False,
355+
trusted_for_delegate_call: bool | None = False,
356356
) -> int:
357357
"""
358358
Update the contract metadata for all the chains
359359
360360
:param address:
361361
:param name:
362362
:param display_name:
363-
:param trusted_for_delegate:
363+
:param trusted_for_delegate_call:
364364
:return: number of affected rows
365365
"""
366366
query = (
@@ -369,7 +369,7 @@ async def update_contract_info(
369369
.values(
370370
name=name,
371371
display_name=display_name,
372-
trusted_for_delegate=trusted_for_delegate,
372+
trusted_for_delegate_call=trusted_for_delegate_call,
373373
)
374374
)
375375
result = await db_session.execute(query)

app/routers/models.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from datetime import datetime
2-
from typing import Any, Union, cast
2+
from typing import Any, Union
33

44
from pydantic import Field, computed_field, field_validator, model_validator
55

@@ -51,19 +51,19 @@ def convert_bytes_to_hex(cls, abi_hash: bytes):
5151

5252

5353
class ContractsPublic(CamelModel):
54-
address: bytes | ChecksumAddress
54+
address: ChecksumAddress
5555
name: str
5656
display_name: str | None
5757
chain_id: int
5858
project: ProjectPublic | None
5959
abi: AbiPublic
6060
modified: datetime
61-
trusted_for_delegate: bool
61+
trusted_for_delegate_call: bool
6262

6363
class Config:
6464
from_attributes = True
6565

66-
@field_validator("address")
66+
@field_validator("address", mode="before")
6767
@classmethod
6868
def convert_to_checksum_address(cls, address: bytes):
6969
"""
@@ -78,12 +78,7 @@ def convert_to_checksum_address(cls, address: bytes):
7878

7979
@computed_field(return_type=str | None)
8080
def logo_url(self) -> str | None:
81-
if isinstance(self.address, bytes):
82-
addr_str = cast(str, to_0x_hex_str(self.address))
83-
else:
84-
addr_str = str(self.address)
85-
86-
return f"{settings.CONTRACTS_LOGO_BASE_URL}/{addr_str}.png"
81+
return f"{settings.CONTRACT_LOGO_BASE_URL}/{self.address}.png"
8782

8883

8984
class DataDecoderInput(CamelModel):

app/tests/datasources/db/test_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ async def test_update_contract_info(self):
314314
for contract in contracts_to_be_updated:
315315
self.assertIsNone(contract.name)
316316
self.assertIsNone(contract.display_name)
317-
self.assertFalse(contract.trusted_for_delegate)
317+
self.assertFalse(contract.trusted_for_delegate_call)
318318

319319
self.assertIsNone(other_contract.name)
320320
self.assertIsNone(other_contract.display_name)
@@ -323,7 +323,7 @@ async def test_update_contract_info(self):
323323
address=address_to_be_updated,
324324
name=contract_name,
325325
display_name=contract_display_name,
326-
trusted_for_delegate=True,
326+
trusted_for_delegate_call=True,
327327
)
328328
self.assertEqual(rows_affected, 5)
329329
# Other contract remains not updated
@@ -339,4 +339,4 @@ async def test_update_contract_info(self):
339339
)
340340
self.assertEqual(updated_contract.name, contract_name)
341341
self.assertEqual(updated_contract.display_name, contract_display_name)
342-
self.assertTrue(updated_contract.trusted_for_delegate)
342+
self.assertTrue(updated_contract.trusted_for_delegate_call)

app/tests/routers/test_contracts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,10 @@ async def test_view_contracts(self):
5454
self.assertEqual(results[0]["chainId"], 1)
5555
self.assertEqual(results[0]["project"], None)
5656
self.assertEqual(results[0]["modified"], datetime_to_str(contract.modified))
57-
self.assertFalse(results[0]["trustedForDelegate"])
57+
self.assertFalse(results[0]["trustedForDelegateCall"])
5858
self.assertEqual(
5959
results[0]["logoUrl"],
60-
f"{settings.CONTRACTS_LOGO_BASE_URL}/{address_expected}.png",
60+
f"{settings.CONTRACT_LOGO_BASE_URL}/{address_expected}.png",
6161
)
6262
# Test filter by chain_id
6363
contract = Contract(

0 commit comments

Comments
 (0)