import subprocess
import shlex

class AdbWrapper:

    def __init__(self,
        executable_path,
        serial = None
    ):
        self.executable_path = executable_path
        self.serial = serial
        self.selfrooted = False

    def exec(self,comspec,output_streams = False):
        if isinstance(comspec,str):
            comspec = shlex.split(comspec)

        comspec.insert(0,self.executable_path)
        if self.serial:
            comspec = comspec[:1] + ['-s',self.serial] + comspec[1:]

        if output_streams:
            res = subprocess.run(comspec,capture_output=True,check=True)
            return res.stdout

        else:
            ret = subprocess.check_call(comspec,shell=False)

    def shell(self,comspec,output_streams = False):
        if isinstance(comspec,str):
            comspec = shlex.split(comspec)
        comspec.insert(0,'exec-out')
        return self.exec(comspec,output_streams)

    def sudo(self,comspec,output_streams = False):
        if not self.rooted:
            comspec = ['su','--','--'] + comspec
        return self.shell(comspec,output_streams)

    def root(self):
        self.exec('root')
        self.rooted = True

    def unroot(self):
        self.exec('unroot')
        self.rooted = False

if __name__ == "__main__":
    exec_path = r"C:\Program Files\platform-tools\adb.exe"
    wrapper = AdbWrapper(exec_path,'LGD415d60b8c9b')
    wrapper.root()
    wrapper.sudo(['ls','/'])