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
