Subversion Repositories SmartDukaan

from docutils.utils.math.math2html import URL
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)