|
1 import os |
|
2 import shutil |
|
3 import tempfile |
|
4 |
|
5 |
|
6 # stub file paths |
|
7 files = [('foo.txt',), |
|
8 ('foo', 'bar.txt',), |
|
9 ('foo', 'bar', 'fleem.txt',), |
|
10 ('foobar', 'fleem.txt',), |
|
11 ('bar.txt',), |
|
12 ('nested_tree', 'bar', 'fleem.txt',), |
|
13 ('readonly.txt',), |
|
14 ] |
|
15 |
|
16 |
|
17 def create_stub(): |
|
18 """create a stub directory""" |
|
19 |
|
20 tempdir = tempfile.mkdtemp() |
|
21 try: |
|
22 for path in files: |
|
23 fullpath = os.path.join(tempdir, *path) |
|
24 dirname = os.path.dirname(fullpath) |
|
25 if not os.path.exists(dirname): |
|
26 os.makedirs(dirname) |
|
27 contents = path[-1] |
|
28 f = file(fullpath, 'w') |
|
29 f.write(contents) |
|
30 f.close() |
|
31 return tempdir |
|
32 except Exception, e: |
|
33 try: |
|
34 shutil.rmtree(tempdir) |
|
35 except: |
|
36 pass |
|
37 raise e |