adb_python/adb.py

import datetime
import json
import os
import re
import shutil
import subprocess
import time
from load_things import loaded
from decode_parcel import decode_parcel
debug = True

defaults = loaded.defaults
keycodes = loaded.keycodes
exe = defaults.exe
exists = '''if [ -e "{file}" ]; then
    if [ -d "{file}" ]; then
        echo "directory"
    elif [ -f "{file}" ]; then
        echo "file"
    else
        echo "error"
    fi
else
    echo "na"
fi'''
def merge(src, dst,log = False):
    if not os.path.exists(dst):
        return False
    ok = True
    for path, dirs, files in os.walk(src):
        relPath = os.path.relpath(path, src)
        destPath = os.path.join(dst, relPath)
        if not os.path.exists(destPath):
            os.makedirs(destPath)
        for file in files:
            destFile = os.path.join(destPath, file)
            if os.path.isfile(destFile):
                if log:
                    print("Skipping existing file: " + os.path.join(relPath, file))
                ok = False
                continue
            srcFile = os.path.join(path, file)
            shutil.move(srcFile, destFile)
    for path, dirs, files in os.walk(src, False):
        if len(files) == 0 and len(dirs) == 0:
            os.rmdir(path)
    return ok

def _adb(*args,out = False):
    args = [exe] + list(args)
    if out:
        return subprocess.check_output(args,shell = False).decode().rstrip('\r\n')
    else:
        ret = subprocess.call(args,shell = False)
        if ret:
            raise subprocess.CalledProcessError(ret,args)

def get_info():
    thing = _adb("devices","-l",out = True)
    formed = list(filter(bool,thing.split("\r\n")))[1:]
    main = {}
    for device in formed:
        categories = re.split(" +",device)
        device_dict = {
            "serial":categories[0],
            "mode":categories[1]
            }

        device_dict.update(dict(category.split(":") for category in categories[2:]))
        main[categories[0]] = device_dict
    return main

class Device:
    def connect(ip,port=5555):
        if not re.match(r'(\d{1,3}\.){3}\d{1,3}',ip):
            raise TypeError("Invalid ip")
        if not all(int(n) <= 255 and int(n) >= 0 for n in ip.split('.')):
            raise TypeError("Invalid ip")
        if not (port >= 0 and port <= 2**16-1):
            raise TyperError("Port must be in the range 0-65536")
        id = '{}:{}'.format(ip,port)
        _adb('connect','{}:{}'.format(ip,port))
        dev = Device(id)
        dev.tcip = True
        return dev

    def disconnect(self):
        if self.tcip:
            _adb('disconnect',self.serial)
        

    def prim_device():
        cont = True
        while cont:
            try:
                d = Device()
                cont = False
            except IndexError:
                time.sleep(1)
        return d

    def __init__(self,serial=None):
        self.tcip = False
        if serial:
            self.serial = serial
            info = get_info()[serial]
        else:
            serial,info = list(get_info().items())[0]
        self.__dict__.update(info)



    def adb(self,*args,out = False):
        args = ['-s',self.serial]+ list(args)
        return _adb(*args,out = out)
    def shell(self,*args,out=False):
        args = ('shell',)+args
        return self.adb(*args,out=out)
    def sudo(self,*args,out = False):
        if self.mode == 'recovery':
            return self.shell(*args,out=out)
        else:
            args = '"{}"'.format(' '.join(args).replace('"','\\"'))
            return self.shell('su','-c',args,out = out)



    def type(self,file):
        
        e = exists.format(file = file)
        res = self.sudo(e,out=True)
        return res
    def exists(self,file):
        return self.type(file) != "na"
    def isfile(self,file):
        return self.type(file) == 'file'
    def isdir(self,file):
        return self.type(file) == 'directory'

    def delete(self,path):
        return self.sudo("rm","-rf",path,out=True)

    def copy(self,remote,local,del_duplicates = True,ignore_error=True):
        remote_type = self.type(remote)
        if remote_type != "na":
            if remote_type == "directory" and not remote.endswith('/'):
                remote += '/'
            merge_flag = False
            if os.path.exists(local):
                last = os.path.split(local)[-1]
                real_dir = local
                local = os.path.join(defaults['local']['temp'],last)
                merge_flag = True
            try:
                self.adb("pull","-a",remote,local)
            except subprocess.CalledProcessError as e:
                if ignore_error:
                    pass
                else:
                    raise e
            if merge_flag:
                merge(local,real_dir)
                if os.path.exists(local) and del_duplicates:
                    shutil.rmtree(local)
        else:
            print("File not found: {}".format(remote))

    def move(self,remote,local,del_duplicates = True,ignore_error=False):
        if self.exists(remote):
            self.copy(remote,local,del_duplicates = del_duplicates,ignore_error=ignore_error)
            self.delete(remote)
        else:
            print("File not found: {}".format(remote))

    def push(self,local,remote):
        self.adb('push',local,remote)



    def reboot(self,mode = None):
        if mode:
            if mode == "soft":
                if self.mode != 'recovery':
                    pid = self.shell("pidof","zygote",out = True)
                    return self.sudo("kill",pid,out=False)
                else:
                    return self.reboot()

            else:
                self.adb("reboot",mode)
        else:
            self.adb("reboot")
        while True:
            infos = get_info()
            if len(infos) > 0:
                self.__dict__.update(infos[self.serial])
                break
            time.sleep(1)

    def send_keycode(self,code):
        try:
            keycode = keycodes[code]
        except KeyError:
            keycode = str(code)
        self.shell("input","keyevent",keycode)

    def unlock_phone(self,password):
        if self.mode != 'recovery':
            if not decode_parcel(self.shell('service','call', 'power','12',out=True),'int'):
                self.send_keycode('power')
            if decode_parcel(self.shell('service','call','trust','7',out=True),'int'):
                self.send_keycode('space')
                self.shell("input","text",str(password))
                self.send_keycode('enter')



    def backup(self,*partitions,name = None):
        if self.mode != 'recovery':
            self.reboot('recovery')
        backupdir = defaults['local']['TWRP']
        options_dict = {
            "system": "S",
            "data": "D",
            "cache": "C",
            "recovery": "R",
            "spec_part_1": "1",
            "spec_part_2": "2",
            "spec_part_3": "3",
            "boot": "B",
            "as": "A"
        }
        options = "".join(options_dict[option] for option in partitions)

        if not name:
            name = "backup_"+datetime.datetime.today().strftime(defaults['date_format'])

        filename = os.path.join(backupdir,name)
        self.shell("twrp","backup",options,name)
        phone_dir = "/data/media/0/TWRP/BACKUPS/{serial}/{name}".format(serial = self.serial,name = name)
        self.move(phone_dir,filename)

    def wipe(self,partition):
        if self.mode != 'recovery':
            self.reboot('recovery')
        self.shell("twrp","wipe",partition)

    def install(self,name):
        if self.mode != 'recovery':
            self.reboot('recovery')
        if os.path.exists(name):
            local_name = name
            name = os.path.split(name)[-1]
            update_path = '{}/{}'.format(defaults['remote']['updates'],name)
            if not self.exists(update_path):
                self.push(local_name,defaults['remote']['updates'])

        else:
            update_path = '{}/{}'.format(defaults['remote']['updates'],name)
        self.shell("twrp","install",update_path)


if __name__ == "__main__" and debug:
    d = Device.prim_device()