import sys

from mozci.query_jobs import TreeherderApi
from mozci.errors import TreeherderError
from argparse import ArgumentParser


SUCCESS, WARNING, FAILURE, SKIPPED, EXCEPTION, RETRY, CANCELLED = range(7)


def parse_args(argv=None):
    parser = ArgumentParser()

    parser.add_argument("-r", "--revision",
                        dest="rev",
                        type=str,
                        help="It's the revision we need to query job")
    parser.add_argument("-repo", "--repo_name",
                        dest="repo",
                        type=str,
                        help="It's the repo name of revision")
    options = parser.parse_args(argv)
    return options


def main():
    options = parse_args()
    repo_name = options.repo
    revision = options.rev
    success_jobs = []
    failed_jobs = []
    query_api = TreeherderApi()
    jobs = query_api.get_all_jobs(repo_name, revision)
    for job in jobs:
        try:
            job_status = query_api.get_job_status(job)
        except TreeherderError:
            continue
        if job_status == SUCCESS:
            success_jobs.append(job)
        elif job_status == FAILURE:
            failed_jobs.append(job)
    print success_jobs
    print "+++++++++++++++++++++++++++++++++++++++++++++"
    print failed_jobs


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        sys.tracebacklimit = 0
        sys.exit(0)