wechat_jump.py

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from PIL import Image
import math
import time
import os

def pull_screenshot(name = 1):
    os.system('adb shell screencap -p /sdcard/{}.png'.format(name))
    os.system('adb pull /sdcard/{}.png .'.format(name))
    os.system('mv {}.png images/{}.png'.format(name,name))
    

def jump(distance):
    press_time = distance * 1.35
    press_time = int(press_time)
    cmd = ('adb shell input swipe 320 410 320 410 ' + str(press_time))
    print (cmd)
    os.system(cmd)

fig = plt.figure()

pull_screenshot()
img = np.array(Image.open(os.path.join('images','1.png')))


click_count = 0
cor = []
name = 1
def update_data(name):
    return np.array(Image.open(os.path.join('images','{}.png'.format(name))))

im = plt.imshow(img, animated=True)


def updatefig(*args):
    global name
    name += 1
    pull_screenshot(name = name)
    im.set_array(update_data(name))
 
    return im,

def onClick(event):      
   
    global ix, iy
    global click_count
    global cor

    # next screenshot
    
    ix, iy = event.xdata, event.ydata
    coords = []
    coords.append((ix, iy))
    print ('now = ', coords)
    cor.append(coords)
    
    click_count += 1
    if click_count > 1:
        click_count = 0
        
        cor1 = cor.pop()
        cor2 = cor.pop()

        distance = (cor1[0][0] - cor2[0][0])**2 + (cor1[0][1] - cor2[0][1])**2 
        distance = distance ** 0.5
        print ('distance = ', distance)
        jump(distance)
        
        
fig.canvas.mpl_connect('button_press_event', onClick)
ani = animation.FuncAnimation(fig, updatefig, interval=50, blit=True)
plt.show()