p4lock.py #4

#!/usr/bin/env python3

"""
This program should lock a perforce label.

The only command line parameter is the label
p4lock.py LABEL
"""

import isitalabel
import os
import re
import subprocess
import sys

def lock (label):
  x = isitalabel.check(label)
  if x == 0:
    print("There is no label %s." % label)
    sys.exit(1)

  label_pattern = re.compile(r"^Label:\s+.*")
  lock_pattern = re.compile(r"^Options:\s+unlocked")

  linelist = []

  # owner.txt is a one line file generated by the p4unlock.py script. It saves
  # the name of the label owner so it can be put back by this script.
  # The owner line is removed by p4unlock.py because Perforce doesn't allow
  # anyone except the owner of a label to update the label.

  found_owner = True
  owner = ""
  if os.path.isfile("owner.txt"):
    with open("owner.txt", "r") as ownerfile:
      for line in ownerfile.readlines():
        owner = line
    os.remove("owner.txt")
  else:
    found_owner = False

  result = subprocess.run(
    ["p4", "label", "-o", label],
    capture_output=True, text=True
  )
  for line in result.stdout.splitlines(keepends=True):
    if found_owner:
      match = label_pattern.match(line)
      if match is not None:
        linelist.append("Owner:\t%s\n" % owner)
    line = lock_pattern.sub("Options:\tlocked", line)
    linelist.append(line)

  spec_text = "".join(linelist)
  subprocess.run(["p4", "label", "-i"], input=spec_text, text=True)

def main():
  if len(sys.argv) < 2:
    print('Invalid input')
    print('You must specify a label')
    print('usage: p4lock.py LABELNAME')
  else:
    label = sys.argv[1]
    lock(label)

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