DokuWiki

This page is read only. You can view the source, but not change it. Ask your administrator if you think this is wrong.
  import sys
  import subprocess
  import time
  import pexpect
  import smtplib
  
  
  #while dblock is 1 script will fail
  
  
  
  dblock = 1
  CPTMGMT = '10.30.165.21'
  cpstat_output = 'set a true value so script fails if ssh does not run'
  pushpolicy_output = 'set a value here so i know nothing got set at the end'
  CPSTATCMD = 'cpstat'
  PPCMD = 'fwm load -p firewall -m MKTX_DC03 DC03CPTFW'
  
  
  
  #or this, this is the mail alert in case we have any problems pushing policy
  def mail_alert_push(ipaddr,message):
          SERVER = "ash1mail.mycompany4.com"
          FROM = "CheckPointPushAlerts@crpashnetdev01.com"
          TO = 'network@mycompany4.com'
          SUBJECT = "CHECKPOINT PUSH FAILED FOR PUSH " + ipaddr ,  message
          TEXT = "ALERT! The push policy script was not able to run!!!!!!" + message
          message = 'TO: %s ' % (TO) + '\n' + 'Subject: %s\n\n%s' % (SUBJECT, TEXT)
          server = smtplib.SMTP(SERVER)
          server.sendmail(FROM, [TO], message)
          server.quit()
  
  #if we are lucky this is the only alert we will hit
  def mail_alert_final(ipaddr,output):
          SERVER = "ash1mail.mycompany4.com"
          FROM = "CheckPointPushAlerts@crpashnetdev01.com"
          TO = 'network@mycompany4.com'
          SUBJECT = "CHECKPOINT POLICY HAS BEEN PUSHED" #real good dun duh dun dun duh dun dun nun na nun nun nah
          TEXT = "Checkpoint Policy has been pushed via the push policy script"
          message = 'TO: %s ' % (TO) + '\n' + 'Subject: %s\n\n%s \n\n%s' % (SUBJECT, TEXT, output)
          server = smtplib.SMTP(SERVER)
          server.sendmail(FROM, [TO], message)
          server.quit()
    
  try:
      conn = pexpect.spawn('ssh admin@10.30.165.21')
      conn.expect('CPTMGMT')
  except pexpect.TIMEOUT:
      print ('icouldnt ssh in')
      fail_message = ' SSH FAILED'
      mail_alert_push(CPTMGMT, fail_message)
      sys.exit('ssh failed')
  
  conn.sendline('cpstat mg')
  conn.expect('CPTMGMT') 
  cpstat_output = conn.before.split(b'\n')
  conn.sendline('exit')
  cpstat_output = " ".join(str(elm) for elm in cpstat_output)
  
  while dblock is 1:
      try:
          if "true" in cpstat_output:
              dblock = 1
              fail_message = 'the database is locked'
              print('im in the if statement and the dblock value is :' , dblock)
              mail_alert_push(CPTMGMT, fail_message)
              sys.exit(1)
  
          else:
              dblock = 0
              print('im in the else statement and the dblock value is :' , dblock , "so im going to continue on")
  
      except:
           dblock = 1
           "there was an error"
           fail_message = ' the try except failed to check dblock'
           mail_alert_push(CPTMGMT, fail_message)
           sys.exit('failed in except')
  
  ppcmd_output = "this value is before the command ran, this is my comment and my value"
  
  #if the value is 0 that means dblock is off so im going to move forward with the script
  if dblock == 0:
      
  
      conn = pexpect.spawn('ssh admin@10.30.165.21')
      conn.expect('\[Expert')
      
      try:
          conn.sendline('fwm load -p firewall -m MKTX_DC03 DC03CPTFW')
          conn.expect('\[Expert', timeout=330)
      except pexpect.TIMEOUT:
          print ('THERE WAS A TIMEOUT PUSHING POLICY')
          fail_message = 'PUSH TIMEOUT ERROR!!!!!'
          mail_alert_push(CPTMGMT, fail_message)        
          sys.exit('POLICY PUSH FAILED!!!!!!!!!!!!!')   
      
      ppcmd_output = conn.before.split(b'\n')
      conn.sendline('exit')
      #ppcmd_output = " ".join(str(elm) for elm in ppcmd_output)
  
  
      print ('i know this is after the push policy command ran so it should have a bunch of data in it')
      mail_alert_final(CPTMGMT, ppcmd_output)    
      #send output in email
  
  else:
      print('send out error email')
      fail_message = ' SOMETHING UNEXPECTED HAPPENED WITH PUSH POLICY SCRIPT'
      mail_alert_push(CPTMGMT, fail_message)
      sys.exit('i failed at the end')