import pandas as pd
from email_new_test import send_mail, username
import os
import re

reportname = pd.read_csv('reportname.csv', index_col='Reportname')
email_detail_columns = ['filename', 'markets', 'email', 'email1', 'email2', 'email3']
email_details = pd.read_csv('regionsmktclub_emails.csv', header=None, names=email_detail_columns, index_col='filename')
groups = set(list(reportname.group))
# print(email_details)
for detail in email_details.index:
    print(detail)
    email_ids = []
    files = []
    if detail.startswith('MKT'):
        email_ids.append(email_details.loc[detail].email)
        email_ids.append(email_details.loc[detail].email1)
        print(email_details.loc[detail].email2)
        email_ids.append(email_details.loc[detail].email2)

        if type(email_details.loc[detail].email3) == str:
            email_ids.append(email_details.loc[detail].email3)
        for group in groups:
            files = []
            grouped_reportname = reportname[reportname.group == group]
            for report in grouped_reportname.index:
                if grouped_reportname.loc[report].mm == 't':
                    path = os.path.join(report, 'pdf')
                    for r, d, f in os.walk(path):

                        if detail in f:
                            files.append(os.path.join(path, detail))
            send_mail(username, '{0} mm reports- Group {1}'.format(detail,group), 'PFA', email_ids, files=files)
    elif re.match(r'^\d\d\d\d', detail):
        email_ids.append(email_details.loc[detail].email)

        for group in groups:
            files = []
            grouped_reportname = reportname[reportname.group == group]
            for report in grouped_reportname.index:
                if grouped_reportname.loc[report].cm == 't':
                    path = os.path.join(report, 'pdf')
                    for r, d, f in os.walk(path):
                        if detail in f:
                            files.append(os.path.join(path, detail))
            send_mail(username, '{0} cm reports - Group {1}'.format(detail,group), 'PFA', email_ids, files=files)
    elif re.search(r'region.', detail):
        email_ids.append(email_details.loc[detail].email)
        markets = email_details.loc[detail].markets.split(';')
        for report in reportname.index:
            files = []
            if reportname.loc[report].rgm == 't':
                path = os.path.join(report, 'pdf')
                for r, d, f in os.walk(path):
                    for file in f:

                        if file == detail or (file in markets):
                            files.append(os.path.join(path, file))

            send_mail(username, ' {0} regional reports and associated marketing files'.format(report), 'PFA', email_ids, files=files)