init_data.py - master - stepa228/apartment-booking

init_data.py
140 строк5 KB
stepa228
Полный проект с CI/CD пайплайном
07 июн 2026, 17:07
07 июн 2026, 17:07772f8ba
import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
django.setup()

from django.contrib.auth import get_user_model
from properties.models import Amenity

User = get_user_model()


def init_users():
    """Создание или обновление пользователей"""

    # Администратор
    if not User.objects.filter(username='admin').exists():
        admin = User.objects.create_superuser(
            username='admin',
            email='admin@example.com',
            password='admin123',
            phone='+79001234567',
            role='owner'
        )
        print(f"✓ Создан администратор: {admin.username}")
    else:
        admin = User.objects.get(username='admin')
        admin.role = 'owner'
        admin.save()
        print(f"• Администратор уже существует, роль обновлена")

    # Владелец 1
    if not User.objects.filter(username='owner1').exists():
        owner1 = User.objects.create_user(
            username='owner1',
            email='owner1@example.com',
            password='owner123',
            phone='+79001111111',
            role='owner'
        )
        print(f"✓ Создан владелец: {owner1.username}")
    else:
        owner1 = User.objects.get(username='owner1')
        owner1.role = 'owner'
        owner1.save()
        print(f"• Владелец owner1 уже существует, роль обновлена")

    # Владелец 2
    if not User.objects.filter(username='owner2').exists():
        owner2 = User.objects.create_user(
            username='owner2',
            email='owner2@example.com',
            password='owner123',
            phone='+79003333333',
            role='owner'
        )
        print(f"✓ Создан владелец: {owner2.username}")
    else:
        owner2 = User.objects.get(username='owner2')
        owner2.role = 'owner'
        owner2.save()
        print(f"• Владелец owner2 уже существует, роль обновлена")

    # Арендатор 1
    if not User.objects.filter(username='guest1').exists():
        guest1 = User.objects.create_user(
            username='guest1',
            email='guest1@example.com',
            password='guest123',
            phone='+79002222222',
            role='guest'
        )
        print(f"✓ Создан арендатор: {guest1.username}")
    else:
        guest1 = User.objects.get(username='guest1')
        guest1.role = 'guest'
        guest1.save()
        print(f"• Арендатор guest1 уже существует, роль обновлена")

    # Арендатор 2
    if not User.objects.filter(username='guest2').exists():
        guest2 = User.objects.create_user(
            username='guest2',
            email='guest2@example.com',
            password='guest123',
            phone='+79004444444',
            role='guest'
        )
        print(f"✓ Создан арендатор: {guest2.username}")
    else:
        guest2 = User.objects.get(username='guest2')
        guest2.role = 'guest'
        guest2.save()
        print(f"• Арендатор guest2 уже существует, роль обновлена")


def init_amenities():
    """Создание удобств"""
    amenities_data = [
        {'name': 'Wi-Fi', 'icon': 'wifi'},
        {'name': 'Кондиционер', 'icon': 'ac'},
        {'name': 'Стиральная машина', 'icon': 'washing-machine'},
        {'name': 'Телевизор', 'icon': 'tv'},
        {'name': 'Кухня', 'icon': 'kitchen'},
        {'name': 'Парковка', 'icon': 'parking'},
        {'name': 'Балкон', 'icon': 'balcony'},
        {'name': 'Лифт', 'icon': 'elevator'},
        {'name': 'Бассейн', 'icon': 'pool'},
        {'name': 'Спортзал', 'icon': 'gym'},
    ]

    for data in amenities_data:
        amenity, created = Amenity.objects.get_or_create(
            name=data['name'],
            defaults={'icon': data['icon']}
        )
        if created:
            print(f"✓ Создано удобство: {amenity.name}")
        else:
            print(f"• Удобство уже существует: {amenity.name}")


if __name__ == '__main__':
    print("=" * 50)
    print("Инициализация данных...")
    print("=" * 50)

    init_users()
    print("-" * 50)
    init_amenities()

    print("=" * 50)
    print("ГОТОВО!")
    print(f"Пользователей: {User.objects.count()}")
    print(f"Удобств: {Amenity.objects.count()}")
    print("=" * 50)
    print("\nДанные для входа:")
    print("Админ: admin / admin123")
    print("Владелец: owner1 / owner123")
    print("Арендатор: guest1 / guest123")