python自动化：uiautomation、pyautogui操作会计记账系统（3）：处理序时簿

python自动化：uiautomation、pyautogui操作会计记账系统（1）：处理序时簿

#coding=utf-8
#chronological book，查询序时簿
import xlrd, xlwt, pyautogui, time, pyperclip, openpyxl,pandas,os, tkinter,\
    datetime, win32gui, uiautomation, subprocess, xlwings,json, easygui
from tkinter.filedialog import askopenfilename
pyautogui.PAUSE = 1


def creat_file(year_input):
    print('开始运行创建文件夹自定义函数……')
    with open('**电脑端账套号生成村委会文件夹用.txt', 'r') as f:
        dic = []
        for line in f.readlines():
            line = line.strip('\n')  # 去掉换行符\n
            b = line.split('\t')  # 将每一行以空格为分隔符转换成列表
            #print(list(b))
            dic.append(b)

        print(dic)
        admin_villi_dict=dict(dic)
        print('账套所在的村委会文件夹列表是：')
        print(admin_villi_dict)#列表转字典
    if not os.path.exists('D:\**会计系统导出数据'):
        os.mkdir('D:\**会计系统导出数据')

    admin_villi_list=['前', '']

    year = time.strftime("%Y")
    #print("当前年份是:", year)
    if year_input==year:#等于
        time1 = time.strftime("%Y%m%d", time.localtime())  # 加上秒不会出现同名
    else:
        time1= year_input #赋值
    #print('创建第3层文件夹folder3_1')
    folder3_1 = r"D:\**会计系统导出数据" + '\\' + time1 + '序时簿'
    #print('判断文件夹是否存在，不存在则创建，文件保存在：'+folder3_1)
    if not os.path.exists(folder3_1):
        os.mkdir(folder3_1)
    #print('创建第4层文件夹folder4_i')
    for i in range(0,16):
        folder_admin_villi=time1+admin_villi_list[i]+'序时簿'
        folder4_i=r'D:\**会计系统导出数据'+'\\'+ time1 + '序时簿'+'\\' +folder_admin_villi
        if not os.path.exists(folder4_i):
            os.mkdir(folder4_i)
            #print('创建文件夹：'+folder4_i)

    return  admin_villi_dict,folder3_1


def check_file(folder3_1):
    # 打开文件

    path= folder3_1
    dirs = os.listdir(path)
    year = time.strftime("%Y")



    file_accnt_num_list=[]
    for home, dirs, files in os.walk(path):
        for file_name in files:
            #filelist.append(os.path.join(home, file_name))
            if os.path.splitext(file_name)[1] == '.xls':
                #print(file_name)
                file_year = file_name[0:4]  # 截取文件名前4个字符
                #print(file_year)
                # os.rename(file, new)  # 进行重命名

                if file_year == year:
                    # 取整方法是可以去除0，但是遇到字符串不能转化为整型会报错
                    # file_accnt_num=int(file_name[9:12])
                    s = file_name[9:12]
                    file_accnt_num = "".join(s[::1]).lstrip("0")
                    #print(file_accnt_num)
                else:
                    s = file_name[5:8]
                    file_accnt_num = "".join(s[::1]).lstrip("0")
                    #print(file_accnt_num)
                file_accnt_num_list.append((file_accnt_num))

    #print(filelist)
    #print(file_accnt_num_list)
    #print('对列表进行排序，查找时可能减少遍历次数。也便于根据账套号顺序查找')
    #file_accnt_num_list2=file_accnt_num_list.sort()
    #print('sort影响列表本身，sorted不影响列表本身，用sort排序')
    file_exist_list = sorted(file_accnt_num_list)
    print('已导出的文件列表','file_exist_list:')
    print(file_exist_list)
    print('已导出的文件数量',len(file_exist_list))
    return file_exist_list

