Subversion Repositories SmartDukaan

Rev

Rev 1737 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 1737 Rev 4014
Line 11... Line 11...
11
    sys.path.insert(0, os.getcwd())
11
    sys.path.insert(0, os.getcwd())
12
 
12
 
13
from shop2020.logistics.service.impl import DataService
13
from shop2020.logistics.service.impl import DataService
14
from shop2020.logistics.service.impl.DataService import PublicHolidays
14
from shop2020.logistics.service.impl.DataService import PublicHolidays
15
 
15
 
16
def load_holidays(filename, sheet_no):
16
def load_holidays(filename, sheet_no, db_hostname):
17
    DataService.initialize('logistics')
17
    DataService.initialize(dbname='logistics', db_hostname=db_hostname)
18
 
18
 
19
    workbook = xlrd.open_workbook(filename)
19
    workbook = xlrd.open_workbook(filename)
20
    sheet = workbook.sheet_by_index(sheet_no)
20
    sheet = workbook.sheet_by_index(sheet_no)
21
    num_rows = sheet.nrows
21
    num_rows = sheet.nrows
22
 
22
 
Line 39... Line 39...
39
    parser = optparse.OptionParser()
39
    parser = optparse.OptionParser()
40
    parser.add_option("-f", "--file", dest="filename",
40
    parser.add_option("-f", "--file", dest="filename",
41
                   default="Holidays.xls", type="string",
41
                   default="Holidays.xls", type="string",
42
                   help="Read the holiday list",
42
                   help="Read the holiday list",
43
                   metavar="FILE")
43
                   metavar="FILE")
-
 
44
    parser.add_option("-H", "--host", dest="hostname",
-
 
45
                      default="localhost",
-
 
46
                      type="string", help="The HOST where the DB server is running",
-
 
47
                      metavar="HOST")
44
    (options, args) = parser.parse_args()
48
    (options, args) = parser.parse_args()
45
    if len(args) != 0:
49
    if len(args) != 0:
46
        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
50
        parser.error("You've supplied extra arguments. Are you sure you want to run this program?")
47
    filename = options.filename
51
    filename = options.filename
48
    if filename is None:
52
    if filename is None:
49
        parser.error("A filename must be provided. Use -h for more details.")
53
        parser.error("A filename must be provided. Use -h for more details.")
50
    load_holidays(filename, 0)
54
    load_holidays(filename, 0, options.hostname)
51
 
55
 
52
if __name__ == '__main__':
56
if __name__ == '__main__':
53
    main()
57
    main()