mobile-rpa/control/plat/deepcoin_control.py

# from utils.point_utils import UiFactory
import datetime
import logging
import tkinter as tk

from control.base.base_control import AbsControl


class DeepCoinControl(AbsControl):

    def __init__(self, serial: str, *args, **kwargs):
        super().__init__(serial, *args, **kwargs)

        self.package_name = 'com.niocpeed.dna'

        # self.add_func(self.btn_jiao_yi)
        # self.add_func(self.btn_mairu_kaiduo)
        # self.add_func(self.btn_shijia_quanping)

    def btn_jiao_yi(self):
        return self.get_point({
            "name": "btn_jiao_yi",
            "desc": "交易按钮",
            "xpath": '//*[@content-desc="交易"]'
        })

    def btn_mairu_kaiduo(self):
        return self.get_point({
            "name": "btn_mairu_kaiduo",
            "desc": "买入/开多",
            "xpath": '//*[@content-desc="买入/开多"]'
        })

    def btn_mairu_pingkong(self):
        return self.get_point({
            "name": "btn_mairu_pingkong",
            "desc": "买入/平空",
            "xpath": '//*[@content-desc="买入/平空"]'
        })

    def btn_shijia_quanping(self):
        self.to_next_swipe()
        return self.get_point({
            "name": "btn_shijia_quanping",
            "desc": "买入/开多",
            "xpath": '//*[@content-desc="市价全平"]'
        })

    def btn_kaichang(self):
        return self.get_point({
            "name": "btn_kaichang",
            "desc": "开仓",
            "xpath": '//*[@content-desc="开仓"]'
        })

    def btn_gengdan(self):
        """
        跟单(n) 按钮 坐标
        """
        return self.get_point({
            "name": "btn_gengdan",
            "desc": "跟单",
            "xpath": '//*[starts-with(@content-desc, "跟单(")]'
        })

    def btn_weituo(self):
        """
        委托(n) 按钮 坐标
        FIXME 委托按钮点击2次回出现切换委托界面

        """
        return self.get_point({
            "name": "btn_weituo",
            "desc": "委托",
            "xpath": '//*[starts-with(@content-desc, "委托(")]'
        })

    def btn_yijian_chedan(self):
        """
        一键撤单按钮
        """
        return self.get_point({
            "name": "btn_yijian_chedan",
            "desc": "一键撤单",
            "xpath": '//*[@content-desc="一键撤单"]'
        })

    def slider_baozhengjin_kaicang(self, offset=0, sleep=0.1):
        """
      开仓情况  保证金滑块
           :param sleep 休眠
           :param offset 价格偏移量 默认取50% x=368 每偏移69 向下多一个 偏移-69向上多一个 取值范围 [-2,2]
        坐标： 0% 230  25% 299  50% 368  75% 437  100% 506
        :return: 买入的按钮实例
        """
        # self.to_top_swipe()
        # self.d.sleep(0.1)

        # 基于买入按钮相对计算坐标
        x, y, btn = self.btn_mairu_kaiduo()

        x1 = x + offset * 69
        y1 = y - 155
        logging.info(f"开多坐标: ({x},{y})")
        logging.info(f"保证金50%坐标: ({x1},{y1})")

        self.click_point(x1, y1)
        self.d.sleep(sleep)

        return btn

    def slider_baozhengjin_pingcang(self, offset=0, sleep=0.1):
        """
            平仓情况  保证金滑块
           :param sleep 休眠
           :param offset 价格偏移量 默认取50% x=368 每偏移69 向下多一个 偏移-69向上多一个 取值范围 [-2,2]
        坐标： 0% 230  25% 299  50% 368  75% 437  100% 506
        :return: 买入的按钮实例
        """

        # 基于买入按钮相对计算坐标
        x, y, btn = self.btn_mairu_pingkong()
        x1 = x + offset * 69
        y1 = y - 70
        logging.info(f"开多坐标: ({x},{y})")
        logging.info(f"保证金50%坐标: ({x1},{y1})")

        self.click_point(x1, y1)
        self.d.sleep(sleep)

        return btn

    def click_kaicang(self):
        """
        开仓
        """
        self.click_xpath('//*[@content-desc="开仓"]')

    def click_pingchang(self, sleep=0.1):
        """
        开仓
        :param sleep: 等待时间
        """
        self.click_xpath('//*[@content-desc="平仓"]')
        self.d.sleep(sleep)

    def select_kaicang_weituo_xianjia(self):
        """
        开仓选择限价委托
        """
        x, y, _ = self.btn_kaichang()
        logging.info(f"开仓按钮坐标: ({x},{y})")
        x1 = x + 100
        y1 = y + 70
        logging.info(f"选择委托框: ({x1},{y1})")
        self.click_point(x1, y1)
        x2 = x1
        y2 = y1 + 150
        self.d.sleep(0.2)
        logging.info(f"选择限价委托: ({x2},{y2})")
        self.click_point(x2, y2)

    def click_price_red(self, offset=0):
        """
        点击价格 开多 平空， 点击红色价格

        :param offset 价格偏移量 默认取第5个， 每偏移20 向下多一个 偏移-20 向上多一个 取值范围 [-4,4]
        """

        x1 = 50
        y1 = 450 + offset * 20
        logging.info(f"红色部分选择价格: ({x1},{y1})")
        self.click_point(x1, y1)

    def click_price_green(self, offset=0):
        """
        点击价格 开空 平多， 点击红色价格
        :param offset 价格偏移量 默认取第5个， 每偏移20 向下多一个，偏移-20 向上多一个
        """

        # self.select_kaicang_weituo_xianjia()
        #
        # self.d.sleep(0.1)

        x1 = 50
        y1 = 515 + offset
        logging.info(f"绿色部分选择价格: ({x1},{y1})")
        self.click_point(x1, y1)

    def click_mairu_kaiduo(self):
        """
        买入开多
        """
        self.click_xpath('//*[@content-desc="买入/开多"]')
        self.d.sleep(0.1)
        # 弹出了确认框
        if self.check_element_exists('//*[@content-desc="开仓确认"]'):
            logging.info("弹出了确认框")
            self.click_xpath('//*[@content-desc="确认"]')

    def click_mairu_pingkong(self):
        """
        买入平空
        """
        self.click_xpath('//*[@content-desc="买入/平空"]')

    def click_maichu_kaikong(self):
        """
        卖出开空
        """
        self.click_xpath('//*[@content-desc="卖出/开空"]')

        self.d.sleep(0.1)
        # 弹出了确认框
        if self.check_element_exists('//*[@content-desc="开仓确认"]'):
            logging.info("弹出了确认框")
            self.click_xpath('//*[@content-desc="确认"]')

    def click_maichu_pingduo(self):
        """
        卖出平多
        """
        self.click_xpath('//*[@content-desc="卖出/平多"]')

    def text_xiadan_shuliang(self, text):
        """
        设置 下单数量
        """
        x, y, _ = self.get_point({
            "name": "btn_baozhengjin",
            "desc": "保证金",
            "xpath": '//*[@content-desc="保证金"]'
        })
        logging.info(f"保证金坐标: ({x},{y})"'')
        x1 = x - 130
        y1 = y + 50
        self.input_by_position(x1, y1, str(text), True)

    def event_1(self):
        self.click_kaicang()
        self.slider_baozhengjin_kaicang()
        pass

    def event_2(self, **kwargs):
        """
        开仓情况  保证金滑块
        :return:
        """

        self.to_top_swipe(sleep=0.2)
        self.click_kaicang()
        self.select_kaicang_weituo_xianjia()
        self.d.sleep(0.1)

        self.click_price_red()

        self.text_xiadan_shuliang(kwargs["shuliang"])
        # self.slider_baozhengjin_kaicang()

        self.click_mairu_kaiduo()
        pass

    def event_3(self, **kwargs):
        self.to_top_swipe(sleep=0.2)
        self.click_pingchang()
        self.slider_baozhengjin_pingcang(offset=1)
        pass

    def event_4(self, **kwargs):
        self.to_top_swipe(sleep=0.2)
        self.click_pingchang()
        self.select_kaicang_weituo_xianjia()
        self.click_price_green()
        self.text_xiadan_shuliang(kwargs["shuliang"])
        self.to_next_swipe(y=0.6)
        # self.slider_baozhengjin_pingcang(offset=1)
        self.click_maichu_pingduo()
        pass

    def event_5(self, **kwargs):
        self.to_top_swipe(sleep=0.2)
        self.click_kaicang()
        self.slider_baozhengjin_kaicang()
        pass

    def event_6(self, **kwargs):
        self.to_top_swipe(sleep=0.1)
        self.click_kaicang()
        self.select_kaicang_weituo_xianjia()
        self.click_price_red()
        # self.slider_baozhengjin_kaicang()
        self.text_xiadan_shuliang(kwargs["shuliang"])
        self.to_next_swipe(y=0.6)
        self.click_maichu_kaikong()

        pass

    def event_7(self, **kwargs):
        self.to_top_swipe(sleep=0.1)
        self.click_pingchang()
        self.slider_baozhengjin_pingcang(offset=2)
        pass

    def event_8(self, **kwargs):
        self.to_top_swipe(sleep=0.2)

        self.click_pingchang()

        self.click_price_red()

        self.text_xiadan_shuliang(kwargs["shuliang"])
        # self.slider_baozhengjin_pingcang(offset=2)

        self.click_mairu_pingkong()
        pass

    def event_9(self):
        """
        一键撤单
        如果有滑动 ，会导致第一个坐标不准，点不上。 所以要提前滑动，确保第一个坐标点得上
        """
        self.to_top_swipe(sleep=0.1)
        self.to_next_swipe(sleep=0.1)
        logging.info("点击跟单")
        self.click_xpath('//*[starts-with(@content-desc, "跟单(")]')
        self.d.sleep(0.1)
        logging.info("点击委托")
        self.click_xpath('//*[starts-with(@content-desc, "委托(")]')
        self.d.sleep(0.1)

        b = self.check_element_exists('//*[starts-with(@content-desc, "全部委托(")]')
        logging.info("是否弹出了选择框: %s", b)

        if b:
            self.click_xpath('//*[starts-with(@content-desc, "全部委托(")]')
            self.d.sleep(0.2)

        self.click_xpath('//*[@content-desc="一键撤单"]')
        self.d.sleep(0.1)

        b = self.check_element_exists('//*[@content-desc="撤销掉全部委托"]')

        if b:
            x, y, _ = self.get_point({
                "name": "btn_yijian_chedan",
                "desc": "撤销掉全部委托",
                "xpath": '//*[@content-desc="撤销掉全部委托"]'
            })
            logging.info(f"撤销掉全部委托: ({x},{y})")
            #  相对坐标 撤销掉全部委托 点击确定。
            x1 = x + 80
            y1 = y + 80
            logging.info(f"撤销掉全 确定按钮: ({x1},{y1})")
            self.click_point(x1, y1)

    def event_10(self, **kwargs):
        logging.info(f"检查交易结果")
        self.to_top_swipe(sleep=0.2)
        self.to_next_swipe(sleep=0.2)

        log = kwargs.get("log")
        action = kwargs.get("action", "")

        # 获取持仓
        t1 = self.get_content_desc('//*[starts-with(@content-desc, "持仓(")]')
        logging.info(f"获取持仓:{t1}")
        # 获取委托
        t2 = self.get_content_desc('//*[starts-with(@content-desc, "委托(")]')
        logging.info(f"获取委托:{t2}")
        # 获取跟单
        t3 = self.get_content_desc('//*[starts-with(@content-desc, "跟单(")]')
        logging.info(f"获取跟单:{t3}")
        # 获取策略
        t4 = self.get_content_desc('//*[starts-with(@content-desc, "策略(")]')
        logging.info(f"获取策略:{t4}")

        # 获取当前日期和时间
        now = datetime.datetime.now()

        # 格式化日期和时间为 "年月日-月月-日日 时时:分分:秒秒"
        formatted_time = now.strftime("%Y-%m-%d %H:%M:%S")

        log.insert(tk.INSERT, f"======执行结果============\n"
                              f"手机:{self.ctx['name']}\n"
                              f"操作:{action}\n"
                              f"时间:{formatted_time}\n"
                              f"持仓:{t1}\n"
                              f"委托:{t2}\n"
                              f"跟单:{t3}\n"
                              f"策略:{t4}\n"
                              f"===========================\n")
        log.see(tk.END)
        self.to_top_swipe()

    def event_11(self, **kwargs):
        """
        一键全平
        :param kwargs:
        :return:
        """
        self.to_top_swipe(sleep=0.1)
        self.to_next_swipe(sleep=0.1)
        logging.info("点击跟单")
        self.click_xpath('//*[starts-with(@content-desc, "跟单(")]')
        self.d.sleep(0.1)
        logging.info("点击委托")
        self.click_xpath('//*[starts-with(@content-desc, "持仓(")]')
        self.d.sleep(0.1)
        self.click_xpath('//*[@content-desc="一键全平"]')
        self.d.sleep(0.1)
        self.click_xpath('//*[@content-desc="确认"]')
        self.d.sleep(0.1)