1 #!/usr/bin/env python 2 # Copyright 2015 The PDFium Authors. All rights reserved. 3 # Use of this source code is governed by a BSD-style license that can be 4 # found in the LICENSE file. 5 6 import glob 7 import os 8 import subprocess 9 import sys 10 11 def os_name(): 12 if sys.platform.startswith('linux'): 13 return 'linux' 14 if sys.platform.startswith('win'): 15 return 'win' 16 if sys.platform.startswith('darwin'): 17 return 'mac' 18 raise Exception('Confused, can not determine OS, aborting.') 19 20 21 def RunCommand(cmd): 22 try: 23 subprocess.check_call(cmd) 24 return None 25 except subprocess.CalledProcessError as e: 26 return e 27 28 # RunCommandExtractHashedFiles returns a tuple: (raised_exception, hashed_files) 29 # It runs the given command. If it fails it will return an exception and None. 30 # If it succeeds it will return None and the list of processed files extracted 31 # from the output of the command. It expects lines in this format: 32 # MD5:<path_to_image_file>:<md5_hash_in_hex> 33 # The returned hashed_files is a list of (file_path, MD5-hash) pairs. 34 def RunCommandExtractHashedFiles(cmd): 35 try: 36 output = subprocess.check_output(cmd, universal_newlines=True) 37 ret = [] 38 for line in output.split('\n'): 39 line = line.strip() 40 if line.startswith("MD5:"): 41 ret.append([x.strip() for x in line.lstrip("MD5:").rsplit(":", 1)]) 42 return None, ret 43 except subprocess.CalledProcessError as e: 44 return e, None 45 46 47 class DirectoryFinder: 48 '''A class for finding directories and paths under either a standalone 49 checkout or a chromium checkout of PDFium.''' 50 51 def __init__(self, build_location): 52 # |build_location| is typically "out/Debug" or "out/Release". 53 # Expect |my_dir| to be .../pdfium/testing/tools. 54 self.my_dir = os.path.dirname(os.path.realpath(__file__)) 55 self.testing_dir = os.path.dirname(self.my_dir) 56 if (os.path.basename(self.my_dir) != 'tools' or 57 os.path.basename(self.testing_dir) != 'testing'): 58 raise Exception('Confused, can not find pdfium root directory, aborting.') 59 self.pdfium_dir = os.path.dirname(self.testing_dir) 60 # Find path to build directory. This depends on whether this is a 61 # standalone build vs. a build as part of a chromium checkout. For 62 # standalone, we expect a path like .../pdfium/out/Debug, but for 63 # chromium, we expect a path like .../src/out/Debug two levels 64 # higher (to skip over the third_party/pdfium path component under 65 # which chromium sticks pdfium). 66 self.base_dir = self.pdfium_dir 67 one_up_dir = os.path.dirname(self.base_dir) 68 two_up_dir = os.path.dirname(one_up_dir) 69 if (os.path.basename(two_up_dir) == 'src' and 70 os.path.basename(one_up_dir) == 'third_party'): 71 self.base_dir = two_up_dir 72 self.build_dir = os.path.join(self.base_dir, build_location) 73 self.os_name = os_name() 74 75 def ExecutablePath(self, name): 76 '''Finds compiled binaries under the build path.''' 77 result = os.path.join(self.build_dir, name) 78 if self.os_name == 'win': 79 result = result + '.exe' 80 return result 81 82 def ScriptPath(self, name): 83 '''Finds other scripts in the same directory as this one.''' 84 return os.path.join(self.my_dir, name) 85 86 def WorkingDir(self, other_components=''): 87 '''Places generated files under the build directory, not source dir.''' 88 result = os.path.join(self.build_dir, 'gen', 'pdfium') 89 if other_components: 90 result = os.path.join(result, other_components) 91 return result 92 93 def TestingDir(self, other_components=''): 94 '''Finds test files somewhere under the testing directory.''' 95 result = self.testing_dir 96 if other_components: 97 result = os.path.join(result, other_components) 98 return result 99
