#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
通過標(biāo)簽清理微信好友
"""
import time
import uiautomator2 as u2

class WXClearFriendByTag:
    """
    通過標(biāo)簽清理微信好友
    """
    def __init__(self, tag_name: str, max_times_tag: int = 5, max_times_friend: int = 5):
        self.d = u2.connect()  # 連接設(shè)備
        self.clear_tag_name = tag_name  # 需要被清理好友的標(biāo)簽名
        self.max_times_tag = max_times_tag  # 查找指定標(biāo)簽的最大循環(huán)次數(shù)
        self.max_times_friend = max_times_friend  # 清理好友的最大循環(huán)次數(shù)

def enable_debug(self):
    """
    開啟調(diào)試模式
    :return:
    """
    print(f"設(shè)備信息 ==> {self.d.info}")
    print(f"設(shè)備IP ==> {self.d.wlan_ip} 設(shè)備號 ==> {self.d.serial}")
    self.d.implicitly_wait(30)  # 設(shè)置查找元素等待時間，單位秒

def go_to_tag_page(self):
    """
    進(jìn)入標(biāo)簽頁面
    :return:
    """
    print("打開【通訊錄】")
    self.d(text='通訊錄', className='android.widget.TextView').click_exists(timeout=3)
    print("打開【標(biāo)簽】")
    self.d(text='標(biāo)簽', className='android.widget.TextView').click_exists(timeout=3)

def find_tag_click(self):
    i = 1
    while i <= self.max_times_tag:
        print(f"第{i}次查詢")
        status = self.find_tag_by_name()
        if status == 'done':
            break
        else:
            self.d.swipe(100, 1000, 100, 200)
            i += 1
    if i > self.max_times_tag:
        print(f"沒有找到標(biāo)簽：{self.clear_tag_name}")
        exit()

    tag = self.d(text=self.clear_tag_name)
    if tag.exists() is False:
        print(f"沒有找到標(biāo)簽：{self.clear_tag_name}")
        exit()

    print(f"找了{(lán)i}次，找到了標(biāo)簽：{self.clear_tag_name} 點擊")
    tag.click_exists(timeout=5)

def find_tag_by_name(self):
    run_status = 'doing'
    elems = self.d(resourceId="com.tencent.mm:id/hs8")
    for elem in elems:
        tag_name = elem.get_text(timeout=10)
        friend_count = elem.sibling(resourceId="com.tencent.mm:id/hs7").get_text(timeout=10)
        print(f"標(biāo)簽是：{tag_name}，好友數(shù)：{friend_count}")
        if tag_name == self.clear_tag_name:
            run_status = 'done'
            break
    return run_status

def clear_every_friend(self):
    """
    清理每個好友
    :return:
    """
    time.sleep(3)
    elems = self.d(resourceId='com.tencent.mm:id/kbq')
    if elems.exists() is False:
        print("沒有找到好友")
        exit()

    for elem in elems:
        time.sleep(1)
        friend_nickname = elem.get_text(timeout=10)
        print(f'進(jìn)入好友詳情頁面 --> {friend_nickname}')

        elem.click(timeout=5)

        self.d(resourceId='com.tencent.mm:id/coy').click(timeout=5)
        time.sleep(1)
        self.d(text='刪除').click(timeout=5)
        time.sleep(3)
        self.d(text='刪除', resourceId='com.tencent.mm:id/mm_alert_ok_btn').click(timeout=5)
        time.sleep(1)

def clear_friends_in_tag(self):
    """
    在標(biāo)簽中清理好友
    :return:
    """
    i = 1
    while i <= self.max_times_friend:
        print(f"第{i}次清理好友")
        self.clear_every_friend()
        i += 1

    print(f"清理了{(lán)i}次好友")

if __name__ == '__main__':
    tag_name = "清粉-賬號問題"
    wx = WXClearFriendByTag(tag_name)
    wx.enable_debug()
    wx.find_tag_click()
    wx.clear_friends_in_tag()