def update(self, cmd_params: CommandParameters) -> Optional[str]: resource_id = cmd_params.resource_id actor = cmd_params.actor key = cmd_params.key value = cmd_params.value dashboard = DashboardDAO.get_by_id_or_slug(str(resource_id)) if dashboard and value: entry: Entry = cache_manager.filter_state_cache.get( cache_key(resource_id, key)) if entry: user_id = actor.get_user_id() if entry["owner"] != user_id: raise TemporaryCacheAccessDeniedError() # Generate a new key if tab_id changes or equals 0 contextual_key = cache_key(session.get("_id"), cmd_params.tab_id, resource_id) key = cache_manager.filter_state_cache.get(contextual_key) if not key or not cmd_params.tab_id: key = random_key() cache_manager.filter_state_cache.set(contextual_key, key) new_entry: Entry = { "owner": actor.get_user_id(), "value": value } cache_manager.filter_state_cache.set( cache_key(resource_id, key), new_entry) return key
def update(self, cmd_params: CommandParameters) -> Optional[str]: resource_id = cmd_params.resource_id key = cmd_params.key value = cast( str, cmd_params.value) # schema ensures that value is not optional check_access(resource_id) entry: Entry = cache_manager.filter_state_cache.get( cache_key(resource_id, key)) owner = get_user_id() if entry: if entry["owner"] != owner: raise TemporaryCacheAccessDeniedError() # Generate a new key if tab_id changes or equals 0 contextual_key = cache_key(session.get("_id"), cmd_params.tab_id, resource_id) key = cache_manager.filter_state_cache.get(contextual_key) if not key or not cmd_params.tab_id: key = random_key() cache_manager.filter_state_cache.set(contextual_key, key) new_entry: Entry = {"owner": owner, "value": value} cache_manager.filter_state_cache.set(cache_key(resource_id, key), new_entry) return key
def create(self, cmd_params: CommandParameters) -> str: resource_id = cmd_params.resource_id tab_id = cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, resource_id) key = cache_manager.filter_state_cache.get(contextual_key) if not key or not tab_id: key = random_key() value = cast(str, cmd_params.value) # schema ensures that value is not optional check_access(resource_id) entry: Entry = {"owner": get_user_id(), "value": value} cache_manager.filter_state_cache.set(cache_key(resource_id, key), entry) cache_manager.filter_state_cache.set(contextual_key, key) return key
def delete(self, cmd_params: CommandParameters) -> bool: resource_id = cmd_params.resource_id key = cache_key(resource_id, cmd_params.key) check_access(resource_id) entry: Entry = cache_manager.filter_state_cache.get(key) if entry: if entry["owner"] != get_user_id(): raise TemporaryCacheAccessDeniedError() tab_id = cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, resource_id) cache_manager.filter_state_cache.delete(contextual_key) return cache_manager.filter_state_cache.delete(key) return False
def run(self) -> str: self.validate() try: datasource_id = self._cmd_params.datasource_id datasource_type = self._cmd_params.datasource_type chart_id = self._cmd_params.chart_id tab_id = self._cmd_params.tab_id form_data = self._cmd_params.form_data check_access(datasource_id, chart_id, datasource_type) contextual_key = cache_key(session.get("_id"), tab_id, datasource_id, chart_id, datasource_type) key = cache_manager.explore_form_data_cache.get(contextual_key) if not key or not tab_id: key = random_key() if form_data: state: TemporaryExploreState = { "owner": get_user_id(), "datasource_id": datasource_id, "datasource_type": DatasourceType(datasource_type), "chart_id": chart_id, "form_data": form_data, } cache_manager.explore_form_data_cache.set(key, state) cache_manager.explore_form_data_cache.set(contextual_key, key) return key except SQLAlchemyError as ex: logger.exception("Error running create command") raise TemporaryCacheCreateFailedError() from ex
def create(self, cmd_params: CommandParameters) -> str: resource_id = cmd_params.resource_id actor = cmd_params.actor tab_id = cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, resource_id) key = cache_manager.filter_state_cache.get(contextual_key) if not key or not tab_id: key = random_key() value = cmd_params.value dashboard = DashboardDAO.get_by_id_or_slug(str(resource_id)) if dashboard and value: entry: Entry = {"owner": actor.get_user_id(), "value": value} cache_manager.filter_state_cache.set(cache_key(resource_id, key), entry) cache_manager.filter_state_cache.set(contextual_key, key) return key
def delete(self, cmd_params: CommandParameters) -> bool: resource_id = cmd_params.resource_id actor = cmd_params.actor key = cache_key(resource_id, cmd_params.key) dashboard = DashboardDAO.get_by_id_or_slug(str(resource_id)) if dashboard: entry: Entry = cache_manager.filter_state_cache.get(key) if entry: if entry["owner"] != actor.get_user_id(): raise TemporaryCacheAccessDeniedError() tab_id = cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, resource_id) cache_manager.filter_state_cache.delete(contextual_key) return cache_manager.filter_state_cache.delete(key) return False
def get(self, cmd_params: CommandParameters) -> Optional[str]: resource_id = cmd_params.resource_id key = cache_key(resource_id, cmd_params.key) DashboardDAO.get_by_id_or_slug(str(resource_id)) entry = cache_manager.filter_state_cache.get(key) or {} if entry and self._refresh_timeout: cache_manager.filter_state_cache.set(key, entry) return entry.get("value")
def get(self, cmd_params: CommandParameters) -> Optional[str]: resource_id = cmd_params.resource_id key = cache_key(resource_id, cmd_params.key) check_access(resource_id) entry = cache_manager.filter_state_cache.get(key) or {} if entry and self._refresh_timeout: cache_manager.filter_state_cache.set(key, entry) return entry.get("value")
def run(self) -> bool: try: actor = self._cmd_params.actor key = self._cmd_params.key state: TemporaryExploreState = cache_manager.explore_form_data_cache.get( key) if state: dataset_id = state["dataset_id"] chart_id = state["chart_id"] check_access(dataset_id, chart_id, actor) if state["owner"] != actor.get_user_id(): raise TemporaryCacheAccessDeniedError() tab_id = self._cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, dataset_id, chart_id) cache_manager.explore_form_data_cache.delete(contextual_key) return cache_manager.explore_form_data_cache.delete(key) return False except SQLAlchemyError as ex: logger.exception("Error running delete command") raise TemporaryCacheDeleteFailedError() from ex
def run(self) -> Optional[str]: self.validate() try: datasource_id = self._cmd_params.datasource_id chart_id = self._cmd_params.chart_id datasource_type = self._cmd_params.datasource_type key = self._cmd_params.key form_data = self._cmd_params.form_data check_access(datasource_id, chart_id, datasource_type) state: TemporaryExploreState = cache_manager.explore_form_data_cache.get( key) owner = get_user_id() if state and form_data: if state["owner"] != owner: raise TemporaryCacheAccessDeniedError() # Generate a new key if tab_id changes or equals 0 tab_id = self._cmd_params.tab_id contextual_key = cache_key(session.get("_id"), tab_id, datasource_id, chart_id, datasource_type) key = cache_manager.explore_form_data_cache.get(contextual_key) if not key or not tab_id: key = random_key() cache_manager.explore_form_data_cache.set( contextual_key, key) new_state: TemporaryExploreState = { "owner": owner, "datasource_id": datasource_id, "datasource_type": DatasourceType(datasource_type), "chart_id": chart_id, "form_data": form_data, } cache_manager.explore_form_data_cache.set(key, new_state) return key except SQLAlchemyError as ex: logger.exception("Error running update command") raise TemporaryCacheUpdateFailedError() from ex
def cache(dashboard_id, admin_id): entry: Entry = {"owner": admin_id, "value": INITIAL_VALUE} cache_manager.filter_state_cache.set(cache_key(dashboard_id, KEY), entry)