Python os.rmdir方法代码示例

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_prepare(self):
        p = get_data_folder()
        api = FileAPI(p)

        folders = ["one", "two"]
        relative = "/".join(folders) + "/three.txt"
        api.prepare_dir(relative)

        rmdirs = []
        for f in folders:
            p = p + "/" + f
            self.assertTrue(os.path.exists(p))
            rmdirs.append(p)

        self.assertFalse(os.path.exists(api.to_abs(relative)))

        for f in rmdirs[::-1]:
            os.rmdir(f) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def get_cifar10(data_dir):
    if not os.path.isdir(data_dir):
        os.system("mkdir " + data_dir)
    cwd = os.path.abspath(os.getcwd())
    os.chdir(data_dir)
    if (not os.path.exists('train.rec')) or \
       (not os.path.exists('test.rec')) :
        import urllib, zipfile, glob
        dirname = os.getcwd()
        zippath = os.path.join(dirname, "cifar10.zip")
        urllib.urlretrieve("http://data.mxnet.io/mxnet/data/cifar10.zip", zippath)
        zf = zipfile.ZipFile(zippath, "r")
        zf.extractall()
        zf.close()
        os.remove(zippath)
        for f in glob.glob(os.path.join(dirname, "cifar", "*")):
            name = f.split(os.path.sep)[-1]
            os.rename(f, os.path.join(dirname, name))
        os.rmdir(os.path.join(dirname, "cifar"))
    os.chdir(cwd)

