"""Filename globbing utility."""

import sys

import os

import re

import fnmatch

__all__ = ["glob", "iglob"]

def glob(pathname):

    """Return a list of paths matching a pathname pattern.

    The pattern may contain simple shell-style wildcards a la fnmatch.

    """

    return list(iglob(pathname))

def iglob(pathname):

    """Return a list of paths matching a pathname pattern.

    The pattern may contain simple shell-style wildcards a la fnmatch.

    """

    if not has_magic(pathname):

        if os.path.lexists(pathname):

            yield pathname

        return

    dirname, basename = os.path.split(pathname)

    if not dirname:

        for name in glob1(os.curdir, basename):

            yield name

        return

    if has_magic(dirname):

        dirs = iglob(dirname)

    else:

        dirs = [dirname]

    if has_magic(basename):

        glob_in_dir = glob1

    else:

        glob_in_dir = glob0

    for dirname in dirs:

        for name in glob_in_dir(dirname, basename):

            yield os.path.join(dirname, name)

# These 2 helper functions non-recursively glob inside a literal directory.

# They return a list of basenames. `glob1` accepts a pattern while `glob0`

# takes a literal basename (so it only has to check for its existence).

def glob1(dirname, pattern):

    if not dirname:

        dirname = os.curdir

    if isinstance(pattern, unicode) and not isinstance(dirname, unicode):

        dirname = unicode(dirname, sys.getfilesystemencoding() or

                                   sys.getdefaultencoding())

    try:

        names = os.listdir(dirname)

    except os.error:

        return []

    if pattern[0] != '.':

        names = filter(lambda x: x[0] != '.', names)

    return fnmatch.filter(names, pattern)

def glob0(dirname, basename):

    if basename == '':

        # `os.path.split()` returns an empty basename for paths ending with a

        # directory separator.  'q*x/' should match only directories.

        if os.path.isdir(dirname):

            return [basename]

    else:

        if os.path.lexists(os.path.join(dirname, basename)):

            return [basename]

    return []

magic_check = re.compile('[*?[]')

def has_magic(s):

    return magic_check.search(s) is not None