Subversion Repositories SmartDukaan

#!/usr/bin/python

import optparse

import time

import datetime

import sys

from elixir import session

from shop2020.clients.LogisticsClient import LogisticsClient

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

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

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

def change_logistics_provider(order):

    '''

    Ship the order through different vendor.

    '''

    if order.status not in [OrderStatus.COD_VERIFICATION_PENDING, OrderStatus.SUBMITTED_FOR_PROCESSING, OrderStatus.INVENTORY_LOW, OrderStatus.ACCEPTED]:

        print "This order has already been processed. Please seek help from engineering."

        return

    raw_provider_id = raw_input("Enter the ID of the provider that you want to ship through: ")

    try:

        provider_id = int(raw_provider_id)

    except ValueError:

        print("Invalid provider Id")

        return

    if order.logistics_provider_id == provider_id:

        print("Provider Id entered by you is same as provider assigned to order.")

        return

    logistics_client = LogisticsClient().get_client()

    awb_number = logistics_client.getEmptyAWB(provider_id, order.cod)

    order.logistics_provider_id = provider_id

    order.airwaybill_no = awb_number

    order.track_id = awb_number

    session.commit()

    print("Successfully updated the provider id")

def cancel(order):

    '''

    Cancel

    '''

    print("Your session has been closed")

    return

ACTIONS = {0: cancel,

           1: change_logistics_provider

           }

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

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

    raw_order_id = raw_input("Enter Order Id which you want to modify:")

    try:

        order_id = int(raw_order_id)

        print("You want to modify: " + str(order_id))

    except ValueError:

        print("Invalid Order Id.")

        return

    try:

        order = get_order(order_id)

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