blob: 675b97603e4b314d5fce01dc7ccf085f6c0ce377 [file] [log] [blame]
#!/usr/bin/env python
import os
from os import path
import logging
import site
import sys
site.addsitedir(path.join(path.dirname(__file__), "../../lib/python"))
# Use explicit version of python-requests
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../../lib/python/vendor/requests-2.7.0"))
from balrog.submitter.api import Rule, Release
if __name__ == '__main__':
from argparse import ArgumentParser, REMAINDER
parser = ArgumentParser()
parser.add_argument("-a", "--api-root", dest="api_root", required=True,
help="The root of the Balrog API. Eg:")
parser.add_argument("-c", "--credentials-file", dest="credentials_file", required=True,
help="""The file containing the credentials to use when authenticating to Balrog.
It must containing a dict called "balrog_credentials" which must have a key
matching the --username passed.""")
parser.add_argument("-u", "--username", dest="username", required=True,
help="The username to use when authenticating to Balrog.")
parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", default=False)
parser.add_argument("-r", "--rule-id", dest="rule_ids", action="append", required=True,
help="""A rule id to perform an action on. This must be given at least once, but
can be passed multiple times to update multiple rules at the same time.""")
parser.add_argument("-n", "--dry-run", dest="dry_run", action="store_true", default=False,
help="""When set, the script will print out what would have been done. No rules
will be updated.""")
parser.add_argument("action", nargs=1, choices=["lock", "unlock"],
help="""The action to perform on the given rules.
When "lock" is requested, this script will change the mapping for the given
rules to match what they are currently pointing at. If no other arguments
are given, this is accomplished by looking at the current mapping for each
rule, finding the most recent buildid referenced in it, and constructing
that release's name based on the current mapping. If you want to lock rules
to a specific release, you may pass that as the next argument.
When "unlock" is requested, this script will change the mapping for the
given rules to point at their "latest" blob. This is accomplished by
stripping the current mapping of its last section, and appending "-latest".
For example: Firefox-mozilla-central-nightly-201410120103 becomes
parser.add_argument("action_args", nargs=REMAINDER,
help="""See help for "action". Only valid when "action" is "lock".""")
args = parser.parse_args()
logging_level = logging.INFO
if args.verbose:
logging_level = logging.DEBUG
logging.basicConfig(stream=sys.stdout, level=logging_level,
credentials = {}
execfile(args.credentials_file, credentials)
auth = (args.username, credentials['balrog_credentials'][args.username])
if args.action == ["lock"]:
my_args = args.action_args
if len(my_args) > 1:
parser.error("Unsupported number of args for the lock command.")
for rule_id in args.rule_ids:
release_name = None
if len(my_args) == 1:
release_name = args[0]
if not release_name:
rule_data, _ = Rule(api_root=args.api_root, auth=auth,
latest_blob_name = rule_data["mapping"]
release = Release(api_root=args.api_root, auth=auth,
release_data, _ = release.get_data()
buildid = None
for p in release_data["platforms"].values():
enUS = p.get("locales", {}).get("en-US", {})
next_buildid = enUS.get("buildID")
if next_buildid > buildid:
buildid = next_buildid
if not buildid:
logging.error("Couldn't find latest buildid for rule %s", rule_id)
root_name = "-".join(latest_blob_name.split("-")[:-1])
release_name = "%s-%s" % (root_name, buildid)
if not args.dry_run:"Locking rule %s to %s", rule_id, release_name)
Rule(api_root=args.api_root, auth=auth, rule_id=rule_id
else:"Would've locked rule %s to %s", rule_id, release_name)
elif args.action == ["unlock"]:
if args.action_args:
parser.error("Unlock command does not accept any args.")
for rule_id in args.rule_ids:
rule = Rule(api_root=args.api_root, auth=auth, rule_id=rule_id)
rule_data, _ = rule.get_data()
root_name = "-".join(rule_data["mapping"].split("-")[:-1])
release_name = "%s-latest" % root_name
if not args.dry_run:"Unlocking rule %s back to %s", rule_id, release_name)
else:"Would've unlocked rule %s back to %s", rule_id, release_name)