dexorder/backend
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

blockchain column handles int args

Browse Source
This commit is contained in:
Tim
2024-04-18 16:37:49 -04:00
parent e9779d3338
commit 84e194536a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/dexorder/database/column_types.py
View File

@@ -1,5 +1,6 @@
import dataclasses import dataclasses
import math import math
from typing import Union
from sqlalchemy import TypeDecorator, BIGINT from sqlalchemy import TypeDecorator, BIGINT
from sqlalchemy.dialects.postgresql import BYTEA, JSONB from sqlalchemy.dialects.postgresql import BYTEA, JSONB
@@ -24,8 +25,8 @@ class Blockchain(TypeDecorator):
impl = BIGINT impl = BIGINT
cache_ok = True cache_ok = True
def process_bind_param(self, value: NativeBlockchain, dialect): def process_bind_param(self, value: Union[NativeBlockchain,int], dialect):
return value.id return value if type(value) is int else value.id
def process_result_value(self, value: int, dialect): def process_result_value(self, value: int, dialect):
return NativeBlockchain.for_id(value) return NativeBlockchain.for_id(value)