Subversion Repositories SmartDukaan

#!/usr/bin/python

import optparse
import sys
from elixir import session

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

from shop2020.clients.LogisticsClient import LogisticsClient
from shop2020.thriftpy.logistics.ttypes import DeliveryType

from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException
from shop2020.model.v1.order.impl import DataService
from shop2020.model.v1.order.impl.DataAccessors import close_session, get_orders_in_batch

def update_awb(order):
    logistics_client = LogisticsClient().get_client()
    item_id = order.lineitems[0].item_id
    logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, item_id, DeliveryType.COD)
    print order.id,  "OLD AWB: " + order.airwaybill_no  + "  NEW AWB"+ logistics_info.airway_billno
    order.airwaybill_no = logistics_info.airway_billno
    order.tracking_id = order.airwaybill_no
    session.commit()



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")
    (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?")
    DataService.initialize(db_hostname=options.hostname, echoOn=False)

    try:
        orders = get_orders_in_batch([2,3,5,35,36,37,38,39], 0, 0, 0)
        for order in orders:
            #print order.cod, order.logistics_provider_id, order.airwaybill_no
            if not (order.cod == 1 and  order.logistics_provider_id == 1 and order.id < 132904):
                continue
            #print order.id, order.cod, order.logistics_provider_id, order.airwaybill_no
            update_awb(order)
            #print order.airwaybill_no
    except TransactionServiceException as tsex:
        print tsex.message
    finally:
        close_session()
    
if __name__ == '__main__':
    main()