Try to search all profiles

ezesoler · May 23, 2017 · 1fcaa76 · 1fcaa76
1 parent 7a3f183
commit 1fcaa76
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/Linux/lazagne/softwares/browsers/chrome.py b/Linux/lazagne/softwares/browsers/chrome.py
@@ -8,16 +8,24 @@
 import sqlite3
 import tempfile
 
+import os
+
 class Chrome(ModuleInfo):
     def __init__(self):
 		options = {'command': '-C', 'action': 'store_true', 'dest': 'chrome', 'help': 'chrome'}
 		ModuleInfo.__init__(self, 'chrome', 'browsers', options)
 
     def get_paths(self):
-        return homes.get(file=[
-            '.config/google-chrome/Default/Login Data',
-            '.config/chromium/Default/Login Data'
-        ])
+        for profile_dir in homes.get(dir=['.config/google-chrome', '.config/chromium']):
+            try:
+                subdirs = os.listdir(profile_dir)
+            except:
+                continue
+
+            for subdir in subdirs:
+                logins = os.path.join(profile_dir, subdir, 'Login Data')
+                if os.path.isfile(logins):
+                    yield logins
 
     def get_logins(self, path):
         try: