def handle_url_query(self, req: Request) -> None: query_data = req._get_query() logging.info('query_data: {}'.format(query_data)) for key, val in query_data: # don't bother trying to interpret an argument less than 4 characters as some type of timestamp if len(val) > 4: if self.safely_parse(val): self.query_keys.add(key)
def clean_url_query(self, req: Request) -> None: """Replace any problematic values of query parameters with constant data Args: req (Request): The request to modify """ query_data = sorted(req._get_query()) logging.info('fetched query_data: {}'.format(query_data)) updated_query_data = [] if query_data and self.query_keys: for key, val in query_data: if key in self.query_keys: updated_query_data.append((key, self.constant)) else: updated_query_data.append((key, val)) req._set_query(updated_query_data or query_data) logging.info(f'updated query_data: {req._get_query()}')