Fix parsing request responst message

This commit is contained in:
JonnyWong16 2020-07-25 11:59:24 -07:00
commit 0272c35047
No known key found for this signature in database
GPG key ID: B1F1F9807184697A

View file

@ -295,20 +295,20 @@ def server_message(response, return_msg=False):
try:
soup = BeautifulSoup(response.content, "html5lib")
except Exception:
pass
soup = None
# Find body and cleanup common tags to grab content, which probably
# contains the message.
message = soup.find("body")
elements = ("header", "script", "footer", "nav", "input", "textarea")
if soup:
# Find body and cleanup common tags to grab content, which probably
# contains the message.
message = soup.find("body")
elements = ("header", "script", "footer", "nav", "input", "textarea")
for element in elements:
for element in elements:
for tag in soup.find_all(element):
tag.replaceWith("")
for tag in soup.find_all(element):
tag.replaceWith("")
message = message.text if message else soup.text
message = message.strip()
message = message.text if message else soup.text
message = message.strip()
# Second attempt is to just take the response
if message is None: