Subversion Repositories SmartDukaan

from shop2020.clients import TransactionClient
from shop2020.clients.LogisticsClient import LogisticsClient
from shop2020.config.client.ConfigClient import ConfigClient
from shop2020.thriftpy.logistics.ttypes import DeliveryType
from shop2020.thriftpy.model.v1.order.ttypes import Order
import json
import urllib
import urllib2

cc = ConfigClient()
live = cc.get_property("live")

warehouseMap = {}

if live=="true":
    username = 'ecomexpress'
    password = 'Ke$3c@4oT5m6h#$'
    wayBillApi = 'http://staging.ecomexpress.in/apiv2/fetch_awb/'
    forwardApi = 'http://staging.ecomexpress.in/apiv2/manifest_awb/'
    consignee = ''
else:
    username = 'ecomexpress'
    password = 'Ke$3c@4oT5m6h#$'
    wayBillApi = 'http://staging.ecomexpress.in/apiv2/fetch_awb/'
    forwardApi = 'http://staging.ecomexpress.in/apiv2/manifest_awb/'
    consignee = 'TEST'
    
F_AWB_NUMBER = "AWB_NUMBER" 
F_ORDER_NUMBER = "ORDER_NUMBER" 
F_PRODUCT = "PRODUCT" 
F_CONSIGNEE = "CONSIGNEE" 
F_CONSIGNEE_ADDRESS1 = "CONSIGNEE_ADDRESS1" 
F_CONSIGNEE_ADDRESS2 = "CONSIGNEE_ADDRESS2" 
F_DESTINATION_CITY = "DESTINATION_CITY" 
F_PINCODE = "PINCODE" 
F_STATE = "STATE" 
F_MOBILE = "MOBILE" 
F_ITEM_DESCRIPTION = "ITEM_DESCRIPTION" 
F_PIECES = "PIECES" 
F_COLLECTABLE_VALUE = "COLLECTABLE_VALUE" 
F_DECLARED_VALUE = "DECLARED_VALUE" 
F_ACTUAL_WEIGHT = "ACTUAL_WEIGHT" 
F_VOLUMETRIC_WEIGHT = "VOLUMETRIC_WEIGHT" 
F_LENGTH = "LENGTH" 
F_BREADTH = "BREADTH" 
F_HEIGHT = "HEIGHT" 
F_PICKUP_NAME = "PICKUP_NAME" 
F_PICKUP_ADDRESS_LINE1 = "PICKUP_ADDRESS_LINE1" 
F_PICKUP_ADDRESS_LINE2 = "PICKUP_ADDRESS_LINE2" 
F_PICKUP_PINCODE = "PICKUP_PINCODE" 
F_PICKUP_PHONE = "PICKUP_PHONE" 
F_PICKUP_MOBILE = "PICKUP_MOBILE" 
F_RETURN_NAME = "RETURN_NAME" 
F_RETURN_ADDRESS_LINE1 = "RETURN_ADDRESS_LINE1" 
F_RETURN_ADDRESS_LINE2 = "RETURN_ADDRESS_LINE2" 
F_RETURN_PINCODE = "RETURN_PINCODE" 
F_RETURN_PHONE = "RETURN_PHONE" 
F_RETURN_MOBILE = "RETURN_MOBILE" 
F_DG_SHIPMENT = "DG_SHIPMENT" 
F_INVOICE_NUMBER = "INVOICE_NUMBER" 
F_INVOICE_DATE = "INVOICE_DATE" 
F_ITEM_CATEGORY = "ITEM_CATEGORY" 
F_PACKING_TYPE = "PACKING_TYPE" 
F_PICKUP_TYPE = "PICKUP_TYPE" 
F_RETURN_TYPE = "RETURN_TYPE" 
F_PICKUP_LOCATION_CODE = "PICKUP_LOCATION_CODE" 
F_SELLER_GSTIN = "SELLER_GSTIN" 
F_GST_HSN = "GST_HSN" 
F_GST_ERN = "GST_ERN" 
F_GST_TAX_NAME = "GST_TAX_NAME"
F_GST_TAX_BASE = "GST_TAX_BASE"
F_GST_TAX_RATE_CGSTN = "GST_TAX_RATE_CGSTN"
F_GST_TAX_RATE_SGSTN = "GST_TAX_RATE_SGSTN"
F_GST_TAX_RATE_IGSTN = "GST_TAX_RATE_IGSTN"
F_GST_TAX_TOTAL = "GST_TAX_TOTAL"

