feat(token_url): format token uri with quote_plus in pg (HemeraProtocol#177)

ideal93 · web-flow · commit c0ff9cada71f · 2024-10-08T13:16:47.000+08:00
diff --git a/common/models/erc1155_token_id_details.py b/common/models/erc1155_token_id_details.py
@@ -1,9 +1,11 @@
 from datetime import datetime
+from typing import Type
 
 from sqlalchemy import Column, Index, PrimaryKeyConstraint, desc, func, text
 from sqlalchemy.dialects.postgresql import BIGINT, BOOLEAN, BYTEA, JSONB, NUMERIC, TIMESTAMP, VARCHAR
 
 from common.models import HemeraModel, general_converter
+from common.models.erc721_token_id_details import token_uri_format_converter
 
 
 class ERC1155TokenIdDetails(HemeraModel):
@@ -31,7 +33,7 @@ def model_domain_mapping():
                 "domain": "ERC1155TokenIdDetail",
                 "conflict_do_update": False,
                 "update_strategy": None,
-                "converter": general_converter,
+                "converter": token_uri_format_converter,
             },
             {
                 "domain": "UpdateERC1155TokenIdDetail",
diff --git a/common/models/erc721_token_id_details.py b/common/models/erc721_token_id_details.py
@@ -1,11 +1,21 @@
 from datetime import datetime
+from typing import Type
+from urllib import parse
 
 from sqlalchemy import Column, Index, PrimaryKeyConstraint, desc, func, text
 from sqlalchemy.dialects.postgresql import BIGINT, BOOLEAN, BYTEA, JSONB, NUMERIC, TIMESTAMP, VARCHAR
 
 from common.models import HemeraModel, general_converter
 
 
+def token_uri_format_converter(table: Type[HemeraModel], data, is_update=False):
+
+    if data.token_uri is not None:
+        data.token_uri = parse.quote_plus(data.token_uri)
+
+    return general_converter(table, data, is_update)
+
+
 class ERC721TokenIdDetails(HemeraModel):
     __tablename__ = "erc721_token_id_details"
 
@@ -31,7 +41,7 @@ def model_domain_mapping():
                 "domain": "ERC721TokenIdDetail",
                 "conflict_do_update": False,
                 "update_strategy": None,
-                "converter": general_converter,
+                "converter": token_uri_format_converter,
             },
             {
                 "domain": "UpdateERC721TokenIdDetail",
