diff --git a/bots/markpages.py b/bots/markpages.py index 971e9c2..d2e8d7f 100644 --- a/bots/markpages.py +++ b/bots/markpages.py @@ -62,6 +62,9 @@ class MarkPagesBot( CurrentPageBot ): # sets 'current_page' on each treat() # Init attribute self.__redfams = None # Will hold a generator with our redfams + if "famhash" in kwargs: + self.famhash = kwargs["famhash"] + # We do not use predefined genFactory as there is no sensefull case to # give a generator via cmd-line for this right now self.genFactory = pagegenerators.GeneratorFactory() @@ -102,8 +105,15 @@ class MarkPagesBot( CurrentPageBot ): # sets 'current_page' on each treat() end_after = datetime.strptime( jogobot.config["red.markpages"]["mark_done_after"], "%Y-%m-%d" ) - self.__redfams = list( RedFamWorker.gen_by_status_and_ending( - "archived", end_after) ) + + if hasattr(self, "famhash"): + self.__redfams = list( + RedFamWorker.session.query(RedFamWorker).filter( + RedFamWorker.famhash == self.famhash ) ) + + else: + self.__redfams = list( RedFamWorker.gen_by_status_and_ending( + "archived", end_after) ) return self.__redfams