mirror of
https://github.com/jaypyles/Scraperr.git
synced 2025-11-04 08:24:55 +00:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
# STL
|
|
import logging
|
|
from typing import Any
|
|
|
|
# LOCAL
|
|
from api.backend.database.utils import format_list_for_query
|
|
from api.backend.database.common import query as common_query
|
|
from api.backend.database.common import insert as common_insert
|
|
from api.backend.database.common import update as common_update
|
|
from api.backend.database.queries.job.job_queries import JOB_INSERT_QUERY
|
|
|
|
LOG = logging.getLogger("Job")
|
|
|
|
|
|
def insert(item: dict[str, Any]) -> None:
|
|
common_insert(
|
|
JOB_INSERT_QUERY,
|
|
(
|
|
item["id"],
|
|
item["url"],
|
|
item["elements"],
|
|
item["user"],
|
|
item["time_created"],
|
|
item["result"],
|
|
item["status"],
|
|
item["chat"],
|
|
item["job_options"],
|
|
item["agent_mode"],
|
|
item["prompt"],
|
|
),
|
|
)
|
|
|
|
LOG.debug(f"Inserted item: {item}")
|
|
|
|
|
|
async def get_queued_job():
|
|
query = (
|
|
"SELECT * FROM jobs WHERE status = 'Queued' ORDER BY time_created DESC LIMIT 1"
|
|
)
|
|
res = common_query(query)
|
|
LOG.debug(f"Got queued job: {res}")
|
|
return res[0] if res else None
|
|
|
|
|
|
async def update_job(ids: list[str], field: str, value: Any):
|
|
query = f"UPDATE jobs SET {field} = ? WHERE id IN {format_list_for_query(ids)}"
|
|
res = common_update(query, tuple([value] + ids))
|
|
LOG.debug(f"Updated job: {res}")
|
|
|
|
|
|
async def delete_jobs(jobs: list[str]):
|
|
if not jobs:
|
|
LOG.debug("No jobs to delete.")
|
|
return False
|
|
|
|
query = f"DELETE FROM jobs WHERE id IN {format_list_for_query(jobs)}"
|
|
res = common_update(query, tuple(jobs))
|
|
|
|
return res > 0
|