from app import db
from app.models import User, Profile, Notification, BillingInfo, SubscriptionPlan

users = User.query.all()

for user in users:
    if not user.profile:
        user.profile = Profile(
            user_id=user.id,
            street_address='',
            city='',
            state='',
            phone_number='',
            location='',
            currency=''
        )
    if not user.notification:
        user.notification = Notification(
            user_id=user.id,
            notification_email=user.email,
            newsletter_notifications=False,
            invoice_notifications=False,
            delivery_notifications=False,
            email_notifications=False
        )
    if not user.billing_info:
        user.billing_info = BillingInfo(
            user_id=user.id,
            business_name=user.company_name,
            business_address='',
            first_name=user.first_name,
            last_name=user.last_name,
            business_sector='',
            country=''
        )
    if not user.subscription_plan:
        user.subscription_plan = SubscriptionPlan(
            user_id=user.id,
            plan_name='Free',
            sites_per_month=0
        )
    db.session.add(user)

db.session.commit()