蚂蚁森林自动能量收取

之前在知乎上看到了一个UI测试框架Airtest，感觉很好。自己每天要收取蚂蚁森林能量，就想着能不能用程序实现。于是试着写来下，效果看来很好。

直接贴代码吧，很简单

from airtest.core.api import *


def unlock_device(dev):
    """
    解锁设备(图案锁屏)
    :param dev: 连接的设备
    :return:
    """
    dev.shell('input keyevent 26')
    swipe((100, 400), vector=[-0.0149, -0.9193])

    dev.minitouch.swipe_along(
        [[0, 0]])


def collect_energy(is_home=False):
    """
    收取能量，自己的和好友的
    :param is_home: 是否是自己的
    :return:
    """
    if is_home:

        if not exists(Template(r"data/mayisenlin/tpl1557374412969.png", record_pos=(0.32, -0.203), resolution=(1080, 1920))):
            return

        energy_list = find_all(Template(r"data/mayisenlin/tpl1557374412969.png", record_pos=(0.32, -0.203), resolution=(1080, 1920)))
    else:
        if exists(Template(r"data/mayisenlin/tpl1557373442966.png", record_pos=(-0.062, -0.566), resolution=(1080, 1920))):
            return
        energy_list = find_all(Template(r"data/mayisenlin/tpl1557370999417.png", record_pos=(0.034, -0.215), resolution=(1080, 1920)))

    if not energy_list:
        return
    for energy in energy_list:
        if is_home:
            touch(energy["result"])
        else:
            touch((energy["result"][0], energy["result"][1] - 10))


def find_friends_energy():
    """
    判断好友列表中是否有能量可以收取，如果发现有就收取
    :return:
    """
    energy_friends = find_all(Template(r"data/mayisenlin/tpl1557371179958.png", record_pos=(0.475, 0.15), resolution=(1080, 1920)))
    if not energy_friends:
        return
    for friend in energy_friends:
        touch(friend["result"])
        sleep(1)
        collect_energy()
        touch(Template(r"data/mayisenlin/tpl1557373711838.png", record_pos=(-0.443, -0.765), resolution=(1080, 1920)))


def start_alipy_forest(dev):
    """
    启动支付宝，并且打开蚂蚁森林，前提是蚂蚁森林在支付宝首页
    :param dev: 连接的设备
    :return:
    """
    package = 'com.eg.android.AlipayGphone'
    dev.stop_app(package)
    dev.start_app(package)
    wait(Template(r"data/mayisenlin/tpl1557371626529.png", record_pos=(-0.37, -0.141), resolution=(1080, 1920)))
    touch(Template(r"data/mayisenlin/tpl1557371626529.png", record_pos=(-0.37, -0.141), resolution=(1080, 1920)))
    sleep(1)
    wait(Template(r"data/mayisenlin/tpl1557372598598.png", record_pos=(-0.344, -0.759), resolution=(1080, 1920)))


def swipe_next(check_more=True):
    """
    滑动界面
    :param check_more: 查看更多好友
    :return:
    """
    find_friends_energy()
    if check_more:
        result = exists(
            Template(r"data/mayisenlin/tpl1557374693373.png", threshold=0.86, record_pos=(0.062, -0.115), resolution=(1080, 1920)))
        if result:
            check_more = False
            touch(result)
    if exists(Template(r"data/mayisenlin/tpl1557371700856.png", threshold=0.85, record_pos=(0.012, 0.819), resolution=(1080, 1920))):
        return
    look_more_result = exists(
        Template(r"data/mayisenlin/tpl1557379057683.png", threshold=0.85, record_pos=(0.005, 0.845), resolution=(1080, 1920)))
    if look_more_result:
        touch(look_more_result)
    swipe((100, 500), vector=[0, -0.99])
    return swipe_next(check_more=check_more)


def main():
    dev = connect_device('Android://127.0.0.1:5037/XXX')  # 填写自己的设备号
    if dev.is_locked():
        unlock_device(dev)
    start_alipy_forest(dev)
    collect_energy(is_home=True)
    swipe_next()
    dev.shell('input keyevent 26')


if __name__ == "__main__":
    main()

有一点需要注意：请用github上最新的代码

代码：https://github.com/woshimanong1990/auto_collect_alipay_antforest

如何运行脚本：

1） usb连接上手机，开启usb调试模式，具体可以百度

2） clone https://github.com/AirtestProject/Airtest代码，Python setup.py install (pip install 比较老，有bug)

3） airtest/core/android/static/adb/windows/adb.exe devices 获取手机的设备号 比如s12d45

4） unlock_device(dev)这个代码可以去掉，如果想要自行解锁，参考这个http://airtest.netease.com/docs/cn/9_faq/3_api_faq.html#id15， 主要是获取图案的坐标

5） 当成正常的Python程序运行吧

里面的参数可能需要自己调整下，比如阀值。建议先看看官方教程

 

补充：

建了一个qq群：389954854，主要目的是互帮互助，通过交流促进python技术。很多初学者往往不知道如何下手，还等什么，快快加入进来吧！

或者关注微信公众号：python码码有趣的