import pymysql

# --- Подключение к базе ---
conn = pymysql.connect(
    host='localhost',
    user='b2b_leads',
    password='v01lEHeGIdJUXSjQ',
    database='b2b_leads',
    charset='utf8mb4'
)
cursor = conn.cursor()

# --- Ищем ИНН, в которых есть 'E' ---
cursor.execute("SELECT id, inn FROM companies WHERE inn LIKE '%E%' OR inn LIKE '%e%'")
rows = cursor.fetchall()
print(f"Нашел подозрительных ИНН: {len(rows)}")

# --- Преобразуем их в нормальный формат ---
count = 0
for i, (cid, inn) in enumerate(rows):
    try:
        fixed_inn = '{:.0f}'.format(float(inn.replace(',', '.')))
        if fixed_inn != inn:
            cursor.execute("UPDATE companies SET inn = %s WHERE id = %s", (fixed_inn, cid))
            count += 1
        if i % 100 == 0:
            print(f"Обработано: {i}/{len(rows)} | Исправлено: {count}")
            conn.commit()
    except Exception as e:
        print(f"Ошибка на ID {cid} / ИНН {inn}: {e}")
        conn.rollback()

conn.commit()
print(f"Финиш! Исправлено {count} ИНН.")