Skip to content

Instantly share code, notes, and snippets.

@chrK

chrK/treewalker.py

Created July 28, 2012 18:56
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save chrK/3194414 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save chrK/3194414 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Christian Müller revised this gist Jul 28, 2012. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion treewalker.py
    View file
    Original file line number Diff line number Diff line change
    @@ -5,7 +5,7 @@
    import re


    root = settings.PROJECT_ROOT
    root = os.getcwd()

    includes = ['includes', ] # For files only.
    excludes = ['excludes', ] # For dirs and files.
  2. Christian Müller created this gist Jul 28, 2012.
    32 changes: 32 additions & 0 deletions treewalker.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,32 @@
    # -*- coding: utf-8 -*-

    import os
    import fnmatch
    import re


    root = settings.PROJECT_ROOT

    includes = ['includes', ] # For files only.
    excludes = ['excludes', ] # For dirs and files.

    # Transform glob patterns to regular expressions:
    includes = r'|'.join([fnmatch.translate(x) for x in includes])
    excludes = r'|'.join([fnmatch.translate(x) for x in excludes]) or r'$.'

    for root, dirs, files in os.walk(project_root):

    # Exclude dirs:
    dirs[:] = [os.path.join(root, d) for d in dirs]
    dirs[:] = [d for d in dirs if not re.match(excludes, d)]

    # Exclude/include files:
    files = [os.path.join(root, f) for f in files]
    files = [f for f in files if not re.match(excludes, f)]
    files = [f for f in files if re.match(includes, f)]

    for fname in files:
    (dir_name, file_name) = os.path.split(fname)
    (file_base_name, file_extension) = os.path.splitext(file_name)

    print dir_name, file_name