Now re.findall is used

1 files changed, 8 insertions, 11 deletions

diff --git a/pwndb_credentials.py b/pwndb_credentials.py
index 8fa6896..c0fbc1d 100644
--- a/pwndb_credentials.py
+++ b/pwndb_credentials.py
@@ -36,19 +36,16 @@ class FindPasswords():
 
     def request_data(self):
         result = []
-        if self.session is None:
-            return []
-        try:
+        if self.session is not None:
             response = self.session.post(self.url, data=self.data, timeout=5)
-            soup = soup = BeautifulSoup(response.text, features="html.parser")
-            passwords = soup.find("pre").get_text()
-            if passwords:
-                i = 0
-                for line in passwords.split("\n"):
-                    if "[password]" in line:
+            if response.status_code == 200:
+                soup = BeautifulSoup(response.text, features="html.parser")
+                try:
+                    passwords = soup.find("pre").get_text()
+                    for line in re.findall("\[password\] => .*", passwords):
                         result.append(line.split("=> ")[1])
-        except Exception as e:
-            print(e)
+                except Exception as e:
+                    print(e)
 
         return result