Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
Alerts regarding inventory update.
'''

import sys
import MySQLdb
import optparse
import time
import datetime
import smtplib
from email.mime.text import MIMEText


parser = optparse.OptionParser()
usage = "Usage: %prog -H [DB Address]"
parser.add_option("-H", "--host", dest="host",default="",type="string",help="Database Address",metavar="HOST")
(options, args) = parser.parse_args()
if ( not(options.host)):
    print "Fatal: Required arguments are missing!"
    print "Use: -h / --help to get help."
    sys.exit(1)
db = MySQLdb.connect(options.host,"root","shop2020","inventory" )
now = datetime.datetime.now()
cursor = db.cursor()
MESSAGE=[]


def getUpdate(warehouseName,warehouseId,checkInterval):
    
    try:
        print "Warehouse Name:",warehouseName
        print "Warehouse Id:",warehouseId
        print "Time Period:",checkInterval
        print str(now)
        sql = '''SELECT lastCheckedOn FROM warehouse where id=%d''' % \
        (warehouseId)
        cursor.execute(sql)
        data = cursor.fetchone()
        print "Data [0]:",data[0]
        if (data[0] is None):
            print "Inside None if"
            mailText = 'Last Checked On is NULL for '+warehouseName+" Id = "+str(warehouseId)+"\n\n"
            print mailText
            MESSAGE.append(mailText)
        else:
            diffMinutes =int((time.mktime(now.timetuple()) - time.mktime(data[0].timetuple()))/60)
            print "Diff",diffMinutes
        if ( diffMinutes > checkInterval):
            print "Diffcheck"
            mailText = "Warehouse "+warehouseName+" Id = "+str(warehouseId)+" was last checked "+str(diffMinutes)+" minutes ago while its interval is "+str(checkInterval)+" minutes"+"\n\n"
            print mailText
            MESSAGE.append(mailText)
            print "checking timediff"
    
    except:
        "MYSQL Exception"

        
def checkWarehouseHoliday(warehouseHolidays):
    currentDay=str(now.weekday())
    print "Current Day",currentDay
    print "WarehouseHoliday",warehouseHolidays
    if currentDay in warehouseHolidays:
        return True
    else:
        return False
        
def sendMail():
    
    print MESSAGE
    msg = ''.join(MESSAGE)
    smtpServer = smtplib.SMTP('localhost')
    smtpServer.set_debuglevel(1)
    msg = MIMEText(msg)
    sender = 'inventory-monitor@shop2020.in'
    recipients = ['kshitij.sood@shop2020.in', 'amar.kumar@shop2020.in']
    msg['Subject'] = "Warehouse Inventory Monitor"
    msg['From'] = sender
    msg['To'] = ", ".join(recipients)
    
    try:
        smtpServer.sendmail(sender, recipients, msg.as_string())
        print "Successfully sent email"
    
    except:
        print "Error: unable to send email"
    

def main():
    try:
        sql = '''select * from warehouseMonitor'''
        cursor.execute(sql)
        listWarehouse = cursor.fetchall()
        for warehouse in listWarehouse:
            if ( checkWarehouseHoliday(str(warehouse[3])) ):
                continue
            if ( warehouse[4]==1 ):
                getUpdate(warehouse[1],warehouse[2],warehouse[5])
        if (MESSAGE[0] is not None):
            sendMail()    
    
    except:
        "MYSQL Exception"
    
    finally:
        print "Closing db connection"
        db.close()  
        
if __name__ == "__main__":
    main()