42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime, TIMESTAMP
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.sql import func
|
|
|
|
Base = declarative_base()
|
|
|
|
class UploadLog(Base):
|
|
__tablename__ = "upload_logs"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
filename = Column(String, nullable=False)
|
|
table_name = Column(String, nullable=False)
|
|
file_type = Column(String, nullable=False)
|
|
rows_count = Column(Integer)
|
|
uploaded_at = Column(TIMESTAMP, server_default=func.now())
|
|
status = Column(String)
|
|
message = Column(Text)
|
|
|
|
|
|
class Organization(Base):
|
|
__tablename__ = "organizations"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(100), unique=True, nullable=False)
|
|
address = Column(String(200), nullable=True)
|
|
|
|
users = relationship("User", back_populates="organization")
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String(50), unique=True, nullable=False)
|
|
password_hash = Column(String(255), nullable=False)
|
|
role = Column(String(50), nullable=False, default="user") # <── Added role
|
|
organization_id = Column(Integer, ForeignKey("organizations.id"), nullable=True)
|
|
active_token = Column(String(255), nullable=True)
|
|
token_expired_at = Column(DateTime, nullable=True)
|
|
last_login = Column(DateTime, nullable=True)
|
|
|
|
organization = relationship("Organization", back_populates="users") |