dokucama

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')