def fetch_result_size(self, notebook, snippet): resp = { 'rows': None, 'size': None, 'message': '' } if snippet.get('status') != 'available': raise QueryError(_('Result status is not available')) if snippet['type'] not in ('hive', 'impala'): raise OperationNotSupported(_('Cannot fetch result metadata for snippet type: %s') % snippet['type']) if snippet['type'] == 'hive': resp['rows'], resp['size'], resp['message'] = self._get_hive_result_size(notebook, snippet) else: resp['rows'], resp['size'], resp['message'] = self._get_impala_result_size(notebook, snippet) return resp