import uiautomator2 as u2

d = u2.connect()

# 启动应用（以微信为例）
d.app_start("com.tencent.mm")

# 截微信图保存
d.screenshot("home.jpg")

# 关闭应用
d.app_stop("com.tencent.mm")