def accnt_login(account_Num):
    wc1 = uiautomation.WindowControl(searchDepth=1, Name='帐套登陆')
    # 设置为顶层
    #pyautogui.alert(text="按回车键继续程序", title="提示")  # 窗口前置才能用键盘快捷键
    wc1.SetTopmost(True)  # 置顶能用快捷键
    pc2_1= wc1.PaneControl(searchDepth=1, Name='帐套信息')
    ec3_8=pc2_1.EditControl(searchDepth=1, Name='', foundIndex=2)
    ec3_8.SetFocus()
    #print('ec3_8')
    #print('账套号输入框的位置是：')
    #print(ec3_8.BoundingRectangle)
    pyperclip.copy(account_Num) #account_Num是账套号
    pyautogui.hotkey('ctrl', 'v')  # 再粘贴
    pyautogui.press('enter')



def user_login():
    wc1=uiautomation.WindowControl(searchDepth=1, Name='用户注册')
    # 设置为顶层
    wc1.SetTopmost(True)  # 置顶能用快捷键
    pyautogui.press('down')  # 向下选择超级用户
    pyautogui.press('enter')  # 点击确定
    pyautogui.press('enter')  # 点击确定
    #print('超级用户登录成功')

def  select_year(screenWidth, screenHeight, year_input):
    #print('选择年份')
    #print('深度1')
    wc1 = uiautomation.WindowControl(searchDepth=1, Name='**经营管理系统V2007 - [凭证查询]')
    # 设置为顶层
    #wc1.SetTopmost(True)
    #print(wc1.Name)

    #print('深度2') #isum自创字母后的数字表示深度
    #sbc2_1= wc1.StatusBarControl(searchDepth=1, ClassName= 'TStatusBar')
    pc2_2= wc1.PaneControl(searchDepth=1,ClassName='MDIClient')
    #print(pc2_2.Name)#工作区

    #print('深度3')
    wc3_1= pc2_2.WindowControl(searchDepth=1,Name='凭证查询')
    #print(wc3_1.Name)  # 工作区

    #print('深度4')
    pc4_2= wc3_1.PaneControl(searchDepth=1, ClassName='TPanel',Name= '')
    #print(pc4_2.BoundingRectangle) #Rect: (0,183,311,1013)[311x830]

    #print('深度5')
    #Rect: (1,214,310,442)[309x228]
    pc5_1 = pc4_2.PaneControl(searchDepth=1, ClassName='TPanel',Name= '')

    #print('深度6')
    rbc6_14= pc5_1.RadioButtonControl(searchDepth=1, Name='指定年份')
    #print('点击：'+rbc6_14.Name)
    rbc6_14.Click()
    year = time.strftime("%Y")
    #print("当前年份是:", year)
    if year_input!=year:
        pyautogui.press('tab')
        pyperclip.copy(year_input)
        pyautogui.hotkey('ctrl', 'v')
        # 选项出现较慢，需要等待
        #print('点击生成')
        #pyautogui.click(screenWidth/8.3478, screenHeight /13.5)  # 系统中等字体显示，点击生成按钮(230,80)
        pyautogui.click(screenWidth /10.6667, screenHeight / 16.6154)  # 系统小字体显示，点击生成按钮(180,65)
        time.sleep(1)










# #代替三方鼠标定位软件
# x1,y1,x2,y2=700,27,788,51 #另选账套位置
# x1,y1,x2,y2=316,184,1919,986 #序时簿显示的范围
# x1,y1,x2,y2=259,560,1607,700#空白的区域
# x1,y1,x2,y2=315,183,1920,1013
# # pyautogui.moveTo(x1,y1,duration=2)
# # pyautogui.moveTo(x2,y1,duration=2)
# # pyautogui.moveTo(x2,y2,duration=2)
# # pyautogui.moveTo(x1,y2,duration=2)
# # pyautogui.moveTo(x1,y1,duration=2)
#
#




