Subversion Repositories SmartDukaan

#!/usr/bin/python

'''
This script loads AWB numebers of a given type for a given
provider into the database.

It expects the filename of a XLS file having just one column
containing AWB numbers. The header row is ignored.

@author: Chandranshu
'''
import optparse
import xlrd
from elixir import *

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

from shop2020.logistics.service.impl import DataService
from shop2020.logistics.service.impl.DataService import Awb, Provider

def load_awb_numbers(filename, provider_id, type, db_hostname):
    DataService.initialize(dbname='logistics', db_hostname=db_hostname)

    provider = Provider.get_by(id=provider_id)
    if provider is None:
        exit()
    
    workbook = xlrd.open_workbook(filename)
    sheet = workbook.sheet_by_index(0)
    num_rows = sheet.nrows
    for rownum in range(1, num_rows):
        awb_number = sheet.row_values(rownum)[0]
        awb = Awb()
        awb.awb_number = int(awb_number)
        awb.is_available = True
        awb.provider = provider
        awb.type = type
    session.commit()

def main():
    parser = optparse.OptionParser()
    parser.add_option("-f", "--file", dest="filename",
                   default="AwbNumbers.xls", type="string",
                   help="Read the AWB numbers from FILE",
                   metavar="FILE")
    parser.add_option("-P", "--provider", dest="provider_id",
                      type="int", help="Load these entries for PROVIDER",
                      metavar="PROVIDER")
    parser.add_option("-t", "--type", dest="type",
                      type="string", help="The TYPE of orders these AWB numbers are for",
                      metavar="TYPE")
    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?")
    filename = options.filename
    provider_id = options.provider_id
    type = options.type
    if filename is None or provider_id is None or type is None:
        parser.error("A provider's name and a filename must be provided. Use -h for usage details.")
    load_awb_numbers(filename, provider_id, type, options.hostname)

if __name__ == '__main__':
    main()