import os
import sys
import time

import uiautomator2 as u2


def spider_poc(serial):
    d = u2.connect(serial)
    d.uiautomator.start()
    d.screen_on()
    d.unlock()
    d.debug = False

    d.app_stop("com.instagram.android")
    d.app_start("com.instagram.android", use_monkey=True)

    time.sleep(3)

    root_dir = os.path.dirname(sys.argv[0])
    screenshot_dir = os.path.join(root_dir, "screenshots")
    d.screenshot().save(screenshot_dir)


if __name__ == '__main__':
    spider_poc("5D7HSG5XXCKVBI85")