| #!/usr/bin/env python |
| |
| import imp |
| import inspect |
| import logging |
| import os |
| import shutil |
| |
| from webkitpy.common.memoized import memoized |
| |
| |
| _log = logging.getLogger(__name__) |
| |
| |
| # Borrow following code from stackoverflow |
| # Link: http://stackoverflow.com/questions/11461356/issubclass-returns-flase-on-the-same-class-imported-from-different-paths |
| def is_subclass(child, parent_name): |
| return inspect.isclass(child) and parent_name in [cls.__name__ for cls in inspect.getmro(child)] |
| |
| |
| def load_subclasses(dirname, base_class_name, base_class_file, loader): |
| filelist = [base_class_file] + [f for f in os.listdir(dirname) if f.endswith('_' + base_class_file)] |
| filelist += [f for f in os.listdir(dirname) if f.endswith('.py') and f not in ['__init__.py'] + filelist] |
| for filename in filelist: |
| module_name = os.path.splitext(filename)[0] |
| module = imp.load_source(module_name, os.path.join(dirname, filename)) |
| for item_name in dir(module): |
| item = getattr(module, item_name) |
| if is_subclass(item, base_class_name): |
| loader(item) |
| |
| |
| def get_path_from_project_root(relative_path_to_project_root): |
| # Choose the directory containing current file as start point, |
| # compute relative path base on the parameter, |
| # and return an absolute path |
| return os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), relative_path_to_project_root)) |
| |
| |
| def force_remove(path): |
| try: |
| if os.path.isdir(path): |
| shutil.rmtree(path) |
| else: |
| os.remove(path) |
| except Exception as error: |
| # Directory/file does not exist or privilege issue, just ignore it |
| _log.info("Error removing %s: %s" % (path, error)) |
| pass |
| |
| |
| @memoized |
| def get_driver_binary_path(browser_name): |
| if browser_name.startswith('chrome'): |
| from webkitpy.autoinstalled import chromedriver |
| return chromedriver.executable |
| elif browser_name.startswith('firefox'): |
| from webkitpy.autoinstalled import geckodriver |
| return geckodriver.executable |
| |
| |
| def write_defaults(domain, key, value): |
| # Returns whether the key in the domain is updated |
| from Foundation import NSUserDefaults |
| defaults = NSUserDefaults.standardUserDefaults() |
| defaults_for_domain = defaults.persistentDomainForName_(domain) |
| if not defaults_for_domain: |
| return False |
| old_value = defaults_for_domain.get(key) |
| if old_value == value: |
| return False |
| mutable_defaults_for_domain = defaults_for_domain.mutableCopy() |
| mutable_defaults_for_domain[key] = value |
| defaults.setPersistentDomain_forName_(mutable_defaults_for_domain, domain) |
| defaults.synchronize() |
| return True |