Grupo de Usuarios de Tecnologías Libres

Código: Seleccionar todo

#! /usr/bin/python
from datetime import datetime
import shutil,subprocess, os, smtplib, gzip, tarfile
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart

now = str(datetime.now())
data = {'SERVER_DIR'    :'/home/backups/Servers/Virtue',
        'WORK_DIR'      :'/home/backups/Servers/Virtue/'+now+'.logs/',
        'WORK_CREATE'   :'/home/backups/Servers/Virtue/'+now+'.logs/1',
        'ADMINS'        :('kzkggaara@ipichcb.rimed.cu', 'elav@ipichcb.rimed.cu'),
        'SMTP_SERVER'   :'smtp.ipichcb.rimed.cu'}

def enviar_email(para, subject, texto, smtp_server):
    COMMASPACE = ', '
    text = MIMEText(texto)
    msg = MIMEMultipart()
    msg['Subject'] = subject
    msg['From'] = 'error@server'
    msg['To'] = COMMASPACE.join(para)
    msg.attach(text)
    s = smtplib.SMTP(smtp_server)
    s.sendmail('error@server', para, msg.as_string())
    s.quit()

def copiar_archivo(desde, hasta, subject, mensaje_error, data, scp = False):
    try:
        if scp:
            os.system("scp "+desde+' '+hasta)
        else:
            shutil.copy(desde, hasta)
    except IOError as (errno, strerror):
        mensaje_error = mensaje_error + '\nI/O error({0}): {1}'.format(errno, strerror)+'\n '+desde+'->'+hasta+'\n El error ocurrio ' + now
        enviar_email(data['ADMINS'], subject, mensaje_error, data['SMTP_SERVER'])
        return False
    return True

def comprimir_dir(directorio, subject, mensaje_error, data):
    try:
        tar = tarfile.open(now+'.logs.tar', 'w')
        tar.add(directorio)
        tar.close()
    except IOError as (errno, strerror):
        mensaje_error = mensaje_error + '\nI/O error({0}): {1}'.format(errno, strerror)+'\n El error ocurrio ' + now
        enviar_email(data['ADMINS'], subject, mensaje_error, data['SMTP_SERVER'])
        return False
    f_in = open(now+'.logs.tar', 'rb')
    f_out = gzip.open(now+'.logs.tar.gz', 'wb')
    f_out.writelines(f_in)
    f_out.close()
    f_in.close()
    return True

def principal():
    if not os.path.exists(data['WORK_CREATE']):
        os.mkdir(data['WORK_CREATE'])
    os.execlp('/etc/init.d/squid','', 'stop')

    mensaje_error = 'La salva de logs del servidor Virtue ha dado ERROR, se adjunta log. '
    subject = '[Virtue] | [ERROR] Status de salva de logs.'
    flag = copiar_archivo('/var/log/squid/access.*', data['WORK_DIR'], subject, mensaje_error, data)
    if not flag:
        return

    flag = copiar_archivo('/var/log/auth.*', subject, data['WORK_DIR'], mensaje_error, data)
    if not flag:
        return

    os.remove('/var/log/squid/access.*')
    os.remove('/var/log/auth.*')
    os.execlp('/etc/init.d/squid','', 'start')
    open('/var/log/auth.log', 'w')
    os.lchown('/var/log/auth.log', 'adm', 'adm')
    os.chmod('/var/log/auth.log', 640)
    os.chdir(data['SERVER_DIR'])

    mensaje_error1 = 'La salva de logs del servidor Virtue ha dado ERROR. Especificamente a la hora de COMPRIMIR el directorio, se adjunta log. '
    flag = comprimir_dir(data['WORK_DIR'], subject, mensaje_error1, data)
    if not flag:
        return

    os.remove(data['WORK_DIR'])
    
    flag = copiar_archivo(data['SERVER_DIR']+now+'.logs.tar.gz', 'root@192.168.200.4:/home/backups/Servers/Virtue/', subject, mensaje_error, data, True)
    if not flag:
        return

    subject1 = '[Virtue] | [OK] Status de salva de logs.'
    mensaje_error2 = 'La salva de logs del servidor Virtue (192.168.200.3) se ha efectuado satisfactoriamente. \n'+now
    enviar_email(data['ADMINS'], subject1, mensaje_error2, data['SMTP_SERVER'])

if __name__ == '__main__':
    principal()