maintain_user_from_groups.py #5

#!/usr/bin/env python3
#
# Usage:
#	maintain_user_from_groups.py [instance]
#
# Defaults to instance 1 if parameter not given.
#
# What this scripts does:
#	Reads users from groups
#	Creates any missing user accounts
#   Removes accounts that are not in the group

# Set this for your environment to get the correct email address for your users.

import os
import os.path
import re
import subprocess
import sys
import sdputils

if len(sys.argv) > 1:  # see params above
  SDP_INSTANCE = str(sys.argv[1])
else:
  SDP_INSTANCE = '1'

utils = sdputils.SDPUtils(SDP_INSTANCE)
p4 = utils.p4
utils.login()
domain = (utils.get('domain'))

clientlist = []
users = []
groupusers = []

###############################################################################
def getusers():
  with open('users.txt', 'r') as userfile:
    for user in userfile.readlines():
      user = re.sub(r'<.*', r'', user)
      user = user.strip()
      if user:
        users.append(user)

###############################################################################
def getgroupusers():
  with open('groups.txt', 'r') as groups:
    for group in groups.readlines():
      group = group.rstrip()
      result = subprocess.run(
        [p4, "group", "-o", group],
        capture_output=True, text=True
      )
      inusers = 0
      for user in result.stdout.splitlines():
        user = user.strip()
        if re.search('^Users:', user):
          inusers = 1
          continue
        if inusers:
          if user:
            groupusers.append(user)

###############################################################################
def createuser(user):
  result = subprocess.run(
    [p4, "-c", domain, "user", "-f", "-o", user],
    capture_output=True, text=True
  )
  subprocess.run(
    [p4, "user", "-f", "-i"],
    input=result.stdout, text=True
  )

###############################################################################
def delete_users(userlist):
  for user in userlist:
    print("Deleting User: %s" % (user))
    subprocess.run([p4, "user", "-FDy", user])

###############################################################################
def cleanup():
  for f in ['users.txt', 'groups.txt', 'ztagfiles.txt']:
    if os.path.isfile(f):
      os.remove(f)

###############################################################################
def setup():
  with open('users.txt', 'w') as f:
    subprocess.run([p4, "users"], stdout=f)
  with open('groups.txt', 'w') as f:
    subprocess.run([p4, "groups"], stdout=f)
  with open('ztagfiles.txt', 'w') as f:
    subprocess.run([p4, "-Ztag", "opened", "-a"], stdout=f)

###############################################################################
def main():
  removeusers = []
  addusers = []

  setup()
  getusers()
  getgroupusers()

  for user in users:
    if user not in groupusers:
      removeusers.append(user)
      print('Adding user %s to removeusers' % user)

  for user in groupusers:
    if user not in users:
      addusers.append(user)
      print('Adding user %s to addusers' % user)

  delete_users(removeusers)

  for user in addusers:
    createuser(user)

  cleanup()

###############################################################################
if __name__ == '__main__':
  main()