blob: 80379df835515f31afabbff86b7b27b2a53d2284 [file] [log] [blame]
# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import glob
import os
from telemetry.core import util
import py_utils as catapult_util
# TODO(aiolos): Move these functions to catapult_util or here.
GetBaseDir = util.GetBaseDir
GetTelemetryDir = util.GetTelemetryDir
GetUnittestDataDir = util.GetUnittestDataDir
GetChromiumSrcDir = util.GetChromiumSrcDir
GetBuildDirectories = util.GetBuildDirectories
IsExecutable = catapult_util.IsExecutable
def _HasWildcardCharacters(input_string):
# Could make this more precise.
return '*' in input_string or '+' in input_string
def FindInstalledWindowsApplication(application_path):
"""Search common Windows installation directories for an application.
application_path: Path to application relative from installation location.
A string representing the full path, or None if not found.
search_paths = [os.getenv('PROGRAMFILES(X86)'),
search_paths += os.getenv('PATH', '').split(os.pathsep)
for search_path in search_paths:
if not search_path:
path = os.path.join(search_path, application_path)
if _HasWildcardCharacters(path):
paths = glob.glob(path)
paths = [path]
for p in paths:
if IsExecutable(p):
return p
return None
def IsSubpath(subpath, superpath):
"""Returns True iff subpath is or is in superpath."""
subpath = os.path.realpath(subpath)
superpath = os.path.realpath(superpath)
while len(subpath) >= len(superpath):
if subpath == superpath:
return True
subpath = os.path.split(subpath)[0]
return False
def ListFiles(base_directory, should_include_dir=lambda _: True,
should_include_file=lambda _: True):
matching_files = []
for root, dirs, files in os.walk(base_directory):
dirs[:] = [dir_name for dir_name in dirs if should_include_dir(dir_name)]
matching_files += [os.path.join(root, file_name)
for file_name in files if should_include_file(file_name)]
return sorted(matching_files)