lnbits-legend/lnbits/extensions/tpos/models.py

from sqlite3 import Row
from typing import Optional

from fastapi import Query
from pydantic import BaseModel


class CreateTposData(BaseModel):
    name: str
    currency: str
    tip_options: str = Query(None)
    tip_wallet: str = Query(None)


class TPoS(BaseModel):
    id: str
    wallet: str
    name: str
    currency: str
    tip_options: Optional[str]
    tip_wallet: Optional[str]

    @classmethod
    def from_row(cls, row: Row) -> "TPoS":
        return cls(**dict(row))