Subversion Repositories SmartDukaan

Rev

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

Rev 1504 Rev 1730
Line 4... Line 4...
4 
@author: rajveer
 4 
@author: rajveer
5 
'''
 5 
'''
6 
 
 6 
 
7 
from shop2020.logistics.service.impl import DataService
 7 
from shop2020.logistics.service.impl import DataService
8 
from shop2020.logistics.service.impl.DataService import Awb, AwbUpdate, Provider, \
 8 
from shop2020.logistics.service.impl.DataService import Awb, AwbUpdate, Provider, \
9 
     DeliveryEstimate, WarehouseAllocation, DestinationProviderAllocation
 9 
     DeliveryEstimate, WarehouseAllocation, DestinationProviderAllocation,\
- 
 
 10 
    PublicHolidays
10 
from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException
 11 
from shop2020.thriftpy.logistics.ttypes import LogisticsServiceException
11 
from shop2020.utils.Utils import log_entry, to_py_date, to_java_date
 12 
from shop2020.utils.Utils import log_entry, to_py_date, to_java_date
12 
from elixir import session
 13 
from elixir import session
13 
import datetime
 14 
import datetime, time
14 
import sys
 15 
import sys
15 
 
 16 
 
16 
def initialize(dbname="logistics"):
 17 
def initialize(dbname="logistics"):
17 
    log_entry("initialize@DataAccessor", "Initializing data service")
 18 
    log_entry("initialize@DataAccessor", "Initializing data service")
18 
    DataService.initialize(dbname)
 19 
    DataService.initialize(dbname)
Line 93... Line 94...
93 
    awb = Awb.get_by(awb_number = awb_number, provider_id = provider_id)
 94 
    awb = Awb.get_by(awb_number = awb_number, provider_id = provider_id)
94 
    query = AwbUpdate.query.filter_by(awb = awb)
 95 
    query = AwbUpdate.query.filter_by(awb = awb)
95 
    info = query.all()
 96 
    info = query.all()
96 
    return info
 97 
    return info
97 
 
 98 
 
- 
 
 99 
def get_holidays(start_date, end_date):
- 
 
 100 
    query = PublicHolidays.query
- 
 
 101 
    if start_date != -1:
- 
 
 102 
        query = query.filter(PublicHolidays.date >= to_py_date(start_date))
- 
 
 103 
    if end_date != -1:
- 
 
 104 
        query = query.filter(PublicHolidays.date <= to_py_date(end_date))
- 
 
 105 
    holidays = query.all()
- 
 
 106 
    holiday_dates = [to_java_date(datetime.datetime(*time.strptime(str(holiday.date), '%Y-%m-%d')[:3])) for holiday in holidays]
- 
 
 107 
    return holiday_dates
- 
 
 108 
 
98 
def close_session():
 109 
def close_session():
99 
    if session.is_active:
 110 
    if session.is_active:
100 
        print "session is active. closing it."
 111 
        print "session is active. closing it."
101 
        session.close()
112 
        session.close()