47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from __future__ import annotations
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import (
|
|
Integer,
|
|
String,
|
|
)
|
|
from sqlalchemy.orm import (
|
|
Mapped,
|
|
mapped_column,
|
|
relationship,
|
|
)
|
|
|
|
from .Base import Base
|
|
if TYPE_CHECKING:
|
|
from .UserProducts import UserProducts
|
|
from .Transaction import Transaction
|
|
|
|
class User(Base):
|
|
__tablename__ = 'users'
|
|
name: Mapped[str] = mapped_column(String(10), primary_key=True)
|
|
credit: Mapped[str] = mapped_column(Integer)
|
|
card: Mapped[str | None] = mapped_column(String(20))
|
|
rfid: Mapped[str | None] = mapped_column(String(20))
|
|
|
|
products: Mapped[set[UserProducts]] = relationship(back_populates="user")
|
|
transactions: Mapped[set[Transaction]] = relationship(back_populates="user")
|
|
|
|
name_re = r"[a-z]+"
|
|
card_re = r"(([Nn][Tt][Nn][Uu])?[0-9]+)?"
|
|
rfid_re = r"[0-9a-fA-F]*"
|
|
|
|
def __init__(self, name, card, rfid=None, credit=0):
|
|
self.name = name
|
|
if card == '':
|
|
card = None
|
|
self.card = card
|
|
if rfid == '':
|
|
rfid = None
|
|
self.rfid = rfid
|
|
self.credit = credit
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def is_anonymous(self):
|
|
return self.card == '11122233' |