autoRun.py

# coding: utf-8
"""
@Author: ZhaoBin
@Date: 2018-08-05 13:30:41
@Last Modified by:   ZhaoBin
@Last Modified time: 2018-08-08 17:00:41
"""
from time import time

from auto_us import AllPage, UsPipeline
from auto_us.settings import UsError, os, sleep, strftime

st = 0


class UsRun:
    '''美国签证自动化录入程序逻辑控制模块
    '''

    def __init__(self):
        self.control = {
            0: self.con0,
            1: self.con1,
            2: self.con2,
        }
        self.urList = [
            "Personal1", "Personal2", "AddressPhone", "PptVisa", "Travel",
            "TravelCompanions", "PreviousUSTravel", "USContact", "Relatives",
            "Spouse", "DeceasedSpouse", "PrevSpouse", "WorkEducation1",
            "WorkEducation2", "WorkEducation3", "SecurityandBackground1",
            "SecurityandBackground2", "SecurityandBackground3",
            "SecurityandBackground4", "SecurityandBackground5", "UploadPhoto",
            "ConfirmPhoto", "ReviewPersonal", "ReviewTravel", "ReviewUSContact",
            "ReviewFamily", "ReviewWorkEducation", "ReviewSecurity",
            "ReviewLocation", "SignCertify"
        ]

    def fillInfo(self):
        if self.au.resPublic['aacode']:
            self.au.continueGo()
        else:
            self.au.default

    def sendInfo(self):
        ans = self.au.run
        if ans:
            return 1
        return 0

    def done(self):
        self.au.signCertify()
        self.au.done()
        self.au.uploadPdf
        print('upload Pdf file over')

    def con0(self):
        global st
        st = time()
        if self.au.getNode not in self.urList:
            self.fillInfo()
        self.sendInfo()

    def con1(self):
        if hasattr(self.au, "driver") and '--headless' in self.au.chrome_options._arguments:
            self.au.driver.quit()
            self.au.getDriver(False)
        self.fillInfo()
        self.done()

    def con2(self):
        global st
        self.au.renamePdf()
        self.au.uploadPdf
        print('=' * 20, "End time :" + time() - st, '=' * 20, '\n\n')

    @property
    def run(self):
        # ========
        # 开始执行
        # ========
        while True:
            print(f"\n{'#':=<8}#\n# DS160 #\n{'#':=<8}#")
            self.auto = UsPipeline()
            # print('数据库连接完毕...')

            # 获取数据库信息
            data = self.auto.selDBInfo
            # 判断是否需要申请
            if data:
                # =======
                # 数据处理
                # =======
                print('有数据进行提交\n')
                self.au = AllPage(data=self.auto.data,
                                  noWin=False, usPipe=self.auto)
                try:
                    self.control[self.au.resPublic["visa_status"]]()
                except Exception as e:
                    print(e)
                    if hasattr(self.au, 'driver'):
                        self.au.driver.quit()
                        self.au.getDriver()

            self.auto = None
            try:
                self.au.driver.quit()
            except Exception:
                pass

            self.au = None

            print('没有数据, 等待中...', strftime('%m/%d %H:%M:%S'))
            sleep(5)
            os.system("cls")


def main():
    while True:
        # r.run
        try:
            r = UsRun()
            r.run
        except UsError as ue:
            print("in ue error")
            print(ue)
        except Exception as e:
            print("in e error")
            print(f"other:\n{e}")
            print("sleep 30s")
        finally:
            del r
        sleep(5)


if __name__ == '__main__':
    # sleep(60)
    print(strftime("%Y-%m-%d %H-%M-%S"))
    main()