Another useful python script

Today I was helping a colleague debug a program and I found myself doing the same thing over and over. I would download the log from the server, zip the file and then e-mail the zipped file to him. The second time doing this I decided this is too much work so I wrote a little python program to do all the steps. Here's how it's run:

sendFile.py --mailto "Bob Jones" --comment "Here's the log you wanted" testserver:/aplic/jboss/server/default/log/server.log

It gets the file via scp (10 lines), zips it up (another 10 lines), mails it (11 lines using a helper class). and then cleans up the temporary files (3 lines).

Here's the code minus the stuff for talking to outlook (via win32com).

#!/usr/bin/env python

from subprocess import call
import os
import zipfile
import OutlookMail

class SendFile:
    def __init__(self, dbInfo):
         self.tempfilename = None
         self.zippedfile = None
         
    def getFile(self, filename):
        fname = os.path.basename(filename)
        outname = os.path.join('/temp/', fname)
        args = ['scp', filename, outname]
        print ' '.join(args)
        if 0 != call(args):
            print "Problem getting file '%s'" % (filename)

        self.tempfilename = outname
        return self.tempfilename
        
    def zipFile(self, filename = None):
        if filename == None:
             filename = self.tempfilename
        
        zippedfile = filename + ".zip"
        zf = zipfile.ZipFile(zippedfile, 'w', zipfile.ZIP_DEFLATED) 
        zf.write(filename, os.path.basename(filename), zipfile.ZIP_DEFLATED) 
        zf.close()
        
        self.zippedfile = zippedfile
        return self.zippedfile

    def mailTo(self,  mailto, strComment = None, zipfilename = None):
        if zipfilename == None:
            zipfilename = self.zippedfile
        
        print "Mailing %s to %s" % (zipfilename , options.strMailTo)
        
        if strComment == None:
            strComment = 'Em anexo'
        
        shortzipfilename = os.path.basename(zipfilename)
        OutlookMail.sendToOutlook(strTos=options.strMailTo, strSubject=shortzipfilename, strBody=strComment, files = [zipfilename])

    def cleanup(self):
        os.remove(self.tempfilename)
        os.remove(self.zippedfile)
        
if __name__ == "__main__":
    import optparse
    
    parser = optparse.OptionParser()
    parser.add_option("", "--mail", "--mailto", dest="strMailTo", default=None,
        help="Who to send mail to")
    parser.add_option("-c", "--comment", dest="strComment", default=None,
        help = "Some info on the file")
    (options, args) = parser.parse_args()
    
    sf = SendFile(options.strMailTo)
    sf.getFile(args[0])
    sf.zipFile()
    sf.mailTo(options.strMailTo, options.strComment)
    sf.cleanup()

It's missing the ability to set the subject and to send multiple files, and maybe it should have some intelligence not to zip file which are already small or already compressed, but it's usable. Taking a 15 minute break to make a little utility like this is a lot more fun repeating the same boring button clicks over and over again.