Archived
feat(bot): add bot account ownership and balance
This commit is contained in:
+31
-2
@@ -1,7 +1,7 @@
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from sqlalchemy import BigInteger, Boolean, DateTime, String
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, String, Text
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from bot.db.base import Base
|
||||
|
||||
@@ -19,9 +19,38 @@ class User(Base):
|
||||
first_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
||||
last_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
||||
is_admin: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
balance_credits: Mapped[int] = mapped_column(BigInteger, default=0, nullable=False)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True),
|
||||
default=utc_now,
|
||||
onupdate=utc_now,
|
||||
)
|
||||
bots: Mapped[list["BotAccount"]] = relationship(back_populates="owner", cascade="all, delete-orphan")
|
||||
|
||||
|
||||
class AppSetting(Base):
|
||||
__tablename__ = "app_settings"
|
||||
|
||||
key: Mapped[str] = mapped_column(String(128), primary_key=True)
|
||||
integer_value: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
updated_by: Mapped[int | None] = mapped_column(BigInteger, nullable=True)
|
||||
|
||||
|
||||
class BotAccount(Base):
|
||||
__tablename__ = "bot_accounts"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True)
|
||||
owner_id: Mapped[int] = mapped_column(ForeignKey("users.id"), index=True)
|
||||
bot_name: Mapped[str] = mapped_column(String(255), unique=True, index=True)
|
||||
steam_id: Mapped[str | None] = mapped_column(String(64), nullable=True)
|
||||
source_filename: Mapped[str] = mapped_column(String(255))
|
||||
config_sha256: Mapped[str] = mapped_column(String(64), index=True)
|
||||
upload_state: Mapped[str] = mapped_column(String(32), default="attached")
|
||||
last_error: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
is_attached: Mapped[bool] = mapped_column(Boolean, default=False)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now)
|
||||
updated_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
attached_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
owner: Mapped[User] = relationship(back_populates="bots")
|
||||
|
||||
Reference in New Issue
Block a user