import os
from appium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
from time import sleep

class Auto():
    driver = None

    def open_weixin(self, host='localhost', port=4723):
        # 打开WinAppDriver服务
        # 注意：如果手动开启，则可以注释掉
        os.system(r'start "" /d "C:\Program Files (x86)\Windows Application Driver\"  "WinAppDriver.exe"')

        # 配置信息
        # 包含：平台名、系统、应用程序绝对路径
        desired_caps = {'platformName': 'Windows', 'deviceName': 'WindowsPC',
                        'app': r"C:\Program Files (x86)\Tencent\WeChat\WeChat.exe"}

        try:
            # 连接WinAppDriver服务，打开目标软件
            self.driver = webdriver.Remote('http://{}:{}'.format(host, port), desired_caps)
        except Exception as e:
            raise AssertionError(e)

    # 给文件传输助手发送一条信息
    def send_msg(self, element_name, msg):
        """
        :param element_name:元素name值
        :param msg:
        :return:
        """
        # 通过name属性，找到目标元素
        chat_element = self.driver.find_element_by_name(element_name)

        # 点击元素，进入聊天界面
        chat_element.click()

        # 找到输入框，并输入
        self.driver.find_element_by_name("输入").send_keys(msg)

        # 点击右下角的发送，发送消息出去
        # self.driver.find_element_by_name("发送(S)").click()
        # sendBtn
        self.driver.find_element_by_name("sendBtn").click()

    # 释放资源及关闭服务
    def tearDownFunc(self):
        print("准备退出")
        sleep(2)

        # 1、释放资源
        self.driver.quit()

        # 2、关闭WinAppDriver应用程序
        os.system(' @taskkill /f /im WinAppDriver.exe')


if __name__ == '__main__':
    auto = Auto()
    auto.open_weixin()
    auto.send_msg("文件传输助手", "我是自动发送的哈哈")
    auto.tearDownFunc()