Subversion Repositories SmartDukaan

#!/usr/bin/python

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

from_user = "cnc.center@shop2020.in"

from_pwd = "5h0p2o2o"

def alert(provider_id, threshold):

    logistics_client = LogisticsClient().get_client()

    provider = logistics_client.getProvider(provider_id)

    count = logistics_client.getFreeAwbCount(provider_id)

    print count

    if count <= threshold:

        text = "Only %d AWB Numbers left for Bluedart. Kindly get it replineshed." % count

        mail(from_user, from_pwd, from_user, "AWB alert for " + provider.name, 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", "--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.threshold)

if __name__ == '__main__':

    main()