from faker import Faker
import random

# Инициализация генератора на русском
fake = Faker('ru_RU')

# Возможные специальности и категории
specialties = ['Терапевт', 'Хирург', 'Педиатр', 'Офтальмолог', 'Кардиолог', 'Невролог']
categories = [ '1-я', '2-я', '3-я']

# Открываем файл для записи SQL-команд
with open("insert_vrachi.sql", "w", encoding="utf-8") as f:
    for _ in range(20):  # Генерируем 250 строк
        last_name = fake.last_name()
        first_name = fake.first_name()
        patronymic = fake.middle_name()
        specialty = random.choice(specialties)
        category = random.choice(categories)

        # Пишем INSERT-запрос
        f.write(f"INSERT INTO врачи (фамилия, имя, отчество, специальность, категория) "
                f"VALUES ('{last_name}', '{first_name}', '{patronymic}', '{specialty}', '{category}');\n")