def to_excel(screenWidth, screenHeight, year_input, accnt_Num_name, i,admin_villi_dict):
    wc1 = uiautomation.WindowControl(searchDepth=1, Name='**经营管理系统V2007 - [凭证查询]')
    # time1 =time.strftime("%Y%m%d%H%M%S", time.localtime())#加上秒不会出现同名
    year = time.strftime("%Y")
    # print("当前年份是:", year)
    if year_input == year:  # 等于
        time1 = time.strftime("%Y%m%d", time.localtime())  # 加上秒不会出现同名
    else:
        time1 = year_input  # 赋值
    # file_path= r"D:\**会计系统导出数据"+'\\'+time1+'现金三栏明细账'+'\\'+file_name+'.xls'
    # 字典必须传入字符串，admin_villi_dict[str(i)]

    file_path = r"D:\**会计系统导出数据" + '\\' + time1 + '序时簿' + '\\' + time1 + admin_villi_dict[str(i)] + '序时簿'
    if not os.path.exists(file_path):
        # print('如果没有预先归类到行政村文件夹，就放在上一级文件夹')
        file_path = r"D:\**会计系统导出数据" + '\\' + time1 + '序时簿'
    # print('判断文件夹是否存在，不存在则创建，文件保存在：'+file_path)
    # if not os.path.exists(file_path):
    #     os.mkdir(file_path)
    file_name = time1 + " " + accnt_Num_name + "序时簿"
    file_title = "序时簿（" + accnt_Num_name + "）"

    print('文件保存路径是：' + file_path)
    file_full_path = file_path + '\\' + file_name + '.xls'
    print("文件名是:", end="")
    print(file_name)
    if os.path.exists(file_full_path):  # 如果文件不存在（即未导出）
        print('{}--文件已经存在，不需导出。'.format(file_full_path))
    else:
        #pyautogui.click(screenWidth/3.5888, screenHeight/13.5)  #系统中等字体点击导出(535,80)
        pyautogui.click(screenWidth / 4.4651, screenHeight /15.4286)  # 系统小等字体点击导出(430,70)
        time.sleep(0.5)
        # pyautogui.click(590,80)  #点击导出
        # time.sleep(1)
        pyautogui.press('down') #按向下键，选择导出Excel表格
        time.sleep(0.5)
        pyautogui.press('enter')  # 点击确定
        # pyautogui.click(640,345)  #点击桌面，弹出来的对话框位置不断变化的
        # #pyautogui.click(650,475)  #点击我的文档
        # pyautogui.click(650,475)  #点击文件名输入框
        # #pyautogui.press('backspace')#删除
        pyautogui.keyDown('backspace')#长摁删除键
        time.sleep(0.5)
        pyautogui.keyUp('backspace')
        #pyautogui.typewrite(file_name) #不能写入中文
        pyperclip.copy(file_full_path)  # 写入文件名
        pyautogui.hotkey('ctrl', 'v')  # 再粘贴
        pyautogui.press('enter')  # 点击确定
        print('导出Excel表会自动打开，所以需要等待较长时间。')
        wc1.Minimize()  # 最小化缩小窗口
        time.sleep(20)#等待自动打开Excel，必须足够长时间，不然会出现文件名出现不了
        pyautogui.click(screenWidth / 3.2, screenHeight / 4.6957)  # 点击标题，使得WPS前置。(600,230),Excel的显示方式必须是全屏
        pyautogui.typewrite(" ")
        pyautogui.press('backspace')
        #yautogui.press('insert')  #
        #print('修改表格标题:'+file_title)
        pyperclip.copy(file_title)  # 写入表格标题
        pyautogui.hotkey('ctrl', 'v')  # 再粘贴
        time.sleep(0.5)
        pyautogui.hotkey('alt', 'f4')  # 关闭Excel
        time.sleep(0.5)
        pyautogui.press('enter')  # 点击确定，保存更改
        time.sleep(1)
    wc1.Restore()  # 将窗口回复到最大化
    return  file_full_path
    #这里需要加一个，假如保存文件失败之后，**系统错误提示
    #标题是错误，内容是路径不存在或者文件不可写！ 确定。
    #pyautogui.press('enter')


