test_script.py - main - NeuralNetworkCorporation/TechTalentHub

test_script.py
48 строк2 KB
Tim Gray Fan
Prepare project for defense and 1C artifacts
03 июн 2026, 09:49
03 июн 2026, 09:49f87494a
"""Ручной сценарий для проверки демо-данных.

Файл больше не выполняет запросы к базе при импорте, поэтому `python manage.py test`
не создаёт побочные эффекты. Запускать вручную так:

    python manage.py shell -c "import test_script; test_script.run_demo_request()"
"""

from datetime import timedelta

from django.utils import timezone

from users.models import Employee
from vacations.models import VacationRequest


def run_demo_request():
    print("ВСЕ СОТРУДНИКИ:")
    for emp in Employee.objects.all():
        print(f"ID: {emp.id}, Имя: {emp.name}")

    ivanov = Employee.objects.filter(name__icontains='Ivan').first()
    petrov = Employee.objects.filter(name__icontains='Petr').first()
    anna = Employee.objects.filter(name__icontains='Anna').first()

    print(f"\nИванов: {ivanov.name if ivanov else 'Не найден'}")
    print(f"Петров: {petrov.name if petrov else 'Не найден'}")
    print(f"Анна: {anna.name if anna else 'Не найден'}")

    if not (ivanov and petrov):
        print("❌ Не все сотрудники найдены. Выполните: python manage.py create_demo_data")
        return

    ivanov.manager = petrov
    ivanov.save(update_fields=['manager', 'updated_at'])
    print(f"✅ Руководитель Иванова: {ivanov.manager.name}")

    new_request = VacationRequest.objects.create(
        employee=ivanov,
        request_type='vacation',
        start_date=timezone.now().date() + timedelta(days=5),
        end_date=timezone.now().date() + timedelta(days=12),
        comment='Тестовая заявка',
        status='submitted',
        current_step=1,
        step_started_at=timezone.now(),
    )
    print(f"✅ Создана заявка #{new_request.id}")