Coverage for fix_folders_simple.py: 0%

1import os 

2import sys 

3import django 

4 

5# Setup Django 

6base_dir = os.path.dirname(os.path.abspath(__file__)) 

7os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings.local') 

8sys.path.insert(0, base_dir) 

9django.setup() 

10 

11from django.db import connection 

12 

13cursor = connection.cursor() 

14 

15# Account 1: bb97f5aaf5e744049533fd1c84640084 

16acc_id = 'bb97f5aaf5e744049533fd1c84640084' 

17tenant_id = '7231bde9e867484faf5eef08ac2ac68d' 

18 

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

20 

21# Delete duplicate INBOX 

22cursor.execute("DELETE FROM email_emailfolder WHERE id = 2") 

23print('Deleted duplicate INBOX (id=2)') 

24 

25# Create missing folders 

26folders = [ 

27 ('Sent', 'sent'), 

28 ('Drafts', 'draft'), 

29 ('Trash', 'trash'), 

30 ('Spam', 'spam'), 

31 ('Outbox', 'outbox'), 

32] 

33 

34for name, ftype in folders: 

35 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) 

36 cursor.execute(sql) 

37 print('Created:', name) 

38 

39# Verify 

40cursor.execute('SELECT id, name, folder_type, account_id FROM email_emailfolder ORDER BY account_id, folder_type') 

41print() 

42print('All folders:') 

43for f in cursor.fetchall(): 

44 print(' Account:', f[3], 'Name:', f[1], 'Type:', f[2])