diff --git a/src/solo_turnier/html_locator.py b/src/solo_turnier/html_locator.py index e4be68f..cd5a449 100644 --- a/src/solo_turnier/html_locator.py +++ b/src/solo_turnier/html_locator.py @@ -4,21 +4,24 @@ import logging class HtmlLocator: def __init__(self): self.l = logging.getLogger('solo_turnier.html_locator') - self.fileName = 'erg.htm' - def __findRecursivelyCandidates(self, path: str): + def __findRecursivelyCandidates(self, path: str, fileName: str): ret = [] ls = os.listdir(path) - if self.fileName in ls and os.path.isfile(os.path.join(path, self.fileName)): - ret.append(os.path.join(path, self.fileName)) + if fileName in ls and os.path.isfile(os.path.join(path, fileName)): + ret.append(os.path.join(path, fileName)) for p in ls: subPath = os.path.join(path, p) if os.path.isdir(subPath): - ret = ret + self.__findRecursivelyCandidates(subPath) + ret = ret + self.__findRecursivelyCandidates(subPath, fileName) return ret def findCandidates(self, path: str): - return self.__findRecursivelyCandidates(path) + return self.__findRecursivelyCandidates(path, 'erg.htm') + + def findPreviewRoundCandidates(self, path: str): + candidates = self.__findRecursivelyCandidates(path, 'tabges.htm') + return candidates