# data 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_create_new_folder_parallel():
    folder = os.path.join('output', str(uuid.uuid4()))

    def run_makedirs(folder, exceptions):
        try:
            makedirs_safe(folder)
        except Exception as exc:
            exceptions.put(exc)

    for _ in range(5):
        exceptions = queue.Queue()
        thread1 = threading.Thread(target=run_makedirs, args=(folder, exceptions))
        thread2 = threading.Thread(target=run_makedirs, args=(folder, exceptions))
        thread1.start()
        thread2.start()
        thread1.join()
        thread2.join()

        assert exceptions.qsize() == 0
        assert os.path.isdir(folder)
        os.rmdir(folder) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def _delete_measures_files_for_metric(self, metric_id, files):
        for f in files:
            try:
                os.unlink(self._build_measure_path(metric_id, f))
            except OSError as e:
                # Another process deleted it in the meantime, no prob'
                if e.errno != errno.ENOENT:
                    raise
        try:
            os.rmdir(self._build_measure_path(metric_id))
        except OSError as e:
            # ENOENT: ok, it has been removed at almost the same time
            #         by another process
            # ENOTEMPTY: ok, someone pushed measure in the meantime,
            #            we'll delete the measures and directory later
            # EEXIST: some systems use this instead of ENOTEMPTY
            if e.errno not in (errno.ENOENT, errno.ENOTEMPTY, errno.EEXIST):
                raise 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_prepare_find_m2m(self, sphere3_fn):
        l = sim_struct.LEADFIELD()
        l.pathfem = ''
        l.fnamehead = sphere3_fn
        path, n = os.path.split(sphere3_fn)
        dir_fn = os.path.join(path, 'm2m_'+n[:-4])
        os.mkdir(dir_fn)
        l._prepare()
        assert os.path.abspath(l.subpath) == os.path.abspath(dir_fn)
        os.rmdir(dir_fn)
        l = sim_struct.LEADFIELD()
        l.pathfem = ''
        l.fnamehead = sphere3_fn
        path, n = os.path.split(sphere3_fn)
        dir_fn = os.path.join(path, 'm2m_'+n[:-4])
        os.mkdir(dir_fn)
        l._prepare()
        assert os.path.abspath(l.subpath) == os.path.abspath(dir_fn)
        os.rmdir(dir_fn) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_log_file(self):
        tmpdir = tempfile.mkdtemp()
        try:
            self.options.log_file_prefix = tmpdir + '/test_log'
            enable_pretty_logging(options=self.options, logger=self.logger)
            self.assertEqual(1, len(self.logger.handlers))
            self.logger.error('hello')
            self.logger.handlers[0].flush()
            filenames = glob.glob(tmpdir + '/test_log*')
            self.assertEqual(1, len(filenames))
            with open(filenames[0]) as f:
                self.assertRegexpMatches(f.read(), r'^\[E [^]]*\] hello$')
        finally:
            for handler in self.logger.handlers:
                handler.flush()
                handler.close()
            for filename in glob.glob(tmpdir + '/test_log*'):
                os.unlink(filename)
            os.rmdir(tmpdir) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_log_file_with_timed_rotating(self):
        tmpdir = tempfile.mkdtemp()
        try:
            self.options.log_file_prefix = tmpdir + '/test_log'
            self.options.log_rotate_mode = 'time'
            enable_pretty_logging(options=self.options, logger=self.logger)
            self.logger.error('hello')
            self.logger.handlers[0].flush()
            filenames = glob.glob(tmpdir + '/test_log*')
            self.assertEqual(1, len(filenames))
            with open(filenames[0]) as f:
                self.assertRegexpMatches(
                    f.read(),
                    r'^\[E [^]]*\] hello$')
        finally:
            for handler in self.logger.handlers:
                handler.flush()
                handler.close()
            for filename in glob.glob(tmpdir + '/test_log*'):
                os.unlink(filename)
            os.rmdir(tmpdir) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def remove(self, rec=1, ignore_errors=False):
        """ remove a file or directory (or a directory tree if rec=1).
        if ignore_errors is True, errors while removing directories will
        be ignored.
        """
        if self.check(dir=1, link=0):
            if rec:
                # force remove of readonly files on windows
                if iswin32:
                    self.chmod(0o700, rec=1)
                import shutil
                py.error.checked_call(
                    shutil.rmtree, self.strpath,
                    ignore_errors=ignore_errors)
            else:
                py.error.checked_call(os.rmdir, self.strpath)
        else:
            if iswin32:
                self.chmod(0o700)
            py.error.checked_call(os.remove, self.strpath) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def tearDown(self):
        '''Remove any files generated during the test'''
        #unittest.TestCase.tearDown(self)
        
        root = os.path.join(".", "files")
        endingList = os.listdir(root)
        endingDir = os.getcwd()
        rmList = [fn for fn in endingList if fn not in self.startingList]
        
        if self.oldRoot == root:
            for fn in rmList:
                fnFullPath = os.path.join(root, fn)
                if os.path.isdir(fnFullPath):
                    os.rmdir(fnFullPath)
                else:
                    os.remove(fnFullPath)
        
        os.chdir(self.oldRoot) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_initial_pysat_load(self):
        import shutil
        saved = False
        try:
            root = os.path.join(os.getenv('HOME'), '.pysat')
            new_root = os.path.join(os.getenv('HOME'), '.saved_pysat')
            shutil.move(root, new_root)
            saved = True
        except:
            pass

        re_load(pysat)

        try:
            if saved:
                # remove directory, trying to be careful
                os.remove(os.path.join(root, 'data_path.txt'))
                os.rmdir(root)
                shutil.move(new_root, root)
        except:
            pass

        assert True 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def _download(self, root):
        if os.path.isdir(os.path.join(root, 'train')):
            return  # Downloaded already

        os.makedirs(root, exist_ok=True)

        zip_file = os.path.join(root, self.UNZIPPED_DIR_NAME + '.zip')

        print('Downloading {}...'.format(os.path.basename(zip_file)))
        download_file_from_google_drive(self.GOOGLE_DRIVE_FILE_ID, zip_file)

        print('Extracting {}...'.format(os.path.basename(zip_file)))
        with zipfile.ZipFile(zip_file, 'r') as fp:
            fp.extractall(root)
        os.remove(zip_file)

        os.rename(os.path.join(root, self.UNZIPPED_DIR_NAME, self.UNZIPPED_TRAIN_SUBDIR),
                  os.path.join(root, 'train'))
        os.rename(os.path.join(root, self.UNZIPPED_DIR_NAME, self.UNZIPPED_VAL_SUBDIR),
                  os.path.join(root, 'val'))
        os.rmdir(os.path.join(root, self.UNZIPPED_DIR_NAME)) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def create_floppy_image(vlock_filename, floppy_image):
    """ Creates the floppy image used to install the vlock file
    """

    # Delete any existing image to start clean
    if os.access(floppy_image, os.R_OK):
        os.unlink(floppy_image)

    subprocess.check_call("mkfs.vfat -C {} 1440".format(floppy_image),
                          shell=True)

    floppy_mnt = "/tmp/mnt-dashboard"
    os.mkdir(floppy_mnt)

    subprocess.check_call("mount -o loop {} {}".format(floppy_image,
                                                       floppy_mnt),
                          shell=True)

    shutil.copy(vlock_filename, os.path.join(floppy_mnt, "versionlock.list"))

    subprocess.check_call("umount {}".format(floppy_mnt), shell=True)
    os.rmdir(floppy_mnt) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def rollback(self):
        if not self.dry_run:
            for f in list(self.files_written):
                if os.path.exists(f):
                    os.remove(f)
            # dirs should all be empty now, except perhaps for
            # __pycache__ subdirs
            # reverse so that subdirs appear before their parents
            dirs = sorted(self.dirs_created, reverse=True)
            for d in dirs:
                flist = os.listdir(d)
                if flist:
                    assert flist == ['__pycache__']
                    sd = os.path.join(d, flist[0])
                    os.rmdir(sd)
                os.rmdir(d)     # should fail if non-empty
        self._init_record() 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def test_subdirectory_deleted(self):
    """Tests that internal _directory_to_subdirs is updated on delete."""
    path = self._create_directory('test')
    sub_path = self._create_directory('test/test2')
    self._watcher.start()

    self.assertEqual(
        set([sub_path]),
        self._watcher._directory_to_subdirs[path])
    os.rmdir(sub_path)
    self.assertEqual(
        set([sub_path]),
        self._watcher._get_changed_paths())
    self.assertEqual(
        set(),
        self._watcher._directory_to_subdirs[path])

    os.rmdir(path)
    self.assertEqual(
        set([path]),
        self._watcher._get_changed_paths()) 

# 需要导入模块: import os [as 别名]
# 或者: from os import rmdir [as 别名]
def _delete_cgroup(cgroup_path):
    paths = _get_cgroup_full_path(cgroup_path)

    try:
        os.rmdir(paths[CgroupType.CPU])
    except FileNotFoundError:
        log.warning('cpu cgroup "{}" not found'.format(cgroup_path))

    try:
        os.rmdir(paths[CgroupType.MEMORY])
    except FileNotFoundError:
        log.warning('memory cgroup "{}" not found'.format(cgroup_path))

    try:
        os.rmdir(paths[CgroupType.CPUSET])
    except FileNotFoundError:
        log.warning('cpuset cgroup "{}" not found'.format(cgroup_path))

    try:
        os.rmdir(paths[CgroupType.PERF_EVENT])
    except FileNotFoundError:
        log.warning('perf_event cgroup "{}" not found'.format(cgroup_path))