Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
If a order has been in the payment pending state for more
than 2 days, it marks it as failed.

Created on 09-Aug-2011

@author: chandranshu
'''

import datetime
import optparse
import sys

if __name__ == '__main__' and __package__ is None:
    import os
    sys.path.insert(0, os.getcwd())

from shop2020.thriftpy.model.v1.order.ttypes import OrderStatus
from shop2020.model.v1.order.impl import DataService
from shop2020.model.v1.order.impl.DataService import Order
from elixir import session

def move_payment_pending_orders_to_failed_state(days, db_hostname):
    DataService.initialize(db_hostname=db_hostname, echoOn=True)
    current_time = datetime.datetime.now()
    to_datetime = datetime.datetime(current_time.year, current_time.month, current_time.day)
    to_datetime = to_datetime - datetime.timedelta(days)
    orders = Order.query.filter_by(status=OrderStatus.PAYMENT_PENDING).filter(Order.created_timestamp <= to_datetime).all()
    for order in orders:
        order.status = OrderStatus.PAYMENT_FAILED
    session.commit()

def main():
    parser = optparse.OptionParser()
    parser.add_option("-d", "--days", dest="days",
                   default=2, type="int",
                   help="move orders in the PAYMENT_PENDING state for more than NUM days to PAYMENT_FAILED state",
                   metavar="NUM")
    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?")
    move_payment_pending_orders_to_failed_state(options.days, options.hostname)

if __name__ == '__main__':
    main()