import re, sys, time, os, java

if len(sys.argv)==0:

    print __doc__

    sys.exit(1)

# timing of loop to collect measure

loopTime=int(sys.argv[0])

print "Loop Time: " + str(loopTime)

perfObject = AdminControl.completeObjectName('type=Perf,*') #@UndefinedVariable - comes from WAS

if perfObject is None or perfObject == "":

    print "Can not retrieve the performance MBean. Make sure PMI (performance metrics) is enabled on the server"

perfOName = AdminControl.makeObjectName(perfObject) #@UndefinedVariable

objNameString = AdminControl.completeObjectName('WebSphere:type=Server,*')

processName =  AdminControl.getAttribute(objNameString, 'name') 

# Output file for collections

OutFile = '../Collection/SysMetrics.' + processName + "_" + time.strftime("%d%m%Y", time.localtime()) + '.csv'

print "output Stored in: " + OutFile

writeHeader=1

try:

    if os.path.isfile(OutFile):

        writeHeader=0 # do not write a header to a file that is being appended 

    f_out=open(OutFile,"a")

except:

    print "Error opening file %s\n%s" % (OutFile,sys.exc_info()[1])

    sys.exit(2)

if len(sys.argv) > 1:

    print "Argomento 2: " + sys.argv[1]

    if sys.argv[1]=='-l':

        # list objects with stats

        print "Listing objects ..."

        for a in AdminControl.queryNames('*').split('\n'): #type=ConnectionPool, @UndefinedVariable

            try:

                config=AdminControl.invoke_jmx(perfOName, 'getConfig', [AdminControl.makeObjectName(a)], ['javax.management.ObjectName']) 

                if config is not None:

                    print       f_out, a

            except:

                pass

        print "done."

        sys.exit(0)

# Enable PMI data.

AdminControl.invoke_jmx (perfOName, 'setStatisticSet', ['all'], ['java.lang.String']) 

WASobjects={}

# definition object 

WASobjects['JVM']=AdminControl.makeObjectName(AdminControl.completeObjectName('type=JVM,*')) 

WASobjects['MQThread'] = AdminControl.makeObjectName(AdminControl.completeObjectName('name=WebContainer,type=ThreadPool,*')) 

WASobjects['System'] = AdminControl.makeObjectName(AdminControl.completeObjectName('name=SystemMetrics,*')) 

namelist=['Date','Time'] 

# list all possible data to collect

for obj in WASobjects.keys():

    statconfig=AdminControl.invoke_jmx(perfOName, 'getConfig', [WASobjects[obj]], ['javax.management.ObjectName']) 

    for d in statconfig.listAllData():

        namelist.append(obj+d.name)

if writeHeader: 

    print     f_out,",".join(namelist)

def run(cmd):

  process = java.lang.Runtime.getRuntime().exec(cmd)

  stdoutstream = '' 

  errorstream = ''

  running = 1 

  while running: 

    while process.getInputStream().available(): # > 0: 

        stdoutstream += chr(process.getInputStream().read())

    while process.getErrorStream().available(): # > 0: 

        errorstream += chr(process.getErrorStream().read())

    try: 

        process.exitValue() 

        running = 0 

        print "done..." 

        return (stdoutstream, errorstream)

    except java.lang.IllegalThreadStateException, e:

        #print "running..." 

#pass

time.sleep(0.1)

#return stdoutstream, errorstream

def get_value(s):

    value=1

    if str(s.getClass())=='com.ibm.ws.pmi.stat.BoundedRangeStatisticImpl': 

        value=s.current

    elif str(s.getClass())=='com.ibm.ws.pmi.stat.CountStatisticImpl': 

        value=s.count

    elif str(s.getClass())=='com.ibm.ws.pmi.stat.DoubleStatisticImpl':

        value=s.double

    elif str(s.getClass())=='com.ibm.ws.pmi.stat.TimeStatisticImpl':  

        value=s.count

    elif str(s.getClass())=='com.ibm.ws.pmi.stat.RangeStatisticImpl':   

        value=s.current

    elif str(s.getClass())=='com.ibm.ws.pmi.stat.AverageStatisticImpl': 

        value=s.count   

    else:

        print s.getClass()

    return value

print "Pulling WebSphere statistics...Press Ctrl-C to interrupt"

cycle=0

