diff --git a/red_page.py b/red_page.py index e9fcfc6..1a790ec 100644 --- a/red_page.py +++ b/red_page.py @@ -7,7 +7,7 @@ from red_fam import RED_FAM_PARSER class RED_PAGE: """Class for handling redundance discussion pages and archives""" - def __init__( self, page ): + def __init__( self, page, archive=False ): """ Generate a new RED_PAGE object based on the given pywikibot page object @@ -16,6 +16,7 @@ class RED_PAGE: # Safe the pywikibot page object self.page = page + self._archive = archive self.__handle_db( ) @@ -54,7 +55,7 @@ class RED_PAGE: """ Detects wether current page is an archive of discussions """ - if u"/Archiv" in self.page.title: + if self._archive or ( u"/Archiv" in self.page.title() ): return True else: return False