def drop_temp_tables(self, alias): data = { 'prev_table': get_prev_agg_tablename(alias), 'prev_local': f"{alias}_prev_local", } with connections[get_icds_ucr_citus_db_alias()].cursor() as cursor: cursor.execute(self.DROP_QUERY.format(**data))
def create_temp_tables(self, table, day): data = { 'prev_table': get_prev_agg_tablename(table), 'prev_local': f"{table}_prev_local", 'prev_month': day, 'current_table': get_table_name(DASHBOARD_DOMAIN, table), 'alias': table } with connections[get_icds_ucr_citus_db_alias()].cursor() as cursor: cursor.execute(self.CREATE_QUERY.format(**data))
def create_temp_tables(self, table, day): alias, table, ucr_alias, id_column_name = table data = { 'prev_table': get_prev_agg_tablename(alias), 'prev_local': f"{alias}_prev_local", 'prev_month': day, 'current_table': table, 'alias': alias, 'ucr_prev_local': f"{ucr_alias}_prev_local", 'id_column': id_column_name } with connections[get_icds_ucr_citus_db_alias()].cursor() as cursor: cursor.execute(self.CREATE_QUERY.format(**data))
def create_temp_tables(self, table, day): next_month_start = day + relativedelta(months=1) six_months_ago = day - relativedelta(months=6) alias, table = table data = { 'prev_table': get_prev_agg_tablename(alias), 'prev_local': f"{alias}_prev_local", 'prev_month': day, 'current_table': table, 'alias': alias, 'six_months_ago': six_months_ago, 'next_month_start': next_month_start } with connections[get_icds_ucr_citus_db_alias()].cursor() as cursor: cursor.execute(self.CREATE_QUERY.format(**data))
def get_table(self, table_id): if not is_current_month( self.month) and ICDS_LOCATION_REASSIGNMENT_AGG.enabled( self.domain): return get_prev_agg_tablename(table_id) return get_table_name(self.domain, table_id)
def prev_tablename(self): return get_prev_agg_tablename(self.helper_key)
def temp_ucr_tablename(self): if ICDS_LOCATION_REASSIGNMENT_AGG.enabled(self.domain): return get_prev_agg_tablename(self.ucr_data_source_id) return self.ucr_tablename