def excel_modifier(file_full_path):
    # 创建app，打开工作表
    app = xlwings.App(visible=False, add_book=False)
    app.screen_updating = False
    app.display_alerts = False
    load_wb = app.books.open(file_full_path)
    load_ws = load_wb.sheets.active
    print('\t已打开工作表……')
    # 处理列，将指定列从大到小删除（避免先删除小列导致后续列号变动）
    #错误方法：for cell in ['J1', 'M1', 'N1', 'O1', 'P1', 'Q1']:
    for cell in ['Q1', 'P1', 'O1', 'N1', 'M1', 'J1']:
        load_ws.range(str(cell)).api.EntireColumn.Delete()  # 删除 ’J1‘ 单元格所在的列。删除对应出纳单号、制单、审核、过帐、审核标志、过帐标志
        print('删除{}列'.format(cell))
        #load_ws.api.columns('M').delete

    # load_ws.range('P1').api.EntireColumn.Delete()
    # load_ws.range('J1').api.EntireColumn.Delete()
    # load_ws.range('M1').api.EntireColumn.Delete()
    # load_ws.range('N1').api.EntireColumn.Delete()
    # load_ws.range('O1').api.EntireColumn.Delete()
    # load_ws.range('P1').api.EntireColumn.Delete()
    # load_ws.range('Q1').api.EntireColumn.Delete()

    print('\t开始处理表格……需要等待一定时间')
    load_wb.save()
    load_wb.close()
    app.quit()
    print('处理完毕===>>>', file_full_path, '\n\n')


def chage_accnt():
    #print('深度1')
    wc1 = uiautomation.WindowControl(searchDepth=1, Name='**经营管理系统V2007 - [凭证查询]')
    # 设置为顶层
    #wc1.SetTopmost(True)
    #print(wc1.Name)

    #print('深度2') #isum自创字母后的数字表示深度
    #sbc2_1= wc1.StatusBarControl(searchDepth=1, ClassName= 'TStatusBar')
    pc2_2= wc1.PaneControl(searchDepth=1,ClassName='MDIClient')
    #print(pc2_2.Name)#工作区

    mbc2=wc1.MenuBarControl(searchDepth=1, Name= '应用程序')
    #print('mbc2')
    #print(mbc2.Name)

    mic3_9=mbc2.MenuItemControl(searchDepth=1, Name='', foundIndex=9)
    #print('mic3_9')
    #print(mic3_9.BoundingRectangle)#(700,27,788,51)[88x24]
    #print('点击另选账套')
    mic3_9.Click()

    #另选账套的坐标：Rect: (700,27,788,51)
    #这四个 参数 分别代表的意思是： left   top   right   bottom   是 左 上 右 下。


def open_fold():
    # #open folder，打开文件夹,用win+E快捷键
    pyautogui.hotkey('win', 'e')
    pyautogui.press('tab')
    #pyautogui.press('insert')
    pyautogui.press('enter')
    pyperclip.copy('D:\**会计系统导出数据')
    pyautogui.hotkey('ctrl', 'v')
    pyautogui.press('enter')



