from faker import Faker
import random

fake = Faker('ru_RU')

# Префиксы и возможные названия диагнозов
prefixes = ['A', 'B', 'C', 'D', 'E', 'F']
diagnosis_names = [
    "Грипп", "ОРВИ", "Пневмония", "Гастрит", "Мигрень", "Астма", "Гипертония",
    "Сахарный диабет", "Артрит", "Бронхит", "Невралгия", "Тонзиллит", "Ангина",
    "Гепатит", "Цистит", "Остеохондроз", "Конъюнктивит", "Панкреатит", "Радикулит"
]

used_codes = set()

with open("insert_diagnoses.sql", "w", encoding="utf-8") as f:
    while len(used_codes) < 20:
        # Генерируем уникальный код диагноза
        code = random.choice(prefixes) + str(random.randint(0, 99)).zfill(2)
        if code in used_codes:
            continue
        used_codes.add(code)

        # Случайное наименование
        name = random.choice(diagnosis_names)

        # Записываем SQL-запрос
        f.write(
            f"INSERT INTO диагнозы (код_диагноза, наименование) "
            f"VALUES ('{code}', '{name}');\n"
        )