#!/usr/bin/env python
#
#*******************************************************************************
#
#Copyright (c) 2009, Perforce Software, Inc.  All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# 1.  Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 
# 2.  Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL PERFORCE SOFTWARE, INC. BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#*******************************************************************************
#
#Author: Stephen Moon
#Date: 7/17/2010
#Last Modifed: 7/24/2010
#Summary: Delete old clients
#
#*******************************************************************************

from subprocess import Popen,PIPE,STDOUT
from datetime import date,datetime,time,timedelta
import sys,os,re,time,logging,smtplib

#Enable logging of the backup script
logging.basicConfig(level=logging.DEBUG,
                    format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s',
                    datefmt='%m-%d %H:%M',
                    filename='p4client.log',
                    filemode='w')
# define a Handler which writes INFO messages or higher to the sys.stderr
console = logging.StreamHandler()
console.setLevel(logging.INFO)
# set a format which is simpler for console use
formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
# tell the handler to use this format
console.setFormatter(formatter)
# add the handler to the root logger
logging.getLogger('').addHandler(console)

#define all the environmental variables
p4debug = logging.getLogger('p4debug')
p4error = logging.getLogger('p4error')

#p4 = "c:\program files\perforce\p4.exe"
p4 = "c:\p4root\p4.exe"

mailhost = 'smtp.perforce.com'
server = 'localhost'
port = '20101'

P4USER = 'smoon'
#P4CHARSET = 'utf8'  #set it to empty string if your server is not Unicode
P4PORT = server + ':' + port

os.environ['P4PORT'] = P4PORT
os.environ['P4USER'] = P4USER
#os.environ['P4CHARSET'] = P4CHARSET

class Client(object):

   def __init__(self):
      self.name = ""
      self.access = "" 
      self.owner = "" 

   def getName(self):
      return self.name

class ClientDict(dict):

   def addClient(self,obj):
      self[obj.getName()] = obj
  
def getClientInfo(clientsOutput):

   matchClient = "^\.\.\.\sclient\s(\S+)"
   matchAccess = "^\.\.\.\sAccess\s(\d+)"
   matchOwner  = "^\.\.\.\sOwner\s(\S+)"

   clientMatched = re.compile(matchClient)
   accessMatched = re.compile(matchAccess)
   ownerMatched = re.compile(matchOwner)

   clientList = ClientDict()
   for line in clientsOutput:
   
      mClient = clientMatched.match(line)
      mAccess = accessMatched.match(line)
      mOwner = ownerMatched.match(line)

      if mClient is not None:
         clientObj = Client()

         #print mClient.group(1) 
         clientObj.name = mClient.group(1)

      if mAccess is not None:
         #print mAccess.group(1)
         clientObj.access = mAccess.group(1)

      if mOwner is not None:
         #print mOwner.group(1)
         clientObj.owner = mOwner.group(1)
	 clientList.addClient(clientObj)
   
   return clientList

def getClients(today,oldDate):


   getClients = [p4,'-ztag','clients']
   p = Popen(getClients,shell=True,stdout=PIPE,stderr=PIPE)
   clientList = p.stdout.readlines()
   clientErr = p.communicate()[1]

   clients = getClientInfo(clientList)

   p.stdout.close()
   p.stderr.close()

   if clientErr == "":
      for k,v in clients.items():
	 clientDate = date.fromtimestamp(float(v.access))
	 if clientDate > oldDate: 
	    print("%s, %s, %s" % (k,date.fromtimestamp(float(v.access)),v.owner))
	    p4debug.exception("keep: %s, %s, %s" % (k,date.fromtimestamp(float(v.access)),v.owner))
	 else:
	    #print("%s, %s, %s" % (k,date.fromtimestamp(float(v.access)),v.owner))
	    p4debug.exception("delete: %s, %s, %s" % (k,date.fromtimestamp(float(v.access)),v.owner))
      
   else:
      sys.stderr.write(clientErr) 
      p4error.exception('Unable to list clients: \n%s' % clientErr)
      exit(1)

def main():
   inputDays = raw_input("Enter the number of days that the clients have\nnot been used: ") 
   today = date.today()
   #print("Today: %s" % today)
  
   oldDate = timedelta(days=int(inputDays))
   #print("OldDAy: %s" % (today - oldDate))
   getClients(today,today - oldDate)

if __name__ == '__main__':
   main()