maintain_user_from_groups.py #5

#!/usr/bin/env python
# ==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
# ------------------------------------------------------------------------------

#
# Usage:
#	maintain_user_froup_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 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)
config = utils.config
p4 = utils.p4
utils.login()
domain = (config.get('domain'))

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


###############################################################################
def getusers():
    userfile = open('users.txt', 'r')

    for user in userfile.readlines():
        user = re.sub(r'<.*', r'', user)
        user = user.strip()
        if user:
            users.append(user)

    userfile.close()


###############################################################################
def getgroupusers():
    groups = open('groups.txt', 'r')

    for group in groups.readlines():
        group = group.rstrip()
        os.system('%s group -o %s > %s.txt' % (p4, group, group))
        users = open('%s.txt' % group, 'r')
        inusers = 0
        for user in users.readlines():
            user = user.strip()
            if re.search('^Users:', user):
                inusers = 1
                continue
            if inusers:
                if user:
                    groupusers.append(user)
        users.close()
        os.remove('%s.txt' % group)

    groups.close()


###############################################################################
def createuser(user):
    os.system("%s -c %s user -f -o %s > userspec.txt" % (p4, domain, user))
    os.system("%s user -f -i < userspec.txt" % p4)
    os.remove("userspec.txt")


###############################################################################
def get_clients(clientlist, userlist):
    try:
        ztagclients = open("ztagclients.txt")
    except:
        print("Unable to open file ztagclients.txt")
        sys.exit(3)

    for line in ztagclients.readlines():
        if line[0:11] == "... client ":
            clientname = line[11:].strip()
        else:
            if line[0:10] == "... Owner ":
                owner = line[10:].strip()
                # If that client is owned by user in userlist, add it to clientlist for deletion
                if (owner in userlist):
                    print("Adding client %s to list for deletion." % (clientname))
                    clientlist.append(clientname)
    ztagclients.close()

    try:
        ztagfiles = open("ztagfiles.txt")
    except:
        print("Unable to open file ztagfiles.txt")
        sys.exit(3)

    for line in ztagfiles.readlines():
        if line[0:9] == "... user ":
            username = line[9:].strip()
        else:
            if line[0:11] == "... client ":
                clientname = line[11:].strip()
                # If that client is owned by user in userlist, add it to clientlist for deletion
                if (username in userlist):
                    if (clientname in clientlist):
                        print("Client %s already scheduled for deletion." % (clientname))
                    else:
                        print("Adding client %s to list for deletion" % (clientname))
                        clientlist.append(clientname)
    ztagfiles.close()


###############################################################################
def delete_clients(clientlist):
    for client in clientlist:
        print("Deleting Client: %s" % (client))
        os.system("%s client -f -d %s" % (p4, client))


###############################################################################
def delete_users(userlist):
    for user in userlist:
        print("Deleting User: %s" % (user))
        os.system("%s user -f -d %s" % (p4, user))


###############################################################################
def cleanup():
    os.remove('users.txt')
    os.remove('groups.txt')
    os.remove('ztagclients.txt')
    os.remove('ztagfiles.txt')


###############################################################################
def setup():
    os.system('%s users > users.txt' % p4)
    os.system('%s groups > groups.txt' % p4)
    os.system('%s -Ztag clients > ztagclients.txt' % p4)
    os.system('%s -Ztag opened -a > ztagfiles.txt' % p4)


###############################################################################
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)

    get_clients(clientlist, removeusers)
    delete_clients(clientlist)
    delete_users(removeusers)

    for user in addusers:
        createuser(user)

    cleanup()


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