revertAllFilesForUser.py

#!/usr/bin/python

import marshal
import subprocess, shlex
import re
import os
import sys

clienthostcache = {}

def getPerforceResponse(args):
	lst = []
	#args = shlex.split(command)
	p = subprocess.Popen(args, stdout=subprocess.PIPE)
	print "Running " + " ".join(args)
	try:
		while 1:
			dictionary = marshal.load(p.stdout)
			lst.append(dictionary)
	except EOFError:
		pass
	return lst

def login(userID):
	return getPerforceResponse(["p4", "-G", "login", "-a", userID])

def getOpenedFilesForUser(userID):
	return getPerforceResponse(["p4", "-G", "opened", "-u", userID])

def getClient(clientID):
	return getPerforceResponse(["p4", "-G", "client", "-o", clientID])[0]

def getHostForClient(clientID):
	host = "p4client"
	try:
		host = clienthostcache[clientID]
	except KeyError:
		client = getClient(clientID)
		host = client.get('Host', 'p4client')
		clienthostcache[clientID] = host
	return host

def main():
	for user in sys.argv[1:]:
		print "Logging in as " + user
		print login(user)
		openedByUser = getOpenedFilesForUser(user)
		print "Found " + str(len(openedByUser)) + " opened files for " + user
		print "reverting..."
		for f in openedByUser:
			print "Reverting " + f['depotFile'] + " on " + f['client']
			host = getHostForClient(f['client'])
			response = getPerforceResponse(["p4", "-G", "-H", host, "-c", f['client'], "-u", user, "revert", "-k", f['depotFile']])
			print response

main()