while cycle<=1:

    try:

        statshash={}

        for t in namelist:

            if t=="Date":

                statshash[t]=time.strftime("%m/%d/%Y", time.localtime())

            elif t=="Time":

                statshash[t]=time.strftime("%H:%M:%S", time.localtime())

            else:

                statshash[t]=""

        for obj in WASobjects.keys():

            stats=AdminControl.invoke_jmx(perfOName, 'getStatsObject', [WASobjects[obj], java.lang.Boolean ('false')], ['javax.management.ObjectName', 'java.lang.Boolean'])

            for s in stats.statistics:

                statshash[obj+s.name]=get_value(s)

            print     f_out,",".join([str(statshash[v]) for v in namelist])

    except:

            print "%s. Encountered a glitch: %s, %s" % (time.strftime("%m/%d/%Y %H:%M:%S", time.localtime()),sys.exc_info()[0],sys.exc_info()[1])

   if loopTime == -1:

      cycle=cycle+1

    if cycle<1: 

       time.sleep(loopTime) # collection delay

#

# Script to use Parametric method to call a Collection data

# menu item

#

inputData()

{

echo -n "Enter process Id :"

echo "(name of the process make the collection)"

read procesId

echo -n "Enter ip or Fqdn of Server to Collect : " 

read hostIp

echo -n "Enter Port where the JVM listen : " 

read soapPort

echo -n "Enter Admin Id: " 

read wasAdmin

echo -n "Enter Password : " 

read wasPwd

export wasHost=$hostIp

export wasSoapPort=$soapPort

export wasAdminId=$wasAdmin

export wasAdminPwd=$wasPwd

}

# make script to one time run, the script collect until user press ctrl-C in terminal session.

oneTime()

{

echo

echo

echo "Start one time WsAdmin Collection" ;

echo

inputData

echo -n "Enter delay of cycle (in seconds) : " 

read timeCycle

echo "press any key to start......"

read ss

scriptId=/opt/juh/Script/$procesId.sh

echo cd /opt/juh/Script/ > $scriptId

echo /opt/juh/adminClient/wsthinadmin.sh -lang jython -host $hostIp -port $soapPort -user $wasAdmin -password $wasPwd -f /opt/juh/jython/OneCollect.py $timeCycle       $scriptId

chmod 771 $scriptId

echo "Configure Wsadmin Thin Client"

python /opt/juh/jython/setSoapClient.py /opt/juh/adminClient/properties/soap.client.props

echo " Start Collection"

$scriptId

}

#

# make script to schedule run, the script collects one time any run.

pulTime()

{

echo

echo

echo "Start puling WsAdmin Collection" ;

echo

inputData

echo -n "Enter delay of cycle (in seconds) : " 

read timeCycle

echo "press any key to start......"

read ss

scriptId=/opt/juh/Script/skd_$procesId.sh

echo cd /opt/juh/Script/ > $scriptId

echo /opt/juh/adminClient/wsthinadmin.sh -lang jython -host $hostIp -port $soapPort -user $wasAdmin -password $wasPwd -f /opt/juh/jython/collectSkd.py $procesId -1       $scriptId

chmod 771 $scriptId

}

#

tester()

{

echo

inputData

python ./writefile.py /opt/juh/adminClient/properties/soap.client.props

}

#

cd /opt/juh/Script/

clear

while :

 do

    clear

    echo "-------------------------------------"

    echo " Main Menu "

    echo "-------------------------------------"

    echo "[1] Start one time WsAdmin Collection"

    echo "[2] Start puling WsAdmin Collection"

    echo "[5] Exit/Stop"

    echo "======================="

    echo -n "Enter your menu choice [1-5]: "

    read yourch

    case $yourch in

      1) oneTime ; exit 0 ;;

      2) pulTime ; exit 0 ;;

      5) exit 0 ;;

      *) echo "Opps!!! Please select choice 1,2 or 5";

         echo "Press a key. . ." ; read ;;

 esac

done

import sys,os

fileb=sys.argv[1]+".bak"

filet=sys.argv[1]

print fileb

print filet

os.rename (filet,fileb)

f = open(fileb, "r")    # read the file by lines

lines = f.readlines()

f.close()

wasUser= os.environ['wasAdminId']

wasPwd= os.environ['wasAdminPwd']

ft =  open( filet, "w" ) #as target:

for line in lines:

    for word in line.split():

      if line == "\n":

          print ">"+line+ "<"

          break

      if line[0] == '#':

         ft.write(line)

         break

      if word[:25] == 'com.ibm.SOAP.loginUserid=':

         ft.write(word[:25]+wasUser+"\n")

      else:

         if word[:27] == "com.ibm.SOAP.loginPassword=":

            ft.write(word[:27]+wasPwd+"\n")

         else:

            if word[:4] == "com.":

               ft.write(word+"\n")

ft.close()