Эффективный способ создания рекурсивных путей Python - PullRequest
15 голосов
/ 03 июня 2009

Мне нужна простая функция для создания пути в Python, где родитель может существовать или не существовать.

Из документации на питоне os.makedirs потерпит неудачу, если один из родителей существует.

Я написал приведенный ниже метод, который работает так, что создает столько подкаталогов, сколько необходимо.

Это выглядит эффективно?

def create_path(path):
    import os.path as os_path
    paths_to_create = []
    while not os_path.lexists(path):
        paths_to_create.insert(0, path)
        head,tail = os_path.split(path)
        if len(tail.strip())==0: # Just incase path ends with a / or \
            path = head
            head,tail = os_path.split(path)
        path = head

    for path in paths_to_create:
        os.mkdir(path)

Ответы [ 8 ]

48 голосов
/ 03 июня 2009

"Из документации по питону os.makedirs потерпит неудачу, если один из родителей существует."

Нет, os.makedirs завершится ошибкой, если сам каталог уже существует. Он не потерпит неудачу, если какой-либо из родительских каталогов уже существует.

16 голосов
/ 25 января 2014

Вот мое мнение, которое позволяет системным библиотекам справляться со всеми путями. Любые ошибки, кроме уже существующего каталога, распространяются.

import os, errno

def ensure_dir(dirname):
    """
    Ensure that a named directory exists; if it does not, attempt to create it.
    """
    try:
        os.makedirs(dirname)
    except OSError, e:
        if e.errno != errno.EEXIST:
            raise
4 голосов
/ 10 августа 2011

Черновая тяга:

import os


class Path(str):
    """
    A helper class that allows easy contactenation
    of path components, creation of directory trees,
    amongst other things.
    """  
    @property
    def isdir(self):
        return os.path.isdir(self)

    @property
    def isfile(self):
        return os.path.isfile(self)

    def exists(self):
        exists = False
        if self.isfile:
            try:
                f = open(self)
                f.close()
                exists = True
            except IOError:
                exists = False
        else:
            return self.isdir
        return exists

    def mktree(self, dirname):
        """Create a directory tree in this directory."""
        newdir = self + dirname
        if newdir.exists():
            return newdir
        path = dirname.split('/') or [dirname]
        current_path = self + path.pop(0)
        while True:
            try:
                os.mkdir(current_path)
            except OSError as e:
                if not e.args[0] == 17:
                    raise e
                current_path = current_path + path.pop(0)
                continue
            if len(path) == 0:
                break
        return current_path

    def up(self):
        """
        Return a new Path object set a the parent
        directory of the current instance.
        """
        return Path('/'.join(self.split('/')[:-1]))

    def __repr__(self):
        return "<Path: {0}>".format(self)

    def __add__(x, y):
        return Path(x.rstrip('/') + '/' + y.lstrip('/'))
3 голосов
/ 08 апреля 2015

В python (> = 3.4.1) для os.makedirs есть параметр exist_ok.

Если exist_ok имеет значение False (по умолчанию), OSError вызывается, если цель каталог уже существует.

Так что, если вы используете, как exist_ok = True, не будет никаких проблем при создании рекурсивного каталога.

Примечание: существующий_ок поставляется с Python 3.2, с другой стороны, была ошибка о повышении исключения, даже если вы установите True. Так что попробуйте использовать Python> = 3.4.1 (исправлено в этой версии)

2 голосов
/ 01 ноября 2012

Попробуйте этот код, он проверяет, существует ли путь до n уровня подкаталога, и создает каталог, если не существует.

def pathtodir(path):
if not os.path.exists(path):
    l=[]
    p = "/"
    l = path.split("/")
    i = 1
    while i < len(l):
        p = p + l[i] + "/"
        i = i + 1
        if not os.path.exists(p):
            os.mkdir(p, 0755)
1 голос
/ 15 ноября 2017

Это старая ветка, но я не был удовлетворен предоставленными решениями, так как они были в основном слишком сложными для простой задачи.

Из доступных функций в библиотеке я считаю, что самое чистое, что мы можем сделать, это:

