blob: 09ecc9decfd3e37bfdf9052a61f72ceb3adf2872 [file] [log] [blame]
#!/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'):
import webkitpy.thirdparty.autoinstalled.chromedriver
driver_init_file = webkitpy.thirdparty.autoinstalled.chromedriver.__file__
driver_executable = os.path.join(os.path.dirname(os.path.realpath(driver_init_file)), 'chromedriver')
return driver_executable
elif browser_name.startswith('firefox'):
import webkitpy.thirdparty.autoinstalled.geckodriver
driver_init_file = webkitpy.thirdparty.autoinstalled.geckodriver.__file__
driver_executable = os.path.join(os.path.dirname(os.path.realpath(driver_init_file)), 'geckodriver')
return driver_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