utils

最新推荐文章于 2024-05-09 13:14:52 发布

Altium123456789

最新推荐文章于 2024-05-09 13:14:52 发布

阅读量333

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python视觉学习

本文链接：https://blog.youkuaiyun.com/Altium123456789/article/details/78849540

Python视觉学习专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个用于清空指定目录下所有文件及子文件夹的Python函数empty_dir()，以及一个用于创建目录的函数create_dir()。此外，还提供了一个组合函数prepare_dir()，它可以根据需要创建新的目录或清空现有目录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
import errno
import shutil
def empty_dir(path):
    """
    Delete all files and folders in a directory
    :param path: string, path to directory
    :return: nothing
    """
    for the_file in os.listdir(path):
        file_path = os.path.join(path, the_file)
        try:
            if os.path.isfile(file_path):
                os.unlink(file_path)
            elif os.path.isdir(file_path):
                shutil.rmtree(file_path)
        except Exception as e:
            print ('Warning: {}'.format(e))


def create_dir(path):
    """
    Creates a directory
    :param path: string
    :return: nothing
    """
    try:
        os.makedirs(path)
    except OSError as exc:
        if exc.errno != errno.EEXIST:
            raise


def prepare_dir(path, empty=False):
    """
    Creates a directory if it soes not exist
    :param path: string, path to desired directory
    :param empty: boolean, delete all directory content if it exists
    :return: nothing
    """
    if not os.path.exists(path):
        create_dir(path)

    if empty:
        empty_dir(path)