mirror of
https://github.com/kristoferssolo/School.git
synced 2025-10-21 20:10:38 +00:00
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
import os
|
|
|
|
from .config import Configuration
|
|
from .utils import iter_entry_points
|
|
from .utils import trace
|
|
|
|
|
|
def walk_potential_roots(root, search_parents=True):
|
|
"""
|
|
Iterate though a path and each of its parents.
|
|
:param root: File path.
|
|
:param search_parents: If ``False`` the parents are not considered.
|
|
"""
|
|
|
|
if not search_parents:
|
|
yield root
|
|
return
|
|
|
|
tail = root
|
|
|
|
while tail:
|
|
yield root
|
|
root, tail = os.path.split(root)
|
|
|
|
|
|
def match_entrypoint(root, name):
|
|
"""
|
|
Consider a ``root`` as entry-point.
|
|
:param root: File path.
|
|
:param name: Subdirectory name.
|
|
:return: ``True`` if a subdirectory ``name`` exits in ``root``.
|
|
"""
|
|
|
|
if os.path.exists(os.path.join(root, name)):
|
|
if not os.path.isabs(name):
|
|
return True
|
|
trace("ignoring bad ep", name)
|
|
|
|
return False
|
|
|
|
|
|
def iter_matching_entrypoints(root, entrypoint, config: Configuration):
|
|
"""
|
|
Consider different entry-points in ``root`` and optionally its parents.
|
|
:param root: File path.
|
|
:param entrypoint: Entry-point to consider.
|
|
:param config: Configuration,
|
|
read ``search_parent_directories``, write found parent to ``parent``.
|
|
"""
|
|
|
|
trace("looking for ep", entrypoint, root)
|
|
|
|
for wd in walk_potential_roots(root, config.search_parent_directories):
|
|
for ep in iter_entry_points(entrypoint):
|
|
if match_entrypoint(wd, ep.name):
|
|
trace("found ep", ep, "in", wd)
|
|
config.parent = wd
|
|
yield ep
|