84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
import datetime
|
|
from sqlalchemy import create_engine, Column, Integer, String, Boolean, DateTime, Text
|
|
from sqlalchemy.orm import declarative_base, sessionmaker
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class Job(Base):
|
|
__tablename__ = "jobs"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
name = Column(String(200), nullable=False)
|
|
stream_url = Column(String(1000), nullable=False)
|
|
stream_type = Column(String(20), default="auto")
|
|
output_format = Column(String(10), default="")
|
|
max_duration = Column(Integer, nullable=True)
|
|
extra_ffmpeg_args = Column(String(500), default="")
|
|
segment_duration = Column(Integer, nullable=True)
|
|
|
|
schedule_enabled = Column(Boolean, default=False)
|
|
schedule_once = Column(Boolean, default=False)
|
|
schedule_date = Column(String(10), default="")
|
|
schedule_days = Column(String(50), default="*")
|
|
schedule_start = Column(String(5), default="")
|
|
schedule_stop = Column(String(5), default="")
|
|
|
|
plik_enabled = Column(Boolean, default=False)
|
|
delete_after_upload = Column(Boolean, default=False)
|
|
|
|
ntfy_enabled = Column(Boolean, default=False)
|
|
|
|
metadata_monitor_enabled = Column(Boolean, default=False)
|
|
metadata_pattern = Column(String(200), default="")
|
|
metadata_grace_period = Column(Integer, default=300)
|
|
metadata_poll_interval = Column(Integer, default=30)
|
|
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)
|
|
|
|
|
|
class Recording(Base):
|
|
__tablename__ = "recordings"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
job_id = Column(Integer, nullable=True)
|
|
job_name = Column(String(200), nullable=False)
|
|
stream_url = Column(String(1000), nullable=False)
|
|
stream_type = Column(String(20), default="")
|
|
output_dir = Column(String(500), default="")
|
|
output_file = Column(String(500), default="")
|
|
pid = Column(Integer, nullable=True)
|
|
status = Column(String(20), default="running") # running, stopped, completed, error
|
|
started_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
stopped_at = Column(DateTime, nullable=True)
|
|
scheduled_stop = Column(DateTime, nullable=True)
|
|
segment_mode = Column(Boolean, default=False)
|
|
is_scheduled = Column(Boolean, default=False)
|
|
error_message = Column(Text, nullable=True)
|
|
|
|
|
|
class LogEntry(Base):
|
|
__tablename__ = "logs"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
timestamp = Column(DateTime, default=datetime.datetime.utcnow)
|
|
level = Column(String(10), default="INFO")
|
|
message = Column(Text, nullable=False)
|
|
job_name = Column(String(200), nullable=True)
|
|
|
|
|
|
_engine = None
|
|
_SessionLocal = None
|
|
|
|
|
|
def init_db(db_path: str):
|
|
global _engine, _SessionLocal
|
|
_engine = create_engine(f"sqlite:///{db_path}", connect_args={"check_same_thread": False})
|
|
Base.metadata.create_all(_engine)
|
|
_SessionLocal = sessionmaker(bind=_engine)
|
|
|
|
|
|
def get_session():
|
|
return _SessionLocal()
|