Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
Loads the static logic of assigning provider to a particular pincode
into the DestinationProviderAllocation table.

@author: Amit Gupta
'''
from datetime import *
from elixir import *
from shop2020.logistics.service.impl import DataService, EcomExpressService
from shop2020.logistics.service.impl.DataAccessor import add_empty_AWBs
from shop2020.logistics.service.impl.DataService import \
    ServiceableLocationDetails, Awb
from shop2020.logistics.service.impl.EcomExpressService import generate_awb
from shop2020.logistics.service.script import AwbNumberLoader
from shop2020.thriftpy.logistics.ttypes import DeliveryType
from shop2020.utils.EmailAttachmentSender import get_attachment_part, mail
import optparse

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



def main():
    parser = optparse.OptionParser()
    parser.add_option("-H", "--host", dest="hostname",
                  default="localhost",
                  type="string", help="The HOST where the DB server is running",
                  metavar="HOST")
    parser.add_option("-P", "--provider_id", dest="provider_id",
                  default="localhost",
                  type="string", help="The HOST where the DB server is running",
                  metavar="HOST")
    (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?")
    provider_id = options.provider_id
    hostname = options.hostname
    if provider_id is None or hostname is None:
        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
    
    DataService.initialize(dbname='logistics', db_hostname=hostname)
    if provider_id=="49":
        subject = "Tat is missing for Ecom Express active pincodes"
        missingPincodes = []
        try:
            serviceabilityPincodeList = EcomExpressService.getServiceablePinCodes(datetime.now()-timedelta(days=15))
            #print "serviceabilityPincodeList", serviceabilityPincodeList
            for serviceabilityPincode in serviceabilityPincodeList:
                pincode = str(serviceabilityPincode["pincode"])
                serviceableLocationDetails = ServiceableLocationDetails.query.filter_by(provider_id=int(provider_id)).filter_by(dest_pincode=pincode).first()
                if serviceableLocationDetails:
                    if serviceabilityPincode['active']:
                        serviceableLocationDetails.exp=1
                        serviceableLocationDetails.cod=1
                    else:
                        serviceableLocationDetails.exp=0
                        serviceableLocationDetails.cod=0
                        
                elif serviceabilityPincode['active']:
                    missingPincodes.append(pincode)
            print missingPincodes
            if missingPincodes:
                subject += " - " + ", ".join(missingPincodes)
                mail("cnc.center@shop2020.in", "5h0p2o2o", ["deena.nath@smartdukaan.com"], subject, text="")
            
            session.commit()
        except:
            print "problem updating pincodes" 
        print "After Error"
            #add_new_awbs(provider_id, isCod, awbs, awbUsedFor)
        prepaidCount = Awb.query.filter_by(provider_id = provider_id, is_available = True, type='Prepaid').count()
        if prepaidCount < 100:
            prepaidAwbs = EcomExpressService.generate_awb(DeliveryType.PREPAID)
            add_empty_AWBs(prepaidAwbs, 49, "Prepaid")
        
        codCount = Awb.query.filter_by(provider_id = provider_id, is_available = True, type='COD').count()
        if codCount < 100:
            codAwbs = EcomExpressService.generate_awb(DeliveryType.COD)
            add_empty_AWBs(codAwbs, 49, "COD")
            
    session.close()
                
        
if __name__ == '__main__':
    main()