51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, JSON, Boolean
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from app.core._defaults import DEFAULT_ASSET_INITOBJ
|
|
from app.core.models.base import AlchemyBase
|
|
|
|
|
|
class Asset(AlchemyBase):
|
|
__tablename__ = 'assets'
|
|
|
|
id = Column(Integer, autoincrement=True, primary_key=True)
|
|
symbol = Column(String(32), nullable=False)
|
|
name = Column(String(256), nullable=False)
|
|
decimals = Column(Integer, nullable=False)
|
|
|
|
network = Column(String(32), nullable=True)
|
|
address = Column(String(1024), nullable=True)
|
|
meta = Column(JSON, nullable=False, default={})
|
|
rates = Column(JSON, nullable=False, default={})
|
|
|
|
created = Column(DateTime, nullable=False, default=0)
|
|
is_active = Column(Boolean, nullable=False, default=True)
|
|
|
|
balances = relationship('UserBalance', back_populates='asset')
|
|
internal_transactions = relationship('InternalTransaction', back_populates='asset')
|
|
|
|
@classmethod
|
|
def init_table(cls, engine):
|
|
AlchemyBase.metadata.create_all(engine)
|
|
|
|
@classmethod
|
|
def find(cls, session, **kwargs):
|
|
if 'symbol' in kwargs:
|
|
kwargs['symbol'] = kwargs['symbol'].upper()
|
|
|
|
result = session.query(cls).filter_by(**kwargs)
|
|
results_count = result.count()
|
|
if results_count == 0:
|
|
any_count = session.query(cls).count()
|
|
if any_count == 0:
|
|
init_asset = cls(**DEFAULT_ASSET_INITOBJ)
|
|
session.add(init_asset)
|
|
session.commit()
|
|
return cls.find(session, **kwargs)
|
|
|
|
raise Exception(f"Asset not found: {kwargs}")
|
|
elif results_count == 1:
|
|
return result.first()
|
|
else:
|
|
raise Exception(f"Multiple assets found: {results_count}")
|