from ExcelReader import ExcelReader
from ExcelWriter import ExcelWriter
from Macro import Macro
import datetime


# read id and pw from file
def readUserInfo():
    file = open('userinfo.txt', 'r')
    lines = file.read().splitlines()
    file.close()
    return lines

def process(mode):
    userinfo = readUserInfo()
    macro = Macro()
    macro.mode = mode

    macro.login(userinfo[0], userinfo[1])

    if mode == 3:
        macro.printPDF()

    else:
        values = ExcelReader('Input2.xlsx').readAllData()

        macro.index = 1
        for value in values:
            macro.index += 1

            data = None
            datas = None
            if macro.sendData(value):
                if mode == 4:
                    datas = macro.saveEvents(value.typeOfItem)
                else:
                    data = macro.saveData(value.typeOfItem)

                if mode == 1:
                    now = datetime.datetime.now()
                    d = now.strftime('%Y-%m-%d')
                    t = now.strftime('%H:%M:%S')

                    data[0] = str(value.index)
                    data[4] = d
                    data[5] = t

                    excelWriter = ExcelWriter('Output.xlsx')
                    excelWriter.writeData(data)

                elif mode == 4:
                    now = datetime.datetime.now()
                    d = now.strftime('%Y-%m-%d')
                    t = now.strftime('%H:%M:%S')


                    for data in datas:
                        data[0] = str(value.index)
                        data[4] = d
                        data[5] = t

                        excelWriter = ExcelWriter('Output2.xlsx')
                        excelWriter.writeData(data)

answer = int(input('어떤 작업을 하시겠습니까? (숫자만 입력해주세요)\n'
                   '1. 매매목록조회\n'
                   '2. 결제대상에 담기\n'
                   '3. 미발급 등본 PDF 저장\n'
                   '4. 등기신청사건 처리현황 저장\n'))
process(answer)