import os
import sys
import django

# Setup Django
base_dir = os.path.dirname(os.path.abspath(__file__))
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings.local')
sys.path.insert(0, base_dir)
django.setup()

from django.db import connection

cursor = connection.cursor()

# Account 1: bb97f5aaf5e744049533fd1c84640084
acc_id = 'bb97f5aaf5e744049533fd1c84640084'
tenant_id = '7231bde9e867484faf5eef08ac2ac68d'

print('Fixing folders for account:', acc_id)

# Delete duplicate INBOX
cursor.execute("DELETE FROM email_emailfolder WHERE id = 2")
print('Deleted duplicate INBOX (id=2)')

# Create missing folders
folders = [
    ('Sent', 'sent'),
    ('Drafts', 'draft'),
    ('Trash', 'trash'),
    ('Spam', 'spam'),
    ('Outbox', 'outbox'),
]

for name, ftype in folders:
    sql = "INSERT INTO email_emailfolder (created_at, updated_at, tenant_id, name, remote_id, folder_type, unread_count, account_id) VALUES (datetime('now'), datetime('now'), '{0}', '{1}', '{2}', '{2}', 0, '{3}')".format(tenant_id, name, ftype, acc_id)
    cursor.execute(sql)
    print('Created:', name)

# Verify
cursor.execute('SELECT id, name, folder_type, account_id FROM email_emailfolder ORDER BY account_id, folder_type')
print()
print('All folders:')
for f in cursor.fetchall():
    print('  Account:', f[3], 'Name:', f[1], 'Type:', f[2])