Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 1 | # Module 'glob' -- filename globbing. |
| 2 | |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 3 | import os |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 4 | import fnmatch |
Guido van Rossum | c2ef5c2 | 1992-01-12 23:32:11 +0000 | [diff] [blame] | 5 | import regex |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 6 | |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 7 | |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 8 | def glob(pathname): |
Guido van Rossum | c2ef5c2 | 1992-01-12 23:32:11 +0000 | [diff] [blame] | 9 | if not has_magic(pathname): |
| 10 | if os.path.exists(pathname): |
| 11 | return [pathname] |
| 12 | else: |
| 13 | return [] |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 14 | dirname, basename = os.path.split(pathname) |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 15 | if has_magic(dirname): |
| 16 | list = glob(dirname) |
| 17 | else: |
| 18 | list = [dirname] |
| 19 | if not has_magic(basename): |
| 20 | result = [] |
| 21 | for dirname in list: |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 22 | if basename or os.path.isdir(dirname): |
| 23 | name = os.path.join(dirname, basename) |
| 24 | if os.path.exists(name): |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 25 | result.append(name) |
| 26 | else: |
| 27 | result = [] |
| 28 | for dirname in list: |
| 29 | sublist = glob1(dirname, basename) |
| 30 | for name in sublist: |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 31 | result.append(os.path.join(dirname, name)) |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 32 | return result |
| 33 | |
| 34 | def glob1(dirname, pattern): |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 35 | if not dirname: dirname = os.curdir |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 36 | try: |
Guido van Rossum | bba77af | 1992-01-12 23:26:24 +0000 | [diff] [blame] | 37 | names = os.listdir(dirname) |
| 38 | except os.error: |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 39 | return [] |
| 40 | result = [] |
| 41 | for name in names: |
Guido van Rossum | c2ef5c2 | 1992-01-12 23:32:11 +0000 | [diff] [blame] | 42 | if name[0] != '.' or pattern[0] == '.': |
| 43 | if fnmatch.fnmatch(name, pattern): |
| 44 | result.append(name) |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 45 | return result |
| 46 | |
Guido van Rossum | c2ef5c2 | 1992-01-12 23:32:11 +0000 | [diff] [blame] | 47 | |
| 48 | magic_check = regex.compile('[*?[]') |
| 49 | |
Guido van Rossum | 65a9620 | 1991-01-01 18:17:49 +0000 | [diff] [blame] | 50 | def has_magic(s): |
Guido van Rossum | c2ef5c2 | 1992-01-12 23:32:11 +0000 | [diff] [blame] | 51 | return magic_check.search(s) >= 0 |