Subversion Repositories SmartDukaan

#!/usr/bin/python

'''

Created on 12-Oct-2011

@author: Vikas

'''

import optparse

import xlrd

import datetime

from shop2020.model.v1.user.impl.UserDataAccessors import add_address_for_user

from shop2020.thriftpy.model.v1.user.ttypes import Address

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

    import sys

    import os

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

from shop2020.clients.UserClient import UserClient

from shop2020.model.v1.user.impl.CartDataAccessors import create_line_item, create_order

from shop2020.model.v1.user.impl import Dataservice

from shop2020.clients.TransactionClient import TransactionClient

from shop2020.clients.PaymentClient import PaymentClient

from shop2020.thriftpy.model.v1.order.ttypes import Transaction, TransactionStatus

from shop2020.thriftpy.payments.ttypes import Attribute, PaymentStatus

from shop2020.utils.Utils import to_java_date

RTGS_GATEWAY_ID = 6

PAYMETHOD_ATTR = 'payMethod'

RTGS_ID_ATTR = 'rtgsId'

RTGS_BANK_ATTR = 'rtgsBank'

RTGS_BRANCH_ATTR = 'rtgsBranch'

IFSC_ATTR = 'ifscCode'

ITEM_ID_KEY = 'itemId'

ITEM_DESC_KEY = 'itemDesc'

PRICE_KEY = 'price'

QUANTITY_KEY = 'quantity'

AMOUNT_KEY = 'amount'

ADDRESS_KEY = 'address'

user_client = UserClient().get_client()

transaction_client = TransactionClient().get_client()

payment_client = PaymentClient().get_client()

def create_orders(user, items):

    orders = []

    for item in items:

        line_item = create_line_item(int(item[ITEM_ID_KEY]), item[PRICE_KEY], item[QUANTITY_KEY])

        addressId = add_address_for_user(item[ADDRESS_KEY], user.userId, False)

        orders.append(create_order(user.userId, addressId, line_item, 0, 0))

    return orders

def create_transaction(user, items):

    txn = Transaction()

    txn.shoppingCartid = user.activeCartId

    txn.customer_id = user.userId

    txn.createdOn = to_java_date(datetime.datetime.now())

    txn.transactionStatus = TransactionStatus.INIT

    txn.statusDescription = "New Order"

    txn.orders = create_orders(user, items)

    transaction_client = TransactionClient().get_client()

    txn_id = transaction_client.createTransaction(txn)

    return txn_id

def create_payment(user, amount , txn_id, rtgs_id, rtgs_bank, rtgs_branch, ifsc):

    payment_id = payment_client.createPayment(user.userId, amount, RTGS_GATEWAY_ID, txn_id)

    payment_client.getPayment(payment_id)

    attributes = []

    attributes.append(Attribute(PAYMETHOD_ATTR, '4000'))

    attributes.append(Attribute(RTGS_ID_ATTR, str(rtgs_id)))

    attributes.append(Attribute(RTGS_BANK_ATTR, rtgs_bank))

    attributes.append(Attribute(RTGS_BRANCH_ATTR, rtgs_branch))

    attributes.append(Attribute(IFSC_ATTR, str(ifsc)))

    payment_client.updatePaymentDetails(payment_id, None, None, None, None, None, None, None, None, PaymentStatus.SUCCESS, None, attributes)

def load_orders(file):

    workbook = xlrd.open_workbook(file)

    sheet = workbook.sheet_by_index(0)

    num_rows = sheet.nrows

    email, rtgs_id, rtgs_bank, rtgs_branch, ifsc, amount = sheet.col_values(1)[0:6]

    user = user_client.getUserByEmail(email)

    if user.userId == -1:

        print "The email is not registered with us : " + email

        return

    items = []

    total_amount = 0

    for rownum in range(10, num_rows):

        item = {}

        item[ITEM_ID_KEY], item[ITEM_DESC_KEY], item[PRICE_KEY], item[QUANTITY_KEY], item[AMOUNT_KEY] = sheet.row_values(rownum)[0:5]

        name, line1, city, state, pin, phone  = sheet.row_values(rownum)[5:11]

        total_amount += item[AMOUNT_KEY]

        items.append(item)

        address = Address()

        address.name = name

        address.line1 = line1

        address.city = city

        address.state = state

        address.pin = str(pin)

        address.phone = str(phone)

        item[ADDRESS_KEY] = address

    if amount != total_amount:

        print "Amount paid is not equal to the total amount of bulk order."

        return

    txn_id = create_transaction(user, items)

    create_payment(user, amount, txn_id, rtgs_id, rtgs_bank, rtgs_branch, ifsc)

    transaction_client.changeTransactionStatus(txn_id, TransactionStatus.AUTHORIZED, "Payment received for the order");

    transaction_client.changeTransactionStatus(txn_id, TransactionStatus.IN_PROCESS, "RTGS Payment accepted");

def main():

    parser = optparse.OptionParser()

    parser.add_option("-f", "--file", dest="file",

                   help="Excel file with bulk orders.")

    parser.add_option("-d", "--db_server", dest="host",

                      default="localhost",

                   help="Database server.")

    (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.file == None:

        parser.error("Excel File with bulk orders not supplied.")

    Dataservice.initialize(db_hostname=options.host)

    load_orders(options.file)

    #load_orders('/home/vikas/RTGS.xls')

if __name__ == '__main__':

    main()