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: try:
soup = BeautifulSoup(response.content, "html5lib") soup = BeautifulSoup(response.content, "html5lib")
except Exception: except Exception:
pass soup = None
# Find body and cleanup common tags to grab content, which probably if soup:
# contains the message. # Find body and cleanup common tags to grab content, which probably
message = soup.find("body") # contains the message.
elements = ("header", "script", "footer", "nav", "input", "textarea") 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): message = message.text if message else soup.text
tag.replaceWith("") message = message.strip()
message = message.text if message else soup.text
message = message.strip()
# Second attempt is to just take the response # Second attempt is to just take the response
if message is None: if message is None: