from jb.decorators import AMTM, AM
from jb.flow.assignment_tasks import approve_assignment
from jb.managers.amt import APPROVAL_MESSAGE

res = AM.pg_config.execute_sql_query(
    """
SELECT amt_assignment_id, submit_time, h.amt_hit_id, ht.amt_hit_type_id
FROM mtwerk_assignment AS a
LEFT JOIN mtwerk_hit AS h ON h.id = a.hit_id
LEFT JOIN mtwerk_hittype AS ht ON h.hit_type_id = ht.id
WHERE amt_worker_id='A1FMSKDAIK80PQ'
and a.status = 2
AND a.created_at>'2026-04-21 00:20:54.709322'
ORDER BY a.created_at;
"""
)

for x in res:
    approve_assignment(
        amtm=AMTM,
        am=AM,
        amt_assignment_id=x['amt_assignment_id'],
        msg=APPROVAL_MESSAGE,
        amt_hit_type_id=x['amt_hit_type_id'],
        override_rejection=True
    )