42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from sqlalchemy import Column, Integer, String, Enum, DateTime, ForeignKey, JSON
|
|
from sqlalchemy.orm import relationship
|
|
from .database import Base
|
|
import datetime
|
|
import enum
|
|
|
|
class UserRole(str, enum.Enum):
|
|
admin = "admin"
|
|
user = "user"
|
|
|
|
class JobStatus(str, enum.Enum):
|
|
pending = "pending"
|
|
running = "running"
|
|
passed = "passed"
|
|
failed = "failed"
|
|
aborted = "aborted"
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
username = Column(String, unique=True, index=True)
|
|
hashed_password = Column(String)
|
|
role = Column(Enum(UserRole), default=UserRole.user)
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
jobs = relationship("Job", back_populates="owner")
|
|
|
|
class Job(Base):
|
|
__tablename__ = "jobs"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"))
|
|
branch_name = Column(String)
|
|
scenarios = Column(JSON)
|
|
environment = Column(String)
|
|
test_mode = Column(String)
|
|
status = Column(Enum(JobStatus), default=JobStatus.pending)
|
|
result_path = Column(String, nullable=True)
|
|
duration = Column(String, nullable=True)
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)
|
|
|
|
owner = relationship("User", back_populates="jobs")
|