import os
from app import db, app, User, Transaction
from datetime import date, datetime

app.app_context().push()

# Создание таблиц
db.create_all()

# Добавление тестового пользователя
if not User.query.first():
    test_user = User(username="testuser", email="test@example.com")
    test_user.set_password("testpassword")
    db.session.add(test_user)
    db.session.commit()
    
    # Добавление тестовых транзакций
    transactions = [
        Transaction(type='income', amount=50000, category='Зарплата', date=date(2023, 10, 15), description='Основная зарплата'),
        Transaction(type='expense', amount=15000, category='Аренда', date=date(2023, 10, 5), description='Аренда квартиры'),
        Transaction(type='expense', amount=3500, category='Продукты', date=date(2023, 10, 12), description='Супермаркет'),
        Transaction(type='expense', amount=2500, category='Транспорт', date=date(2023, 10, 18), description='Бензин'),
        Transaction(type='income', amount=10000, category='Фриланс', date=date(2023, 10, 20), description='Веб-разработка')
    ]
    
    for t in transactions:
        db.session.add(t)
    db.session.commit()
    print("База данных инициализирована с тестовыми данными")
else:
    print("База данных уже содержит данные")