# -*- coding: utf-8 -*-
"""Sửa nhỏ: số việc 'đang chờ' chỉ đếm việc agent THỰC SỰ làm được (có ảnh hoặc chờ dự đoán)."""
import ast
P = r'C:\Claw 2\Nghien cuu\Du-an-B-He-mau-Stain-Glaze\calib-app\server_calib.py'
s = open(P, encoding='utf-8').read()
old = """        cho = con.execute(\"SELECT COUNT(*) FROM hon_hop WHERE trang_thai IN ('cho_du_doan','da_du_doan')\").fetchone()[0]"""
new = """        cho = con.execute(\"SELECT COUNT(*) FROM hon_hop WHERE trang_thai='cho_du_doan' \"
                          \"OR (trang_thai='da_du_doan' AND anh_thuc_te IS NOT NULL AND anh_thuc_te<>'')\").fetchone()[0]
        doi_anh = con.execute(\"SELECT COUNT(*) FROM hon_hop WHERE trang_thai='da_du_doan' AND (anh_thuc_te IS NULL OR anh_thuc_te='')\").fetchone()[0]"""
assert old in s, 'không tìm thấy đoạn cần sửa'
s = s.replace(old, new)
s = s.replace("""        return (r[0] or '', r[1] or '', r[2] or '', cho)""",
              """        return (r[0] or '', r[1] or '', r[2] or '', cho, doi_anh)""")
s = s.replace("""        return ('', '', '', 0)""", """        return ('', '', '', 0, 0)""")
s = s.replace("""        if nhip[0] else ('<b>Agent đang rảnh</b> — đang chờ %d việc' % nhip[3])""",
              """        if nhip[0] else ('<b>Agent đang rảnh</b> — %d việc đang xếp hàng, %d hỗn hợp đang chờ chủ nhân tải ảnh' % (nhip[3], nhip[4]))""")
open(P, 'w', encoding='utf-8').write(s)
ast.parse(s)
print('đã sửa cách đếm việc đang chờ: cú pháp OK')