def __init__(
self,
profile,
firefox,
geckodriver,
archive,
device_name,
strict,
force_new,
visible,
):
self.force_new = force_new
self.profile = profile
self.geckodriver = geckodriver
self.archive = archive
self.device_name = device_name
self.strict = strict
self.visible = visible
# unpacking a dmg
# XXX do something similar if we get an apk (but later)
# XXX we want to do
# adb install -r target.apk
# and get the installed app name
if firefox is not None and firefox.endswith("dmg"):
target = os.path.join(os.path.dirname(firefox), "firefox.app")
extract_from_dmg(firefox, target)
firefox = os.path.join(target, "Contents", "MacOS", "firefox")
self.firefox = firefox
self.android = self.firefox is not None and self.firefox.startswith(
"org.mozilla")
def main(args=sys.argv[1:]):
parser = argparse.ArgumentParser(description="Profile Creator")
parser.add_argument("archive", help="Archives Dir", type=str, default=None)
parser.add_argument("--firefox",
help="Firefox Binary",
type=str,
default=None)
parser.add_argument("--scenario",
help="Scenario to use",
type=str,
default="all")
parser.add_argument("--profile",
help="Existing profile Dir",
type=str,
default=None)
parser.add_argument("--customization",
help="Profile customization to use",
type=str,
default="all")
parser.add_argument(
"--fresh-profile",
help="Create a fresh profile",
action="store_true",
default=False,
)
parser.add_argument("--visible",
help="Don't use headless mode",
action="store_true",
default=False)
parser.add_argument("--archives-dir",
help="Archives local dir",
type=str,
default="/tmp/archives")
parser.add_argument("--force-new",
help="Create from scratch",
action="store_true",
default=False)
parser.add_argument(
"--strict",
help="Errors out immediatly on a scenario failure",
action="store_true",
default=True,
)
parser.add_argument(
"--geckodriver",
help="Path to the geckodriver binary",
type=str,
default=sys.platform.startswith("win") and "geckodriver.exe"
or "geckodriver",
)
parser.add_argument("--device-name",
help="Name of the device",
type=str,
default=None)
args = parser.parse_args(args=args)
# unpacking a dmg
# XXX do something similar if we get an apk (but later)
# XXX we want to do
# adb install -r target.apk
# and get the installed app name
if args.firefox is not None and args.firefox.endswith("dmg"):
target = os.path.join(os.path.dirname(args.firefox), "firefox.app")
extract_from_dmg(args.firefox, target)
args.firefox = os.path.join(target, "Contents", "MacOS", "firefox")
args.android = args.firefox is not None and args.firefox.startswith(
"org.mozilla")
if not args.android and args.firefox is not None:
LOG("Verifying Desktop Firefox binary")
# we want to verify we do have a firefox binary
# XXX so lame
if not os.path.exists(args.firefox):
if "MOZ_FETCHES_DIR" in os.environ:
target = os.path.join(os.environ["MOZ_FETCHES_DIR"],
args.firefox)
if os.path.exists(target):
args.firefox = target
if not os.path.exists(args.firefox):
raise IOError("Cannot find %s" % args.firefox)
version = get_version(args.firefox)
LOG("Working with Firefox %s" % version)
LOG(os.environ)
args.archive = os.path.abspath(args.archive)
LOG("Archives directory is %s" % args.archive)
if not os.path.exists(args.archive):
os.makedirs(args.archive, exist_ok=True)
LOG("Verifying Geckodriver binary presence")
if shutil.which(args.geckodriver) is None and not os.path.exists(
args.geckodriver):
raise IOError("Cannot find %s" % args.geckodriver)
try:
plat = args.android and "android" or get_current_platform()
changelog = read_changelog(plat)
LOG("Got the changelog from TaskCluster")
except ProfileNotFoundError:
LOG("changelog not found on TaskCluster, creating a local one.")
changelog = Changelog(args.archive)
loop = asyncio.get_event_loop()
async def one_run(scenario, customization):
if args.android:
env = AndroidEnv(
args.profile,
args.firefox,
args.geckodriver,
args.archive,
args.device_name,
)
else:
env = DesktopEnv(
args.profile,
args.firefox,
args.geckodriver,
args.archive,
args.device_name,
)
return await ProfileCreator(scenario, customization, args.archive,
changelog, args.force_new,
env).run(not args.visible)
async def run_all(args):
if args.scenario != "all":
return await one_run(args.scenario, args.customization)
# this is the loop that generates all combinations of profile
# for the current platform when "all" is selected
res = []
for scenario in scenarii.keys():
if args.customization != "all":
try:
res.append(await one_run(scenario, args.customization))
except Exception:
ERROR("Something went wrong on this one.")
if args.strict:
raise
else:
for customization in get_customizations():
try:
res.append(await one_run(scenario, customization))
except Exception:
ERROR("Something went wrong on this one.")
if args.strict:
raise
return res
try:
loop.run_until_complete(run_all(args))
LOG("Saving changelog in %s" % args.archive)
changelog.save(args.archive)
finally:
loop.close()