Adding initial db to queue

This commit is contained in:
Brett Williams
2023-05-23 23:52:29 -05:00
parent 6c3d0e8f6e
commit 4446c61f1d
2 changed files with 27 additions and 9 deletions

View File

@@ -4,10 +4,12 @@ import os
import platform
from datetime import datetime
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
import psutil
import requests
from .scheduled_job import ScheduledJob
from .scheduled_job import ScheduledJob, Base
from .render_workers.render_worker import RenderStatus
logger = logging.getLogger()
@@ -23,6 +25,11 @@ class JobNotFoundError(Exception):
class RenderQueue:
engine = create_engine('sqlite:///database.db')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()
job_queue = []
render_clients = []
maximum_renderer_instances = {'blender': 2, 'aerender': 1, 'ffmpeg': 4}
@@ -45,6 +52,8 @@ class RenderQueue:
cls.job_queue.append(render_job)
if force_start:
cls.start_job(render_job)
cls.session.add(render_job)
cls.session.commit()
else:
# todo: implement client rendering
logger.warning('remote client rendering not implemented yet')