plugins/specialaccounts.py

   1 #!/usr/bin/python -tt
   2 # vim:set ts=4 sw=4 expandtab:
   3 #
   4 # $Id$
   5 # $URL$
   6 #
   7 # NodeManager plugin to create special accounts
   8
   9 """
  10 create/populate accounts/ssh keys for special persons such as root, site_admin, etc.
  11
  12 """
  13
  14 import errno
  15 import os
  16 import random
  17 import string
  18 import tempfile
  19 import grp
  20 import pwd
  21
  22 import logger
  23 import tools
  24
  25 # right after conf_files
  26 priority = 3
  27
  28 def start():
  29     logger.log("specialaccounts: plugin starting up...")
  30
  31 def GetSlivers(data, conf = None, plc = None):
  32     if 'accounts' not in data:
  33         logger.log_missing_data("specialaccounts.GetSlivers",'accounts')
  34         return
  35
  36     for account in data['accounts']:
  37         name = account['name']
  38         new_keys = account['keys']
  39
  40         logger.log('specialaccounts: dealing with account %s'%name)
  41
  42         # look up account name, which must exist
  43         pw_info = pwd.getpwnam(name)
  44         uid = pw_info[2]
  45         gid = pw_info[3]
  46         pw_dir = pw_info[5]
  47
  48         # populate account's .ssh/authorized_keys file
  49         dot_ssh = os.path.join(pw_dir,'.ssh')
  50         if not os.access(dot_ssh, os.F_OK): os.mkdir(dot_ssh)
  51         auth_keys = os.path.join(dot_ssh,'authorized_keys')
  52
  53         # catenate all keys in string, add newlines just in case (looks like keys already have this, but)
  54         auth_keys_contents = '\n'.join(new_keys)+'\n'
  55
  56         changes = tools.replace_file_with_string(auth_keys,auth_keys_contents)
  57         if changes:
  58             logger.log("specialaccounts: keys file changed: %s" % auth_keys)
  59
  60         # always set permissions properly
  61         os.chmod(dot_ssh, 0700)
  62         os.chown(dot_ssh, uid,gid)
  63         os.chmod(auth_keys, 0600)
  64         os.chown(auth_keys, uid,gid)
  65
  66         logger.log('specialaccounts: installed ssh keys for %s' % name)