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)