forked from mirrors/gecko-dev
60 lines
2 KiB
Python
60 lines
2 KiB
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
import pathlib
|
|
import re
|
|
import subprocess
|
|
from datetime import datetime, timedelta
|
|
|
|
import requests
|
|
from mozperftest.system.android_startup import (
|
|
BASE_URL_DICT,
|
|
DATETIME_FORMAT,
|
|
KEY_ARCHITECTURE,
|
|
KEY_COMMIT,
|
|
KEY_DATETIME,
|
|
KEY_NAME,
|
|
KEY_PRODUCT,
|
|
)
|
|
|
|
HTTP_200_OKAY = 200
|
|
|
|
|
|
def before_iterations(kw):
|
|
product = kw["AndroidStartUp_product"]
|
|
architecture = "arm64-v8a"
|
|
if product == "geckoview_example":
|
|
architecture = "aarch64"
|
|
commit_info = subprocess.getoutput("hg log -l 1")
|
|
commit_date = re.search(r"date:\s+([:\s\w]+)\s+", str(commit_info)).group(1)
|
|
download_date = (
|
|
datetime.strptime(commit_date, "%a %b %d %H:%M:%S %Y") - timedelta(days=1)
|
|
).strftime(DATETIME_FORMAT)
|
|
|
|
nightly_url = BASE_URL_DICT[product].format(
|
|
date=download_date, architecture=architecture
|
|
)
|
|
filename = f"{product}_nightly_{architecture}.apk"
|
|
print("Fetching {}...".format(filename), end="", flush=True)
|
|
download_apk_as_date(nightly_url, download_date, filename)
|
|
print(f"Downloaded {product} for date: {download_date}")
|
|
|
|
kw["apk_metadata"] = {
|
|
KEY_NAME: filename,
|
|
KEY_DATETIME: download_date,
|
|
KEY_COMMIT: "",
|
|
KEY_ARCHITECTURE: architecture,
|
|
KEY_PRODUCT: product,
|
|
}
|
|
|
|
|
|
def download_apk_as_date(nightly_url, download_date_string, filename):
|
|
apk = requests.get(nightly_url)
|
|
if apk.status_code != HTTP_200_OKAY:
|
|
raise Exception(
|
|
f"Something went wrong downloading the apk check to make sure you have entered"
|
|
f" a date that is valid and that the apk for the date you have requested("
|
|
f"{download_date_string}) is available and that the URL({nightly_url}) is also "
|
|
f"valid"
|
|
)
|
|
pathlib.Path(filename).write_bytes(apk.content)
|