Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
This script alerts the Operations team if we are running out
of Awb Numbers.

@author : Chandranshu
'''
import optparse
import sys

if __name__ == '__main__' and __package__ is None:
    import os
    sys.path.insert(0, os.getcwd())

from shop2020.clients.LogisticsClient import LogisticsClient
from shop2020.utils.EmailAttachmentSender import mail

to_addresses = ["cnc.center@shop2020.in", "chaitnaya.vats@saholic.com", "sunil.kumar@shop2020.in"]
from_user = "cnc.center@shop2020.in"
from_pwd = "5h0p2o2o"

def alert(provider_id, type, threshold):
    logistics_client = LogisticsClient().get_client()
    provider = logistics_client.getProvider(provider_id)
    count = logistics_client.getFreeAwbCount(provider_id, type)
    print count
    if count <= threshold:
        text = "Only %d AWB Numbers left for Bluedart. Kindly get it replineshed." % count 
        mail(from_user, from_pwd, to_addresses, "AWB alert for " + provider.name + " for " +  type, text)

def main():
    parser = optparse.OptionParser()
    parser.add_option("-p", "--provider", dest="provider",
                   default="1", type="int",
                   help="The PROVIDER this alert is for",
                   metavar="PROVIDER")
    parser.add_option("-t", "--type", dest="type",
                   default="Prepaid", type="string",
                   help="TYPE of AWB nos. to track",
                   metavar="TYPE")
    parser.add_option("-T", "--threshold", dest="threshold",
                   default="200", type="int",
                   help="The THRESHOLD below which the alert should be raised",
                   metavar="THRESHOLD")
    (options, args) = parser.parse_args()
    if len(args) != 0:
        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
    alert(options.provider, options.type, options.threshold)

if __name__ == '__main__':
    main()