1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/moinformat/output/directory.py Sun Jul 29 23:27:23 2018 +0200
1.3 @@ -0,0 +1,60 @@
1.4 +#!/usr/bin/env python
1.5 +
1.6 +"""
1.7 +Directory output context.
1.8 +
1.9 +Copyright (C) 2018 Paul Boddie <paul@boddie.org.uk>
1.10 +
1.11 +This program is free software; you can redistribute it and/or modify it under
1.12 +the terms of the GNU General Public License as published by the Free Software
1.13 +Foundation; either version 3 of the License, or (at your option) any later
1.14 +version.
1.15 +
1.16 +This program is distributed in the hope that it will be useful, but WITHOUT
1.17 +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.18 +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.19 +details.
1.20 +
1.21 +You should have received a copy of the GNU General Public License along with
1.22 +this program. If not, see <http://www.gnu.org/licenses/>.
1.23 +"""
1.24 +
1.25 +from moinformat.output.common import Output
1.26 +from os.path import abspath, commonprefix, join
1.27 +
1.28 +def inside(filename, dirname):
1.29 +
1.30 + "Return whether 'filename' is inside 'dirname'."
1.31 +
1.32 + # Get the directory with trailing path separator.
1.33 +
1.34 + dirname = join(dirname, "")
1.35 + return commonprefix((filename, dirname)) == dirname
1.36 +
1.37 +
1.38 +
1.39 +class DirectoryOutput(Output):
1.40 +
1.41 + "A directory output context."
1.42 +
1.43 + def __init__(self, filename):
1.44 +
1.45 + "Initialise the context with the given 'filename'."
1.46 +
1.47 + Output.__init__(self)
1.48 + self.filename = abspath(filename)
1.49 +
1.50 + def get_filename(self, filename):
1.51 +
1.52 + "Return a file with the given 'filename' within the directory."
1.53 +
1.54 + # Get the absolute path for the combination of directory and filename.
1.55 +
1.56 + pathname = abspath(join(self.filename, filename))
1.57 +
1.58 + if inside(pathname, self.filename):
1.59 + return pathname
1.60 + else:
1.61 + raise ValueError, filename
1.62 +
1.63 +# vim: tabstop=4 expandtab shiftwidth=4