118 lines
4.1 KiB
Python
118 lines
4.1 KiB
Python
import datetime
|
|
from sqlalchemy import create_engine, Column, Integer, String, Boolean, DateTime, Text, 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="")
|
|
schedule_end_date = Column(String(10), default="")
|
|
|
|
pre_buffer = Column(Integer, default=0)
|
|
post_buffer = Column(Integer, default=0)
|
|
|
|
description = Column(Text, default="")
|
|
|
|
plik_enabled = Column(Boolean, default=False)
|
|
delete_after_upload = Column(Boolean, default=False)
|
|
|
|
ntfy_enabled = Column(Boolean, default=False)
|
|
|
|
station_id = Column(Integer, nullable=True)
|
|
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)
|
|
|
|
|
|
class Station(Base):
|
|
__tablename__ = "stations"
|
|
|
|
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")
|
|
description = Column(Text, default="")
|
|
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)
|
|
file_size = Column(Integer, nullable=True)
|
|
plik_url = Column(String(1000), 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)
|
|
_migrate(_engine)
|
|
|
|
|
|
def _migrate(engine):
|
|
with engine.connect() as conn:
|
|
for stmt in [
|
|
"ALTER TABLE recordings ADD COLUMN file_size INTEGER",
|
|
"ALTER TABLE jobs ADD COLUMN station_id INTEGER",
|
|
"ALTER TABLE jobs ADD COLUMN pre_buffer INTEGER DEFAULT 0",
|
|
"ALTER TABLE jobs ADD COLUMN post_buffer INTEGER DEFAULT 0",
|
|
"ALTER TABLE recordings ADD COLUMN plik_url VARCHAR(1000)",
|
|
]:
|
|
try:
|
|
conn.execute(text(stmt))
|
|
conn.commit()
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def get_session():
|
|
return _SessionLocal()
|