diff options
| -rw-r--r-- | morphlib/extensions.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/morphlib/extensions.py b/morphlib/extensions.py index aa0622b8..cc390c50 100644 --- a/morphlib/extensions.py +++ b/morphlib/extensions.py @@ -60,9 +60,19 @@ def _list_morph_extension_filenames(kind): def _get_extension_name(filename): return os.path.basename(filename) +def _get_repo_extension_relpath(name, kind): + repo_dir = _get_root_repo() + files = repo_dir.list_files() + matching = [f for f in files if os.path.splitext(f)[1] == kind and + os.path.basename(os.path.splitext(f)[0]) == name] + if matching: + return matching[0] + return name + kind + def _get_repo_extension_contents(name, kind): repo_dir = _get_root_repo() - return repo_dir.read_file(name + kind) + filepath = _get_repo_extension_relpath(name, kind) + return repo_dir.read_file(filepath) def _get_morph_extension_filename(name, kind): return os.path.join(_get_morph_extension_directory(), name + kind) |