main.py

# -*- coding: utf-8  -*-

import gzip
import os
from os import path

from tqdm import tqdm

import credentials
from pwbdb import gerrit, github, travis


def update_master():
    local_master = gerrit.fetch_master()
    github.push_branch(local_master)


def create_github_pr_from_gerrit():
    update_master()
    github.refresh_pull_requests()
    for change in tqdm(gerrit.get_changes()):
        local_branch = gerrit.fetch_branch(change)
        github_ref = github.push_branch(local_branch)
        pr = github.open_pull_request(github_ref, creds=credentials)
        gerrit.post_comment_about_CI(change, pr, creds=credentials)


def test_downloaded_travis_builds(test_func):
    logs_dir = path.join(path.dirname(__file__), 'travis-logs')
    for b in os.listdir(logs_dir):
        bdir = path.join(logs_dir, b)
        for j in os.listdir(bdir):
            jfile = path.join(bdir, j)
            with gzip.open(jfile) as f:
                if test_func(str(f.read())):
                    yield b, j.split('.')[0]


def show_github_branches_without_gerrit():
    changes = gerrit.get_changes()
    branches = github.get_branches(creds=credentials)
    branches_with_gerrit = {'gerrit-{}-{}'.format(
        change['_number'], gerrit.newest_revision_number(change)
    ) for change in changes}
    branches_without_gerrit = {b.name for b in branches
                               if b.name not in ('master', '2.0') and
                               b.name not in branches_with_gerrit}
    for b in branches_without_gerrit:
        print(b)


def get_travis_logs():
    travis.get_travis_logs()


if __name__ == '__main__':
    while True:
        try:
            create_github_pr_from_gerrit()
            show_github_branches_without_gerrit()
        except:
            pass