prune_daily_builds.py

"""
 @file
 @brief Remove older daily builds from GitHub (so our daily repo doesn't become huge)
 @author Jonathan Thomas <jonathan@openshot.org>

 @section LICENSE

 Copyright (c) 2008-2016 OpenShot Studios, LLC
 (http://www.openshotstudios.com). This file is part of
 OpenShot Video Editor (http://www.openshot.org), an open-source project
 dedicated to delivering high quality video editing and animation solutions
 to the world.

 OpenShot Video Editor is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 OpenShot Video Editor is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with OpenShot Library.  If not, see <http://www.gnu.org/licenses/>.
 """

import sys
from github3 import login
import datetime
import pytz
import time


# Should we actually delete old assets?
SHOULD_DELETE = True
DELETE_DELAY = 0.5
MAXIMUM_ASSET_AGE_DAYS = 180

# Calculate current date (with timezone)
utc=pytz.UTC
now = utc.localize(datetime.datetime.now())


def get_release(repo, tag_name):
    """Fetch the GitHub release tagged with the given tag and return it
    @param repo:        github3 repository object
    @returns:           github3 release object or None
    """
    for release in repo.iter_releases():
        if release.tag_name == tag_name:
            return release


if len(sys.argv) >= 3:
    github_user = sys.argv[1]
    github_pass = sys.argv[2]

    # Login and get "GitHub" object
    gh = login(github_user, github_pass)
    repo = gh.repository("OpenShot", "openshot-qt")

    # Get daily git_release object
    github_release = get_release(repo, "daily")

    # Loop through all assets (for daily release)
    delete_count = 0
    skip_count = 0
    for asset in github_release.assets:
        asset_age = now - asset.updated_at
        if asset_age.days > MAXIMUM_ASSET_AGE_DAYS:
            delete_count += 1

            if SHOULD_DELETE:
                # Wait for a small delay so we don't get blocked from GitHub API
                time.sleep(DELETE_DELAY)

                # Delete the old asset
                print(" - Delete %s (%s days old)" % (asset.name, asset_age.days))
                asset._delete(asset._api)
        else:
            skip_count += 1

    print("------------------")
    print("Deleted %s Assets" % delete_count)
    print("Skipped %s Assets" % skip_count)