Add trivial SSH id files support

ezesoler · May 23, 2017 · 5433fbd · 5433fbd
1 parent bff5bda
commit 5433fbd
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Linux/lazagne/config/homes.py b/Linux/lazagne/config/homes.py
@@ -36,6 +36,24 @@ def get(file=[], dir=[]):
         if not files and not dirs and os.path.isdir(p):
             yield p
 
+def users(file=[], dir=[]):
+    files = file if (type(file) in (tuple, list)) else [file]
+    dirs = dir if (type(dir) in (tuple, list)) else [dir]
+
+    for pw in pwd.getpwall():
+        if files:
+            for file in files:
+                if os.path.isfile(os.path.join(pw.pw_dir, file)):
+                    yield pw.pw_name, os.path.join(pw.pw_dir, file)
+
+        if dirs:
+            for dir in dirs:
+                if os.path.isdir(os.path.join(pw.pw_dir, dir)):
+                    yield pw.pw_name, os.path.join(pw.pw_dir, dir)
+
+        if not files and not dirs and os.path.isdir(pw.pw_dir):
+            yield pw.pw_name, pw.pw_dir
+
 def sessions(setenv=True):
     visited = set()
 

diff --git a/Linux/lazagne/config/manageModules.py b/Linux/lazagne/config/manageModules.py
@@ -62,6 +62,7 @@ def get_modules():
 		Shadow(),
         Aws(),
         Docker(),
+		Ssh(),
 		SQLDeveloper(),
 		Squirrel(),
 		Wifi(),