"""Docstring."""

import csv
import random

from locust import HttpUser
from locust import task

"""
SELECT
    eqd.encoding_queue_detail_id,
    eq.encoding_order_type,
    eq.meta_update,
    eqd.upc,
    eqd.dms_master_master_id
FROM direct_delivery.encoding_queue_detail eqd
JOIN direct_delivery.encoding_queue eq
    ON eq.encoding_queue_id = eqd.encoding_queue_id
WHERE
    TRUE
    -- AND eqd.upc IN ('199350142185', '844942135303')
    AND eqd.dms_master_master_id IN (
        339,
        568,
        1129
    )
    AND eq.meta_update = 'N'
LIMIT 10000
;
"""


class HelloWorldUser(HttpUser):

    job_ids: list[int] = []

    def __init__(self, client) -> None:
        super().__init__(client)
        with open('job_ids.csv', 'r') as f:
            reader = csv.DictReader(f)
            self.job_ids = [
                int(row['encoding_queue_detail_id'])
                for row in reader
            ]

    @task
    def hello_world(self) -> None:
        job_id = random.choice(self.job_ids)
        self.client.get(f'/jobs/{job_id}/eligibility')