#returns list of awb of specified type
#will generate 500 awbs at once
def generate_awb(deliveryType):
    if deliveryType == DeliveryType.COD:
        ecomDeliveryType = "COD"
    if deliveryType == DeliveryType.PREPAID:
        ecomDeliveryType = "PPD"
    values = {'username' : username,
              'password' : password,
              'count' : '500',
              'type' : ecomDeliveryType }

    data = urllib.urlencode(values)
    req = urllib2.Request(wayBillApi, data)
    response = urllib2.urlopen(req)
    the_page = response.read()
    return json.loads(the_page)['awb']

#All orders awbwise
#{'awb':[o1, o2, o3]}
def forward_request(ordersMap): 
    values = {'username' : username,
              'password' : password,
              'json_input' : _forwardMap(ordersMap),
              }
    data = urllib.urlencode(values)
    req = urllib2.Request(wayBillApi, data)
    response = urllib2.urlopen(req)
    the_page = response.read()
    print the_page
    
def _forwardMap(ordersMap):
    shipments = []
    for orders in ordersMap.itervalues():
        shipments.append(_createShipment(orders))
    return shipments    
    
        
def _createShipment(orders):
    shipment = {}
    total_payable = 0
    total_pieces = 0
    declared_value = 0
    total_weight = 0
    order1 = Order()
    for order1 in orders:
        total_payable += order1.net_payable_amount
        total_pieces += order1.lineitems[0].quantity
        declared_value += order1.total_amount + order1.shippingCost
        total_weight += order1.total_weight
    order = Order()
    warehouseAddressObj = get_shipper_object(order.warehouse_address_id)
    shipment[F_AWB_NUMBER] = order.airwaybill_no
    shipment[F_ORDER_NUMBER] = order.logisticsTransactionId 
    shipment[F_PRODUCT] = 'COD' if order.cod and total_payable > 0 else 'PPD' 
    shipment[F_CONSIGNEE] = order.customer_name 
    shipment[F_CONSIGNEE_ADDRESS1] = order.customer_address1 
    shipment[F_CONSIGNEE_ADDRESS2] = order.customer_address2
    shipment[F_DESTINATION_CITY] = order.customer_city 
    shipment[F_PICKUP_PINCODE] = order.customer_pincode 
    shipment[F_STATE] = order.customer_state
    shipment[F_MOBILE] = order.customer_mobilenumber 
    shipment[F_ITEM_DESCRIPTION] = '' 
    shipment[F_PIECES] = total_pieces
    shipment[F_COLLECTABLE_VALUE] = total_payable if order.cod else 0
    shipment[F_DECLARED_VALUE] = declared_value 
    shipment[F_ACTUAL_WEIGHT] = total_weight 
    #shipment[F_VOLUMETRIC_WEIGHT]
    shipment[F_LENGTH] = 10
    shipment[F_BREADTH] = 10
    shipment[F_HEIGHT] = 10
    shipment[F_PICKUP_NAME] = warehouseAddressObj.contact_person
    shipment[F_PICKUP_ADDRESS_LINE1] = warehouseAddressObj.address
    #shipment[F_PICKUP_ADDRESS_LINE2] = ''
    shipment[F_PICKUP_PINCODE] = warehouseAddressObj.pin
    shipment[F_PICKUP_MOBILE] = warehouseAddressObj.contact_number
    shipment[F_PICKUP_PHONE] = warehouseAddressObj.contact_number
    
    shipment[F_RETURN_NAME] = warehouseAddressObj.contact_person
    shipment[F_RETURN_ADDRESS_LINE1] = warehouseAddressObj.address
    #shipment[F_RETURN_ADDRESS_LINE2] = ''
    shipment[F_RETURN_PINCODE] = warehouseAddressObj.pin
    shipment[F_RETURN_MOBILE] = warehouseAddressObj.contact_number
    shipment[F_RETURN_PHONE] = warehouseAddressObj.contact_number
    shipment[F_DG_SHIPMENT] = 'false'
    shipment["ADDITIONAL_INFORMATION"] = {"MULTI_SELLER_INFORMATION": _getMultiSellerInfo(orders)} 
    
def _getMultiSellerInfo(orders):
    pass

def main():
    lc = LogisticsClient().get_client()
    print lc.getEmptyAWB(48, '731467')
    
    

if __name__ == '__main__':
    ordersMap = {}
    forward_request(ordersMap)
    
    
    
    
    
    
    
def get_shipper_object(address_id):
    global shipperMap
    if shipperMap.has_key(address_id):
        return shipperMap.get(address_id)
    else:
        tc = TransactionClient().get_client()
        info = tc.getBuyerByWarehouse(address_id)
        if info is None:
            raise RuntimeError("Buyer address mapping is None")
        shipperMap[address_id] = info
    return shipperMap.get(address_id)