Files
impala/lib/python/impala_py_lib/helpers.py
Joe McDonnell 2357958e73 IMPALA-10304: Fix log level and format for pytests
Recent testing showed that the pytests are not
respecting the log level and format set in
conftest.py's configure_logging(). It is using
the default log level of WARNING and the
default formatter.

The issue is that logging.basicConfig() is only
effective the first time it is called. The code
in lib/python/impala_py_lib/helpers.py does a
call to logging.basicConfig() at the global
level, and conftest.py imports that file. This
renders the call in configure_logging()
ineffective.

To avoid this type of confusion, logging.basicConfig()
should only be called from the main() functions for
libraries. This removes the call in lib/python/impala_py_lib
(as it is not needed for a library without a main function).
It also fixes up various other locations to move the
logging.basicConfig() call to the main() function.

Testing:
 - Ran the end to end tests and custom cluster tests
 - Confirmed the logging format
 - Added an assert in configure_logging() to test that
   the INFO log level is applied to the root logger.

Change-Id: I5d91b7f910b3606c50bcba4579179a0bc8c20588
Reviewed-on: http://gerrit.cloudera.org:8080/16679
Reviewed-by: Tim Armstrong <tarmstrong@cloudera.com>
Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
2020-10-30 15:32:21 +00:00

78 lines
2.3 KiB
Python

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
import fnmatch
import logging
import os
import re
import subprocess
LOG = logging.getLogger('impala_lib_python_helpers')
def exec_local_command(cmd):
"""
Executes a command for the local bash shell and return stdout as a string.
Raise CalledProcessError in case of non-zero return code.
Args:
cmd: command as a string
Return:
STDOUT
"""
proc = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
output, error = proc.communicate()
retcode = proc.poll()
if retcode:
LOG.error("{0} returned status {1}: {2}".format(cmd, retcode, error))
raise subprocess.CalledProcessError()
else:
return output
def find_all_files(fname_pattern, base_dir=os.getenv('IMPALA_HOME', '.')):
"""
General utility to recursively find files matching a certain unix-like file pattern.
Args:
fname_pattern: Unix glob
base_dir: the root directory where searching should start
Returns:
A list of full paths relative to the give base_dir
"""
file_glob = fnmatch.translate(fname_pattern)
matching_files = []
for root, dirs, files in os.walk(base_dir):
matching_files += [os.path.join(root, f) for f in files if re.match(file_glob, f)]
return matching_files
def is_core_dump(file_path):
"""
Determine whether given file is a core file. Works on CentOS and Ubuntu.
Args:
file_path: full path to a possible core file
"""
file_std_out = exec_local_command("file %s" % file_path)
return "core file" in file_std_out and 'ELF' in file_std_out