def test_is_hidden(temp_dir):
from pydgeot.filesystem import is_hidden
with open(os.path.join(temp_dir, 'visible'), 'w') as fh:
assert not is_hidden(fh.name)
if sys.platform != 'win32':
with open(os.path.join(temp_dir, '.dotfile'), 'w') as fh:
assert is_hidden(fh.name)
if sys.platform == 'win32':
import ctypes
with open(os.path.join(temp_dir, 'hidden'), 'w') as fh:
ctypes.windll.kernel32.SetFileAttributesW(fh.name, 2)
assert is_hidden(fh.name)
elif sys.platform == 'darwin':
import stat
with open(os.path.join(temp_dir, 'hidden'), 'w') as fh:
os.chflags(fh.name, os.stat(fh.name).st_flags ^ stat.UF_HIDDEN)
assert is_hidden(fh.name)
def prepare(self, path):
from .dirconfig import DirConfig
config = DirConfig.get(self.app, path)
if not config.is_valid:
return
if os.path.basename(path) == os.path.basename(config.template):
self.app.sources.add_source(path)
path_dir = os.path.dirname(path)
walk = list(os.walk(path_dir))
directories = [path for path, dirs, files in walk if not is_hidden(path)]
directories.append(path_dir)
for directory in directories:
self._generate_directories.add(directory)
return
self.app.sources.add_source(path)
if not os.path.basename(path) == config.index:
target_path = self.app.target_path(path)
directory = os.path.dirname(path)
self.app.sources.set_targets(path, [target_path])
exif_data = self._get_exif_data(path)
for name, value in exif_data.items():
self.app.contexts.add_context(path, name, str(value))
taken_date = datetime.datetime.fromtimestamp(os.stat(path).st_ctime)
for name in ('DateTime', 'DateTimeOriginal', 'DateTimeDigitized'):
if name in exif_data:
taken_date = datetime.datetime.strptime(exif_data[name], '%Y:%m:%d %H:%M:%S')
self.app.contexts.add_context(path, 'date', taken_date)
self._generate_files.add(path)
self._generate_directories.add(directory)
parent_config = DirConfig.get(self.app, os.path.dirname(config.path))
if parent_config.is_valid:
target_directory = self.app.target_path(os.path.dirname(path))
if not os.path.exists(target_directory):
self._generate_directories.add(parent_config.path)
def _generate_index(self, directory):
from .dirconfig import DirConfig
config = DirConfig.get(self.app, directory)
if not config.is_valid:
return
self.app.log.info('Generating index for \'{}\''.format(directory))
directories = []
files = []
for name in os.listdir(directory):
path = os.path.join(directory, name)
if is_hidden(path) or os.path.basename(path) == os.path.basename(config.template):
continue
elif os.path.isfile(path):
files.append(path)
elif os.path.isdir(path):
directories.append(path)
directories = self._contextify_file_list(config, directory, directories)
files = self._contextify_file_list(config, directory, files)
target_directory = self.app.target_path(directory)
target_path = os.path.join(target_directory, config.index)
os.makedirs(target_directory, exist_ok=True)
template = self._env.get_template(self.app.relative_path(config.template))
f = open(target_path, 'w', encoding='utf-8')
rendered = template.render(
config=config,
dir_name=os.path.basename(directory),
has_parent_dir=(directory != self.app.source_root),
dirs=directories,
files=files)
f.write(rendered)
f.close()