mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-21 13:48:20 +00:00
33 lines
1012 B
Python
33 lines
1012 B
Python
import enum
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from sqlmodel import SQLModel, Field, DateTime, Column, Enum, BigInteger, Integer
|
|
|
|
__all__ = (
|
|
"User",
|
|
"UserDataBase",
|
|
"PermissionsEnum",
|
|
)
|
|
|
|
|
|
class PermissionsEnum(int, enum.Enum):
|
|
OWNER = 1
|
|
ADMIN = 2
|
|
PUBLIC = 3
|
|
|
|
|
|
class User(SQLModel):
|
|
__table_args__ = dict(mysql_charset="utf8mb4", mysql_collate="utf8mb4_general_ci")
|
|
id: Optional[int] = Field(default=None, sa_column=Column(Integer(), primary_key=True, autoincrement=True))
|
|
user_id: int = Field(sa_column=Column(BigInteger(), unique=True))
|
|
permissions: Optional[PermissionsEnum] = Field(sa_column=Column(Enum(PermissionsEnum)))
|
|
locale: Optional[str] = Field()
|
|
ban_end_time: Optional[datetime] = Field(sa_column=Column(DateTime(timezone=True)))
|
|
ban_start_time: Optional[datetime] = Field(sa_column=Column(DateTime(timezone=True)))
|
|
is_banned: Optional[int] = Field()
|
|
|
|
|
|
class UserDataBase(User, table=True):
|
|
__tablename__ = "users"
|