Subversion Repositories SmartDukaan

#!/usr/bin/python

import optparse

import time

import datetime

import sys

from elixir import session

from random import randrange

from shop2020.clients.LogisticsClient import LogisticsClient

from shop2020.clients.CatalogClient import CatalogClient

from shop2020.clients.InventoryClient import InventoryClient

from shop2020.model.v1.order.impl.DataService import Attribute

from shop2020.thriftpy.logistics.ttypes import DeliveryType, PickUpType

if __name__ == '__main__' and __package__ is None:

    import os

    sys.path.insert(0, os.getcwd())

from shop2020.thriftpy.model.v1.order.ttypes import OrderStatus, TransactionServiceException, OrderType

from shop2020.model.v1.order.impl import DataService

from shop2020.model.v1.order.impl.DataService import Order, LineItem, Transaction

from shop2020.model.v1.order.impl.DataAccessors import get_order, close_session

from shop2020.model.v1.order.impl.DataAccessors import delhi_pincodes

def cancel(order):

    '''

    Cancel

    '''

    print("Your session has been closed")

    return

def process_order(order):

    '''

    Process Order

    '''

    if order.total_amount < 10998:

        print "Order already processed"

        return

    bulk_order = Order()

    bulk_order.transaction = order.transaction

    bulk_order.customer_id = order.customer_id

    bulk_order.customer_email = order.customer_email

    bulk_order.customer_name = order.customer_name

    bulk_order.customer_pincode = order.customer_pincode

    bulk_order.customer_address1 = order.customer_address1

    bulk_order.customer_address2 = order.customer_address2

    bulk_order.customer_city = order.customer_city

    bulk_order.customer_state = order.customer_state

    bulk_order.customer_mobilenumber = order.customer_mobilenumber

    bulk_order.total_amount = order.total_amount            

    bulk_order.total_weight = order.total_weight

    bulk_order.status = OrderStatus.ACCEPTED

    bulk_order.statusDescription = "In Process";

    bulk_order.created_timestamp = order.created_timestamp

    bulk_order.accepted_timestamp = datetime.datetime.now()

    bulk_order.cod = order.cod

    bulk_order.orderType = OrderType.B2Cbulk

    bulk_order.pickupStoreId = 0

    bulk_order.otg = 0

    bulk_order.insurer = 0

    bulk_order.insuranceAmount = 0

    logistics_client = LogisticsClient().get_client()

    logistics_info = logistics_client.getLogisticsInfo(order.customer_pincode, 1, DeliveryType.PREPAID, PickUpType.SELF)

    bulk_order.logistics_provider_id = logistics_info.providerId;

    bulk_order.tracking_id = logistics_info.airway_billno

    bulk_order.airwaybill_no = logistics_info.airway_billno

    bulk_order.expected_delivery_time = order.expected_delivery_time

    bulk_order.promised_delivery_time = order.promised_delivery_time

    bulk_order.expected_shipping_time = order.expected_shipping_time

    bulk_order.promised_shipping_time = order.promised_shipping_time

    catalog_client = CatalogClient().get_client()

    item = catalog_client.getItem(1173)

    litem = LineItem()

    litem.item_id = item.id

    litem.productGroup = item.productGroup

    litem.brand = item.brand

    litem.model_number = item.modelNumber

    litem.model_name = item.modelName

    litem.color = item.color

    litem.extra_info = "Complementary Order for A110,  Order ID " + str(order.id)

    litem.quantity = 1

    litem.unit_price = order.total_amount - order.insuranceAmount - 9999

    litem.unit_weight = item.weight

    litem.total_price = order.total_amount - order.insuranceAmount - 9999

    litem.total_weight = item.weight

    litem.vatRate = 5.0

    bulk_order.lineitems.append(litem)

    bulk_order.total_amount = order.total_amount - order.insuranceAmount - 9999

    order.total_amount = 9999 + order.insuranceAmount

    order.lineitems[0].unit_price = 9999

    order.lineitems[0].total_price = 9999

    inventory_client = InventoryClient().get_client()

    item_availability = inventory_client.getItemAvailabilityAtLocation(bulk_order.lineitems[0].item_id,1)

    bulk_order.warehouse_id = item_availability[2]

    bulk_order.fulfillmentWarehouseId = item_availability[0]

    session.commit()

    print "New Bulk OrderId : " + str(bulk_order.id)    

    return 

ACTIONS = {0: cancel,

           1: process_order

           }

def get_py_datetime(time_string):

    time_format = "%Y-%m-%d %H%M"

    mytime = time.strptime(time_string, time_format)

    return datetime.datetime(*mytime[:6])

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("-O", "--order", dest="orderId",

                      default =0,

                      type="int", help="A110 OrderId to be processed",

                      metavar="NUM")

    (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?")

    if options.orderId == 0:

        print "Usage : -O OrderId -H Hostname"

        return

    DataService.initialize(db_hostname=options.hostname, echoOn=False)

    orderId = options.orderId

    try:

        order = get_order(orderId)

        print("Please check the details of the order below and ensure that it's the same order which you want to modify:")

        print("Order Id:\t\t" + str(order.id))

        print("Customer Name:\t\t" + order.customer_name)

        print("Pincode:\t\t" + order.customer_pincode)

        print("Amount:\t\t\t" + str(order.total_amount))

        print("Created On:\t\t" + str(order.created_timestamp))

        print("Current Status:\t\t" + str(order.status))

        print("Status Description:\t\t" + order.statusDescription)

        print("Logistics provider id:\t\t" + str(order.logistics_provider_id))

        print("Airway bill number:\t\t" + order.airwaybill_no)

        print("Ordered Items description:")

        for lineitem in order.lineitems:

            print("Item Id:" + str(lineitem.item_id) + "\tBrand: " + str(lineitem.brand) + "\tModel: " + str(lineitem.model_number) + "\tColor: " + str(lineitem.color))

        print

        print

        print("You can perform following operations:")

        for (key, val) in ACTIONS.iteritems():

            print("[" + str(key) + "]" + val.__doc__ )

        raw_action = raw_input("What do you want to do? ")

        if raw_action is None or raw_action == "":

            print("Your session has been closed.")

            return

        try:

            action = int(raw_action)

        except ValueError:

            print("Invalid input.")

            return

        if action > max(ACTIONS.keys()):

            print("Invalid input.")

            return

        ACTIONS[action](order)

    except TransactionServiceException as tsex:

        print tsex.message

    finally:

        close_session()

if __name__ == '__main__':

    main()