Python:minicap实现快速截图

一：minicap简介

minicap属于STF框架的⼀个⼯具，可以⾼速截图、同步⼿机屏幕⾄浏览器等功能，经过试验，截同⼀个屏幕的⼀张图，使⽤adb shell screencap命令进⾏截图速度为2.9秒，⽽minicap仅为0.8秒，效率⾼很多。

import os
import socket
import sys
import time
import struct
from collections import OrderedDict



class Banner:
    def __init__(self):
        self.__banner = OrderedDict(
            [('version', 0),
             ('length', 0),
             ('pid', 0),
             ('realWidth', 0),
             ('realHeight', 0),
             ('virtualWidth', 0),
             ('virtualHeight', 0),
             ('orientation', 0),
             ('quirks', 0)
             ])

    def __setitem__(self, key, value):
        self.__banner[key] = value

    def __getitem__(self, key):
        return self.__banner[key]

    def keys(self):
        return self.__banner.keys()

    def __str__(self):
        return str(self.__banner)


class Minicap:
    def __init__(self, host, port, banner):
        self.buffer_size = 4096
        self.host = host
        self.port = port
        self.banner = banner

    def connect(self):
        try:
            self.__socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        except (socket.error) as e:
            # print(e)
            sys.exit(1)
        self.__socket.connect((self.host, self.port))

    def on_image_transfered(self, data):
        # file_name = str(time.time()) + '.jpg'  # 图片名
        file_name = '1' + '.jpg'  # 图片名
        with open(file_name, 'wb') as f:
            for b in data:
                f.write((b).to_bytes(1,'big'))

    def consume(self):
        readBannerBytes = 0
        bannerLength = 24
        readFrameBytes = 0
        frameBodyLength = 0
        data = []
        while True:
            try:
                chunk = self.__socket.recv(self.buffer_size)
            except (socket.error) as e:
                # print(e)
                sys.exit(1)
            cursor = 0
            buf_len = len(chunk)
            while cursor < buf_len:
                if readBannerBytes < bannerLength:
                    map(lambda i, val: self.banner.__setitem__(self.banner.keys()[i], val),
                        [i for i in range(len(self.banner.keys()))], struct.unpack("<2b5ibB", chunk))
                    cursor = buf_len
                    readBannerBytes = bannerLength
                    # print(self.banner)
                elif readFrameBytes < 4:
                    # print(struct.unpack('B', (chunk[cursor]).to_bytes(1,'big')))
                    # frameBodyLength += (struct.unpack('B', chunk[cursor])[0] << (readFrameBytes * 8)) >> 0
                    frameBodyLength += (chunk[cursor] << (readFrameBytes * 8)) >> 0
                    cursor += 1
                    readFrameBytes += 1
                else:
                    # print("frame length:{0} buf_len:{1} cursor:{2}".format(frameBodyLength, buf_len, cursor))
                    # pic end
                    if buf_len - cursor >= frameBodyLength:
                        data.extend(chunk[cursor:cursor + frameBodyLength])
                        self.on_image_transfered(data)
                        cursor += frameBodyLength
                        frameBodyLength = readFrameBytes = 0
                        data = []
                    else:
                        data.extend(chunk[cursor:buf_len])
                        frameBodyLength -= buf_len - cursor
                        readFrameBytes += buf_len - cursor
                        cursor = buf_len



if '__main__' == __name__:
    # 手动打开调试模式
    # 推送minicap
    path = os.popen('adb shell ls /data/local/tmp/minicap').read().strip()
    print(path)
    if 'No such file or directory' in path:
        os.popen('uiautomator2 init')  # 初始化uiautomator2，用于安装 minicap 和 atx-agent
        time.sleep(2)
    os.popen('adb forward tcp:1717 localabstract:minicap')  # 执行了adb端口转发
    os.popen('adb shell LD_LIBRARY_PATH=/data/local/tmp /data/local/tmp/minicap -P 1920x720@1920x720/0')  # 启动了minicap服务
    time.sleep(1)
    mc = Minicap('localhost', 1717, Banner())
    mc.connect()
    mc.consume()

一.第一次运行时，需要初始化uiautomator2用于推送minicap 和 atx-agent到设备。

二.运行时，会在同目录下生成图片

三.

file_name = str(time.time()) + '.jpg' # 图片名，会生成每次界面刷新的图片

file_name = '1' + '.jpg' # 图片名，只保留最后一次刷新的图片

四.

os.popen('adb shell LD_LIBRARY_PATH=/data/local/tmp /data/local/tmp/minicap -P 1920x720@1920x720/0') # 启动了minicap服务

1920x720是我连接设备屏幕尺寸，这里根据自己连接设备尺寸来修改