def func(year_input):
    #print('当前屏幕分辨率宽 X 高：')
    screenWidth, screenHeight = pyautogui.size()
    #print(screenWidth, screenHeight)  # 屏幕分辨率1920 1080,导出的按钮595,80
    msg = '浏览需要导出的村委会Excel表格文件并打开'
    title = '打开文件'
    filePath = easygui.fileopenbox(msg, title)
    #print(filePath)
    df = pandas.read_excel(filePath)

    #df=pandas.read_excel('**电脑端账套号pandas.xlsx')
    #df = pd.read_excel(src_file, header=1, usecols='B:F')
    #header参数为一个整数，从0开始索引，其为选择的行，比如1表示Excel中的第2行。
    #usecols参数设定选择的Excel列范围范围(A-…)，例如，B:F表示读取B到F列。

    accnt_dict=dict(zip(df['账套号'],df['账套号加单位简称']))

    #print('获取字典的键，要转化为列表。第一个键是：',end='')

    account_Num= list(accnt_dict)[0]
    print(account_Num)





    year = time.strftime("%Y")
    #print("当前年份是:", year)
    #year_input = pyautogui.prompt('请输入账套查询的年份：')
    #print('弹窗输入年份：' + year_input)
    #print('不要把创建文件夹自定义函数放在循环内，先创建文件夹……')
    admin_villi_dict,folder3_1 = creat_file(year_input)
    #print('遍历3级文件夹下所有文件，检查表格上账套号对应的Excel表是否已导出')
    file_exist_list=check_file(folder3_1)

    #print('Excel文件上要导出的文件数量及文件列表是：','list_accnt_dict（要先将字典的键转化为字符串格式，两个列表的元素都是字符串才能做减法）')
    list_accnt_dict= [str(t) for t in list(accnt_dict)]
    #print(len(list_accnt_dict),list_accnt_dict)

    print('已导出的数量及文件列表', 'file_exist_list:')
    print(len(file_exist_list),file_exist_list)
    print('删除两个列表重复值，一个列表减去另一个列表,用set(集合)操作')
    accnt_list2=list(set(list_accnt_dict)-set(file_exist_list))
    print('sort影响列表本身，sorted不影响列表本身，用sort排序')
    accnt_list2 = sorted(accnt_list2)
    print('未导出的文件数量及文件列表是：')
    print(len(accnt_list2),accnt_list2)




    file_numb = 0
    for i in accnt_list2:
        #print('i在之前转化为字符串，要重新转化为数值格式')
        i=int(i)
        #print('直接用遍历字典的键值循环，i就是账套号')
        accnt_Num_name=accnt_dict[i]
        print('账套号是：' + str(i)+'，账套名称是：'+accnt_Num_name)
        time.sleep(1)
        #账套号登录自定义函数
        accnt_login(i)
        time.sleep(1)
        #用户登录自定义函数
        user_login()

        time.sleep(1)#第二次循环时，科目余额点击

        # 选项出现较慢，需要等待
        #print('点击凭证查询')
        #pyautogui.click(screenWidth/5.5652,screenHeight/1.2067)  #系统中等字体，点击三栏明细账(865,895)，凭证查询(345,895)
        pyautogui.click(screenWidth / 7.2453, screenHeight / 1.1676)  # 系统小字体显示，点击凭证查询(265,925)
        time.sleep(1)

        #print('点击序时簿chronological_book,坐标是(165,80)')
        #pyautogui.click(screenWidth/11.6364,screenHeight/13.5)系统中等字体
        #print('系统小字体显示点击序时簿chronological_book,坐标是(135,65)')
        pyautogui.click(screenWidth / 14.2222, screenHeight / 16.6154)
        time.sleep(0.5)
        select_year(screenWidth, screenHeight,year_input)
        #print('等待2秒，数据生成需要时间')
        time.sleep(3)
        #print('导出Excel表涉及到创建文件夹，创建文件，需要输入屏幕长宽、年份、账套号、账套名称、账套所在村委会字典')
        file_full_path= to_excel(screenWidth, screenHeight, year_input, accnt_Num_name, i,admin_villi_dict)
        time.sleep(3)
        excel_modifier(file_full_path)
        #print('系统中等字体显示点击另选账套')
        pyautogui.click(722,37)  #另选账套
        #account_Num+=1
        #print('系统小字体显示点击另选账套坐标135，36')
        pyautogui.click(screenWidth/3.2,screenHeight/30)  #另选账套
        time.sleep(3)#另选账套必须足够长时间，因为窗口切换幅度大
        file_numb+=1
        print('导出第{}个文件'.format(file_numb))
    #pyautogui.alert(text=r"文件保存在D:\**会计系统导出数据", title="提示")  #
