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
