clean_protect.py #4

#!/usr/bin/env python3

"""
Usage:
	python clean_protect.py remove_groups.txt p4.protect

This script will drop all lines in the protect table that have a group referenced from the file
groups.txt passed into the script. The list of groups to drop is passed in as the first parameter
and the protect table is passed in as the 2nd parameter.

remove_groups.txt is generated using protect_groups.py - See that script for details.

Run "p4 protect -o > p4.protect" to generate the protections table.

You can redirect the output of this script to a file called new.p4.protect and then you can compare
the original p4.protect and the new.p4.protect. If everything looks okay, you can update the protections
table by running:

p4 protect -i < new.p4.protect

"""

import re
import sys


def main():
  if len(sys.argv) < 3:
    print("Usage: python clean_protect.py remove_groups.txt p4.protect")
    sys.exit(1)

  p4groups = []

  with open(sys.argv[1], "r") as groups:
    for group in groups.readlines():
      group = group.rstrip()
      group = group.lower()
      p4groups.append(group)

  with open(sys.argv[2], "r") as protect:
    for line in protect.readlines():
      line = line.rstrip()
      origline = line
      line = line.lower()
      match = re.match(".*group (.*) \\* .*", line)
      if match:
        if match.group(1).lower() in p4groups:
          continue
        else:
          print(origline)
      else:
        print(origline)


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