if __name__ == "__main__":
    #creat_file('2019')
    print('桌面不能打开其他程序，程序在左下角运行。程序运行时不能操作鼠标和键盘，不能打开word和WPS。文件保存在D:\**会计系统导出数据.')
    #pyautogui.alert(text="桌面不能打开其他程序，程序在左下角运行。程序运行时不能操作鼠标和键盘。文件保存在D:\**会计系统导出数据",title="提示")
    time.sleep(1)

    #time.sleep(20)
    # 打开程序
    proc_list = ['D:\**经营管理系统2007版\BJACCOUNT.EXE', 'E:\**经营管理系统2007版\BJACCOUNT.EXE', 'C:\**经营管理系统2007版\BJACCOUNT.EXE',
                 'F:\**经营管理系统2007版\BJACCOUNT.EXE', 'G:\**经营管理系统2007版\BJACCOUNT.EXE','H:\**经营管理系统2007版\BJACCOUNT.EXE',
                 'I:\**经营管理系统2007版\BJACCOUNT.EXE', 'J:\**经营管理系统2007版\BJACCOUNT.EXE', 'K:\**经营管理系统2007版\BJACCOUNT.EXE',
                 'D:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE', 'E:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE',
                 'C:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE', 'F:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE',
                 'G:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE', 'H:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE',
                 'I:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE', 'J:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE',
                 'K:\Program Files\**经营管理系统2007版\BJACCOUNT.EXE',
                 'D:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE', 'E:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE',
                 'C:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE', 'F:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE',
                 'G:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE', 'H:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE',
                 'I:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE', 'J:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE',
                 'K:\Program Files (x86)\**经营管理系统2007版\BJACCOUNT.EXE',
                 ]
    for i in range(len(proc_list)):
        try:
            subprocess.Popen(proc_list[i])
            #print('程序在：{}'.format(proc_list[i]))
            break
        except:
            #print('程序不在：{}'.format(proc_list[i]))
            pass

    time.sleep(3)
    wc = uiautomation.WindowControl(searchDepth=1, Name='帐套登陆')
    # 设置为顶层
    # pyautogui.alert(text="按回车键继续程序", title="提示")  # 窗口前置才能用键盘快捷键
    #print('窗口前置才能用键盘快捷键')
    wc.SetTopmost(True)  # 置顶能用快捷键
    # accnt_Num_txt = wc.EditControl()
    # print('accnt_Num_txt',accnt_Num_txt)
    for year_input in range(2021, 2020, -1):
        year_input=str(year_input)
        print(year_input)
        func(year_input)

    open_fold()


    # file_full_path=r"D:\**会计系统导出数据\2020序时簿\2020 219三和村南泮一序时簿 - 副本.xls"
    # #del_datas(file_full_path)
    #
    # excel_modifier(file_full_path)
    # 打开程序










# '''首先，找到xlwings安装目录下的文件：你的python安装位置\Python39\Lib\site-packages\xlwings\_xlwindows.py，
#  然后，找到代码313行 self._xl=COMRetryObjectWrapper(DispatchEx('Excel.Application'))
#  默认使用的是win32com模块下面的DispatchEx方法，此方法会启动一个新进程，把DispatchEx修改为Dispatch就可以啦
#   self._xl = COMRetryObjectWrapper(Dispatch('Excel.Application'))
# 如果你用的是WPS，修改代码为： self._xl = COMRetryObjectWrapper(Dispatch('ket.Application'))
# '''
#
# def del_datas(file_full_path):
#     print('delete column datas from Excel')
#     df = pandas.read_excel(file_full_path,header=1)
#     print(df)
#     df1 = df.drop(['制单','审核','过帐'],axis=1) #删除列
#     print(df1)
#     # 其实问题理解axis有问题，也许简单的来记就是axis=0代表往跨行（down)，而axis=1代表跨列（across)，作为方法动作的副词。换句话说：
#     # 使用0值表示沿着每一列或行标签\索引值向下执行方法；使用1值表示沿着每一行或者列标签模向执行对应的方法。
#     # 轴axis用来为超过一维的数组定义的属性，二维数据拥有两个轴：第0轴沿着行的垂直往下，第1轴沿着列的方向水平延伸。
#     # 所以问题当中df.drop（‘列名’, axis=1）代表将‘列名’对应的列标签（们）沿着水平的方向依次删掉。
#     # header = None准们针对没有表头的表，这也是默认值；
#     # header = 1指定第一行作为表头；
#     for i in accnt_dict.keys():
#         print(i)
#     #nrows = df.shape[0]  # 行数