85 lines
3.8 KiB
Python
85 lines
3.8 KiB
Python
# Copyright (C) 2011 Google Inc. All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are
|
|
# met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above
|
|
# copyright notice, this list of conditions and the following disclaimer
|
|
# in the documentation and/or other materials provided with the
|
|
# distribution.
|
|
# * Neither the name of Google Inc. nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
"""This module is used to find files used by run-webkit-tests and
|
|
perftestrunner. It exposes one public function - find() - which takes
|
|
an optional list of paths, optional set of skipped directories and optional
|
|
filter callback.
|
|
|
|
If a list is passed in, the returned list of files is constrained to those
|
|
found under the paths passed in. i.e. calling find(["LayoutTests/fast"])
|
|
will only return files under that directory.
|
|
|
|
If a set of skipped directories is passed in, the function will filter out
|
|
the files lying in these directories i.e. find(["LayoutTests"], set(["fast"]))
|
|
will return everything except files in fast subfolder.
|
|
|
|
If a callback is passed in, it will be called for the each file and the file
|
|
will be included into the result if the callback returns True.
|
|
The callback has to take three arguments: filesystem, dirname and filename."""
|
|
|
|
import itertools
|
|
|
|
|
|
def find(filesystem, base_dir, paths=None, skipped_directories=None, file_filter=None, directory_sort_key=None):
|
|
"""Finds the set of tests under a given list of sub-paths.
|
|
|
|
Args:
|
|
paths: a list of path expressions relative to base_dir
|
|
to search. Glob patterns are ok, as are path expressions with
|
|
forward slashes on Windows. If paths is empty, we look at
|
|
everything under the base_dir.
|
|
"""
|
|
|
|
paths = paths or ['*']
|
|
skipped_directories = skipped_directories or set(['.svn', '_svn'])
|
|
return _normalized_find(filesystem, _normalize(filesystem, base_dir, paths), skipped_directories, file_filter, directory_sort_key)
|
|
|
|
|
|
def _normalize(filesystem, base_dir, paths):
|
|
return [filesystem.normpath(filesystem.join(base_dir, path)) for path in paths]
|
|
|
|
|
|
def _normalized_find(filesystem, paths, skipped_directories, file_filter, directory_sort_key):
|
|
"""Finds the set of tests under the list of paths.
|
|
|
|
Args:
|
|
paths: a list of absolute path expressions to search.
|
|
Glob patterns are ok.
|
|
"""
|
|
|
|
paths_to_walk = itertools.chain(*(filesystem.glob(path) for path in paths))
|
|
|
|
def sort_by_directory_key(files_list):
|
|
if directory_sort_key:
|
|
files_list.sort(key=directory_sort_key)
|
|
return files_list
|
|
|
|
all_files = itertools.chain(*(sort_by_directory_key(filesystem.files_under(path, skipped_directories, file_filter)) for path in paths_to_walk))
|
|
return all_files
|