'''
Created on Mar 3, 2014

@author: mikewells
'''
import os
import zipfile
import time



def zipdir(output_filename, source_dir):
    relroot = os.path.abspath(os.path.join(source_dir, os.pardir))
    with zipfile.ZipFile(output_filename, "w", zipfile.ZIP_DEFLATED) as zip:
        for root, dirs, files in os.walk(source_dir):
            # add directory (needed for empty dirs)
            zip.write(root, os.path.relpath(root, relroot))
            for file in files:
                filename = os.path.join(root, file)
                if os.path.isfile(filename): # regular files only
                    arcname = os.path.join(os.path.relpath(root, relroot), file)
                    zip.write(filename, arcname)
            
if __name__ == '__main__':
    rootLogDir = '/volume1/IT/Isis_Logs/'
    finalName = time.strftime("%d_%m_%Y") + 'LOGS'
    logdir = rootLogDir + finalName + '/'
    if not os.path.exists(logdir):
        os.makedirs(logdir)
        
    #Tomcat Servers
    servers = ["10.200.1.81","10.200.2.82","10.200.2.83","10.200.1.41","10.200.2.42","10.200.1.43"]
    targetLogDir = '/usr/local/tomcat/logs/*[0-9]*'
    for server in servers:
        print 'Fetching logs from ' + server
        print '-----------------------------'
        finalName = time.strftime("%d_%m_%Y") + 'LOGS'
        logdir = rootLogDir + finalName + '/' + server
        if not os.path.exists(logdir):
            os.makedirs(logdir)
        print 'scp -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + ':' + targetLogDir + ' ' + logdir
        print 'ssh -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + " 'rm  " + targetLogDir + "'"
        os.system('scp -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + ':' + targetLogDir + ' ' + logdir)
        os.system('ssh -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + " 'rm  " + targetLogDir + "'")
        print '-----------------------------' 
        
    servers = ["10.200.1.71","10.200.2.72","10.200.1.73"]
    targetLogDir = '/ionu/logs/*[0-9]*'
    for server in servers:
        print 'Fetching logs from ' + server
        print '-----------------------------'
        finalName = time.strftime("%d_%m_%Y") + 'LOGS'
        logdir = rootLogDir + finalName + '/' + server
        if not os.path.exists(logdir):
            os.makedirs(logdir)
        print 'scp -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + ':' + targetLogDir + ' ' + logdir
        print 'ssh -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + " 'rm  " + targetLogDir + "'"
        os.system('scp -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + ':' + targetLogDir + ' ' + logdir)
        os.system('ssh -i /var/services/homes/admin/.ssh/id_rsa ionu@' + server + " 'rm  " + targetLogDir + "'")
        print '-----------------------------' 
    print  rootLogDir + finalName + '.zip', rootLogDir + finalName   
    print 'rm -r ' + rootLogDir + finalName
    zipdir(rootLogDir + finalName + '.zip', logdir)
    os.system('rm -r ' + logdir)
    pass