1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
# fixtures: Fixtures with cleanups for testing and convenience.
#
# Copyright (c) 2012, Robert Collins <robertc@robertcollins.net>
#
# Licensed under either the Apache License, Version 2.0 or the BSD 3-clause
# license at the users choice. A copy of both licenses are available in the
# project source as Apache-2.0 and BSD. You may not use this file except in
# compliance with one of these two licences.
#
# Unless required by applicable law or agreed to in writing, software
# distributed under these licenses is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# license you chose for the specific language governing permissions and
# limitations under that license.
__all__ = [
'FileTree',
]
import os
from fixtures import Fixture
from fixtures._fixtures.tempdir import TempDir
def normalize_shape(shape):
normal_shape = []
for entry in sorted(shape):
if isinstance(entry, basestring):
if entry[-1] == '/':
normal_shape.append((entry, None))
else:
normal_shape.append((entry, "The file '%s'." % (entry,)))
else:
if entry[0][-1] == '/':
normal_shape.append((entry[0], None))
else:
normal_shape.append(entry)
return normal_shape
class FileTree(Fixture):
"""A structure of files and directories on disk."""
def __init__(self, shape):
"""Create a ``FileTree``.
:param shape: A list of descriptions of files and directories to make.
Files are described as ``("filename", contents)`` and directories
are written as ``"dirname/"``. The trailing slash is necessary.
Directories can also be written as ``("dirname/",)``.
"""
super(FileTree, self).__init__()
self._shape = normalize_shape(shape)
def setUp(self):
super(FileTree, self).setUp()
tempdir = self.useFixture(TempDir())
self.path = path = tempdir.path
for name, contents in self._shape:
name = os.path.join(path, name)
if name[-1] == '/':
os.mkdir(name)
else:
f = open(name, 'w')
f.write(contents)
f.close()
|