diff options
Diffstat (limited to 'modules/title.py')
| -rw-r--r-- | modules/title.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/modules/title.py b/modules/title.py index 96a9ee0b..309b9b00 100644 --- a/modules/title.py +++ b/modules/title.py @@ -50,15 +50,17 @@ class Module(ModuleManager.BaseModule): return -1, None try: - page = utils.http.request(url, parse=True) - except utils.http.HTTPWrongContentTypeException: - return -1, None + page = utils.http.request(url) except Exception as e: self.log.error("failed to get URL title for %s: %s", [url, str(e)]) return -1, None - if page.data.title: - title = utils.parse.line_normalise(page.data.title.text) + if not page.content_type in utils.http.SOUP_CONTENT_TYPES: + return -1, None + page = page.soup() + + if page.title: + title = utils.parse.line_normalise(page.title.text) if not title: return -3, None @@ -135,3 +137,8 @@ class Module(ModuleManager.BaseModule): event["stdout"].write(title) else: event["stderr"].write("Failed to get title") + + @utils.hook("received.command.notitle") + @utils.kwarg("expect_output", False) + def notitle(self, event): + pass |