import os
import os.path
import shutil
import smtplib
import sys
import time
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

import job
from abaqus import *
from abaqusConstants import *
from tools.standard import standard_str

def batch_analysis(foldName, mutiple, to_email, power_off):
    start = time.time()
    fatherdir = os.path.dirname(foldName)
    resultdir = os.path.join(foldName, 'Result')
    """
    驗證文件夾是否已被建立
    """
    try:
        os.mkdir(resultdir)
    except:
        print('directory "Result" already exist!')
    """
    獲取文件夾內部所有文件, 遍歷所有文件, 根據后綴名判斷是否是inp文件
    對于inp文件, 標準化名稱后提交, 按順序隊列式求解
    """
    infor = os.listdir(foldName)
    for item in infor:
        tempdir = os.path.join(foldName, item)
        tempitem = os.path.splitext(item)[0]
        extStr = os.path.splitext(item)[1]
        if os.path.isfile(tempdir):
            if extStr == '.inp':
                jobname = standard_str(tempitem)
                # mdb.JobFromInputFile(name=jobname, inputFileName=item)
                mdb.JobFromInputFile(name=jobname, inputFileName=item, numCpus=mutiple, numDomains=mutiple)
                mdb.jobs[jobname].submit()
                mdb.jobs[jobname].waitForCompletion()
                send_email(start, time.time(), tempitem, to_email)
    """
    獲取求解后的文件夾內所有文件, 
    編制求解信息文件, 
    定義需要被保存的文件的類型
    """
    newinfor = os.listdir(foldName)
    f = open('AnalysisInfo.txt', 'w')
    file_set = set(['.inp', '.cae', '.sta', '.jnl', '.odb', '.py'])
    for item in newinfor:
        tempfile = os.path.join(foldName, item)
        extStr = os.path.splitext(item)[1]
        if os.path.isfile(tempfile):
            if extStr in file_set:
                shutil.move(tempfile, resultdir)
                try:
                    tempStr = 'file ' + item + ' moved!\n'
                except:
                    tempStr = 'record.txt ' + ' remained for search!\n'
            else:
                try:
                    os.remove(tempfile)
                    tempStr = 'file ' + item + ' deleted!\n'
                except:
                    tempStr = 'record.txt ' + ' remained for search!\n'
        f.writelines(tempStr)
    f.close()

    end = time.time()
    # spend = 100000
    if power_off:
        o = "c:\windows\system32\shutdown -s -f -t 60"
        os.system(o)

def send_email(start, end, job, to_email):
    spend = end - start
    s = spend % 60
    m = spend // 60 % 60
    h = spend // 60 // 60
    start_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(start))
    end_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(end))
    spend_str = '%dH:%dM:%dS' % (h, m, s)
    user = 'yours@139.com'
    pwd = 'pwd'
    try:
        msg = MIMEMultipart()
        mail_msg = 'The analysis %s began at %s, ended at %s, Total cost %s.' % (job, start_str, end_str, spend_str)
        msg.attach(MIMEText(mail_msg, 'html', 'utf-8'))
        msg['Subject'] = end_str + ': Analysis of complete'
        msg['From'] = user
        msg['To'] = to_email
        s = smtplib.SMTP_SSL('smtp.139.com', 465)
        s.login(user, pwd)
        # print(msg.as_string())
        s.sendmail(user, [to_email], msg.as_string())
        s.quit()
        print('Success!')
    except Exception as e:
        print(e)