diff --git a/src/dexorder/database/column_types.py b/src/dexorder/database/column_types.py index 4367361..b06cd12 100644 --- a/src/dexorder/database/column_types.py +++ b/src/dexorder/database/column_types.py @@ -1,5 +1,6 @@ import dataclasses import math +from typing import Union from sqlalchemy import TypeDecorator, BIGINT from sqlalchemy.dialects.postgresql import BYTEA, JSONB @@ -24,8 +25,8 @@ class Blockchain(TypeDecorator): impl = BIGINT cache_ok = True - def process_bind_param(self, value: NativeBlockchain, dialect): - return value.id + def process_bind_param(self, value: Union[NativeBlockchain,int], dialect): + return value if type(value) is int else value.id def process_result_value(self, value: int, dialect): return NativeBlockchain.for_id(value)