Das deutsche Python-Forum

Code: Alles auswählen

        self.setup_environment()

       .... do something with virtual box

      clean()

Code: Alles auswählen

        #store the original environment
        try:
            self.__old_user_home = os.environ['VBOX_USER_HOME']
        except KeyError:
            pass

        #set the VBOX_USER_HOME
        os.environ['VBOX_USER_HOME'] = self.__test_suite.get("General", "TEST_SUITE_VM_SETTING_FOLDER")

        #create the vbox_user_home folder, if not exist
        if not os.path.exists(os.environ['VBOX_USER_HOME']):
            os.makedirs(os.environ['VBOX_USER_HOME'])

        #save the orginal machine store of virtual box
        p1 = subprocess.Popen(["VBoxManage", "list", "systemproperties"],
                            env=os.environ.copy(),
                            stdout=subprocess.PIPE)
        p2 = subprocess.Popen(["grep", "Default machine folder:"],
                            env=os.environ.copy(),
                            stdout=subprocess.PIPE,
                            stdin=p1.stdout)
        p1.stdout.close()

        out = p2.communicate()[0]
        self.__old_machine_home = out.split(":")[1].strip()

        #set the machine store of virtual box
        set_command = ["VBoxManage", "setproperty", "machinefolder",
                        self.__test_suite.get("General", "TEST_SUITE_VM_STORE_FOLDER") ]
        subprocess.call(set_command, env=os.environ.copy())

        #create the machine store folder, if not exist
        if not os.path.exists(self.__test_suite.get("General", "TEST_SUITE_VM_STORE_FOLDER")):
            os.makedirs(self.__test_suite.get("General", "TEST_SUITE_VM_STORE_FOLDER"))

Code: Alles auswählen

    def clean(self):
...
            if os.path.exists(self.__test_suite.get("General", "TEST_SUITE_VM_STORE_FOLDER")):
            shutil.rmtree(self.__test_suite.get("General", "TEST_SUITE_VM_STORE_FOLDER"))

        if os.path.exists(self.__test_suite.get("General", "TEST_SUITE_VM_SETTING_FOLDER")):
            shutil.rmtree(self.__test_suite.get("General", "TEST_SUITE_VM_SETTING_FOLDER"))

Code: Alles auswählen

        #create the vbox_user_home folder, if not exist
        if not os.path.exists(os.environ['VBOX_USER_HOME']):
            os.makedirs(os.environ['VBOX_USER_HOME'])

Code: Alles auswählen

VBOX_MANAGE = 'VBoxManage'


def vbox_get_property(name):
    name += ':'
    process = subprocess.Popen(
        [VBOX_MANAGE, 'list', 'systemproperties'], stdout=subprocess.PIPE
    )
    result = None
    for line in process.stdout:
        if line.startswith(name):
            result = line[len(name):]
    if result is None:
        raise KeyError('property {0!r} not found'.format(name[:-1]))
    return result.strip()


def vbox_set_property(name, value):
    # 
    # TODO Maybe use `subprocess.check_call` and analyse the return code.
    # 
    subprocess.call([VBOX_MANAGE, 'setproperty', name, str(value)])

Code: Alles auswählen

try:
    ...rmtree...
except Exception as error:
    print 'rmtree:', error
    raise