Subversion Repositories SmartDukaan

#!/usr/bin/python 

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

    from_datetime = to_datetime - datetime.timedelta(1)

    orders = Order.query.filter_by(status=OrderStatus.INVENTORY_LOW).filter(Order.outofstock_timestamp >= from_datetime).filter(Order.outofstock_timestamp <= to_datetime).all()

    for order in orders:

        order.status = OrderStatus.SUBMITTED_FOR_PROCESSING

    session.commit()

def main():

    parser = optparse.OptionParser()

    parser.add_option("-d", "--days", dest="days",

                   default=2, type="int",

                   help="move orders marked as out of stock in last NUM days",

                   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_out_of_stock_orders_to_pending_queue(options.days, options.hostname)

if __name__ == '__main__':

    main()