AddAdminUser, SetAuthorized_keys

This commit is contained in:
Ian Gulliver
2017-09-22 13:52:06 -07:00
parent 311b874765
commit fa90e36998

View File

@@ -67,13 +67,24 @@ class IconModule(object):
self.InstallPackages('python-pip')
self.ExecChroot('pip', 'install', *packages)
def AddSystemUsers(self, *users):
for user in users:
self.ExecChroot('adduser', '--system', '--group', '--no-create-home', '--disabled-login', user)
def AddDaemonUsers(self, user):
self.ExecChroot('adduser', '--system', '--group', '--no-create-home', '--disabled-login', user)
def AddAdminUser(self, user):
self.ExecChroot('adduser', '--system', '--group', '--disabled-password', '--shell=/bin/bash', user)
with open(os.path.join(self._chroot_path, 'etc', 'sudoers.d', FLAGS.username), 'w') as fh:
fh.write('%s\tALL=(ALL) NOPASSWD: ALL\n' % user)
def AddUserToGroup(self, user, group):
self.ExecChroot('usermod', '--append', '--groups', group, user)
def SetAuthorizedKeys(self, user, path):
dest_dir = os.path.join(self._chroot_path, 'home', user, '.ssh')
dest_path = os.path.join(dest_dir, 'authorized_keys')
os.mkdir(dest_dir)
shutil.copy(path, dest_path)
self.ExecChroot('chown', '%s:%s' % (user, user), os.path.join('home', user, '.ssh', 'authorized_keys'))
def AddKernelModules(self, *modules):
with open(os.path.join(self._chroot_path, 'etc', 'modules'), 'a') as fh:
for module in modules: