def get_department_info(self):
"""
Get imformation about all department in a hotel
Returns:
pd.DataFrame: dataframe contain department infomation.
None: if cannot get department info data
"""
try:
json_data = self.db.get_dept_ids_and_configuration(self.client_id, self.hotel_id)
dep_info = pd.DataFrame(json_data)
except Exception, e:
debug_mess = '%s:%s %s:%s %s' % (col_name.CLIENT, self.client_id, col_name.H_ID, self.hotel_id, e.message)
log.debug(debug_mess)
dep_info = None
def set_hotel_data(self):
"""
If hotel_data set up hotel data as dataframe that index is date, columns is values by
department, segment, type of values
Ex: | Date | dep1_seg1_rv |...|dep1_seg2_rn|...|dep2_seg3_gn|
"""
if self.hotel_data is None:
data_in_json = self.db.get_hotel_data_in_a_specific_time_period(
self.client_id, self.hotel_id, self.from_date, self.to_date)
if len(data_in_json) > 0:
# raw_df = pd.DataFrame(data_in_json)
raw_df = convert_new_revenue(data_in_json)
pivoted_df = pivot_raw_data(raw_df)
pivoted_df.index = pd.to_datetime(pivoted_df.index)
pivoted_df = pivoted_df.astype(float)
self.hotel_data = pivoted_df
else:
debug_mess = '%s:%s %s:%s %s' % (col_name.CLIENT,
self.client_id, col_name.H_ID,
self.hotel_id, 'Data_empty')
log.debug(debug_mess)