os.path.isdir("mydir") or os.makedirs("mydir")
1 голос
/ 28 июня 2016

Этот код сгенерирует дерево каталогов с заданной глубиной и шириной, используя рекурсивный вызов функции:

#!/usr/bin/python2.6

import sys
import os

def build_dir_tree(base, depth, width):
    print("Call #%d" % depth)
    if depth >= 0:
        curr_depth = depth
        depth -= 1
        for i in xrange(width):
                # first creating all folder at current depth
                os.makedirs('%s/Dir_#%d_level_%d' % (base, i, curr_depth))
        dirs = os.walk(base).next()[1]
        for dir in dirs:
                newbase = os.path.join(base,dir)
                build_dir_tree(newbase, depth, width)
    else:
        return

if not sys.argv[1:]:
        print('No base path given')
        sys.exit(1)

print('path: %s, depth: %d, width: %d' % (sys.argv[1], int(sys.argv[2]), int(sys.argv[3])))
build_dir_tree(sys.argv[1], int(sys.argv[2]), int(sys.argv[3]))
1 голос
/ 26 июля 2014

Я нашел этот вопрос, когда искал способ создания простых деревьев каталогов внутри директории проекта.

Я немного новичок в Python, и я испытываю трудности, когда структуры данных становятся слишком сложными, то есть вложенными. В ментальном отображении моего мозга гораздо проще отслеживать небольшие списки итераций, поэтому я придумал две очень простые определения, которые помогут мне в создании дерева каталогов.

В примере для создания дерева используются четыре объекта:

  1. путь к корневому каталогу = PROJECT_HOME
  2. домашний путь = домашний (создается, если он не существует, не перезаписывается)
  3. итерация имен каталогов, которые будут идти внутри home = ветки (создаются внутри дома, не перезаписываются)
  4. словарь ключевых итераций, которые отображаются на ветви = листья (каждое значение создается внутри каждой отображаемой ветви, не перезаписывается)
  5. Если какой-либо каталог существует, он не перезаписывается и ошибка проходит бесшумно.

    import os
    from os.path import join as path_join
    import errno
    
    def make_node(node):
        try:
            os.makedirs(node)
        except OSError, e:
            if e.errno != errno.EEXIST:
                raise
    
    
    def create_tree(home, branches, leaves):
        for branch in branches:
            parent = path_join(home, branch)
            make_node(parent)
            children = leaves.get(branch, [])
            for child in children:
                child = os.path.join(parent, child)
                make_node(child)
    
    if __name__ == "__main__":
        try:  # create inside of PROJECT_HOME if it exists
            PROJECT_HOME = os.environ['PROJECT_HOME']
        except KeyError:  # otherwise in user's home directory
            PROJECT_HOME = os.expanduser('~')
    
        home = os.path.join(PROJECT_HOME, 'test_directory_tree')
        create_tree(home, branches=[], leaves={})
    
        branches = (
            'docs',
            'scripts',
        )
        leaves = (
            ('rst', 'html', ),
            ('python', 'bash', )
        )
        leaves = dict(list(zip(branches, leaves)))
        create_tree(home, branches, leaves)
    
        python_home = os.path.join(home, 'scripts', 'python')
        branches = (
            'os',
            'sys',
            'text_processing',
        )
        leaves = {}
        leaves = dict(list(zip(branches, leaves)))
        create_tree(python_home, branches, leaves)
    
        after_thought_home = os.path.join(home, 'docs', 'after_thought')
        branches = (
            'child_0',
            'child_1',
        )
        leaves = (
            ('sub_0', 'sub_1'),
            (),
        )
        leaves = dict(list(zip(branches, leaves)))
        create_tree(after_thought_home, branches, leaves)
    

Дерево каталогов, которое создает этот пример, выглядит так:

    dev/test_directory_tree/
    ├── docs
    │   ├── after_thought
    │   │   ├── child_0
    │   │   │   ├── sub_0
    │   │   │   └── sub_1
    │   │   └── child_1
    │   ├── html
    │   └── rst
    └── scripts
        ├── bash
        └── python
            ├── os
            ├── sys
            └── text_processing
...