removeusersfromgroup.py #6

#!/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 = """
This script will remove the specified users from given group.

Usage:
 	removeusersfromgroup.py [instance] USER groupname
 	removeusersfromgroup.py [instance] FILE groupname

USER can be a single username or, it can be a FILE with a list of users.

instance defaults to 1 if not given.
"""

import os
import sys
import sdputils

if len(sys.argv) > 3:   # 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()


def removeuser(username, groupname, groupfile):
    newgroupfile = groupfile + ".new"
    infile = open(groupfile, "r")
    outfile = open(newgroupfile, "w")
    for line in infile.readlines():
        if username.lower() != (line.strip()).lower():
            outfile.write(line)
    infile.close()
    outfile.close()
    os.remove(groupfile)
    os.rename(newgroupfile, groupfile)


def cleanup(groupfile):
    print("Cleaning temp files")
    try:
        os.remove(groupfile)
    except:
        print("Failed to delete " + groupfile)


def main():
    if len(sys.argv) < 3:
        print(USAGE)
        sys.exit(1)

    # Handle the optional instance parameter shift
    if len(sys.argv) > 3:
        userorfile = (sys.argv[2])
        groupname = sys.argv[3].strip()
    else:
        userorfile = (sys.argv[1])
        groupname = sys.argv[2].strip()

    global initialized
    initialized = 0
    groupfile = "%s.group" % groupname
    os.system("%s group -o %s > %s" % (p4, groupname, groupfile))

    try:
        userlistfile = open(userorfile, "r")
        for line in userlistfile.readlines():
            removeuser(line.strip(), groupname, groupfile)
    except:
        print("No file %s available, assuming it is the actual user name." % userorfile)
        removeuser(userorfile, groupname, groupfile)

    os.system("%s group -i < %s" % (p4, groupfile))
    cleanup(groupfile)


if __name__ == '__main__':
    main()