1 files changed, 56 insertions, 0 deletions
diff --git a/generators/account.py b/generators/account.py
index e69de29bb..0773ae3d8 100644
--- a/generators/account.py
+++ b/generators/account.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python
+
+from Generator from Generator
+from GeneratorUtils import DirectoryBacked
+from Types import ConfigFile
+
+class account(Generator):
+    __name__ = 'account'
+    __version__ = '$Id$'
+    __author__ = 'bcfg-dev@mcs.anl.gov'
+    __build__ = {'/etc/passwd':"GenFromYP",
+                 '/etc/group':"GenFromYP",
+                 '/etc/security/limits.conf':"GenLimits",
+                 '/root/.ssh/authorized_keys':"GenRootKeys"}
+
+    __doc__ = '''This module generates account config files, based on an internal data repo:
+    static.(passwd|group|limits.conf) -> static entries
+    dyn.(passwd|group) -> dynamic entries (usually acquired from yp)
+    useraccess -> users to be granted login access on some hosts
+    superusers -> users to be granted root privs on all hosts
+    rootlike -> users to be granted root privs on some hosts
+    '''
+
+    def __setup__(self):
+        self.repository = DirectoryBacked(self.data)
+        self.ssh = DirectoryBacked("%s/ssh"%(self.data))
+
+    def GenFromYP(self,filename,client):
+        fname = filename.split('/')[-1]
+        static = self.repository.entries["static.%s"%(fname)].data
+        yp = self.repository.entries["dyn.%s"%(fname)].data
+        return ConfigFile(filename,"root","root",'0644',static+yp)
+
+    def GenLimits(self,filename,client):
+        fname = 'limits.conf'
+        static = self.repository.entries["static.limits.conf"].data
+        superusers = self.repository.entries["superusers"].data.split()
+        useraccess = self.repository.entries["useraccess"].data
+        users = [x[0] for x in useraccess if x[1] == client]
+
+        data = static + join(map(lambda x:"%s hard maxlogins 1024\n"%x, superusers + users), ""),
+
+        if "*" not in users:
+            data += "* hard maxlogins 0\n"
+        
+        return ConfigFile(filename,"root","root",'0644',data)
+
+    def GenRootKeys(self,filename,client):
+        su = self.repository.entries['superusers'].data.split()
+        rl = self.repository.entries['rootlike'].data.split()
+        su += [split(x,':')[0] for x in rl if split(x,':')[1] == client]
+        data = ''
+        for user in su:
+            if self.ssh.entries.has_key(user):
+                data += self.ssh.entries[user].data
+        return ConfigFile(filename,'root','root','0600',data)