vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Fix Github build trigger migration

Browse source
This commit is contained in:
Joseph Schorr 2015-09-19 14:34:46 -04:00
parent 0316f6e471
commit 1c6933a28d
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
util/migrate/migrategithubdeploykeys.py
View file

@ -24,7 +24,8 @@ def backfill_github_deploykeys():
.select(RepositoryBuildTrigger.id) .select(RepositoryBuildTrigger.id)
.where(RepositoryBuildTrigger.private_key >> None) .where(RepositoryBuildTrigger.private_key >> None)
.where(RepositoryBuildTrigger.service == github_service) .where(RepositoryBuildTrigger.service == github_service)
.limit(10)) .where(RepositoryBuildTrigger.used_legacy_github >> None)
.limit(100))
filtered_ids = [trigger.id for trigger in build_trigger_ids if trigger.id not in encountered] filtered_ids = [trigger.id for trigger in build_trigger_ids if trigger.id not in encountered]
if len(filtered_ids) == 0: if len(filtered_ids) == 0:
@ -39,15 +40,22 @@ def backfill_github_deploykeys():
with app.config['DB_TRANSACTION_FACTORY'](db): with app.config['DB_TRANSACTION_FACTORY'](db):
try: try:
query = RepositoryBuildTrigger.select(RepositoryBuildTrigger.id == trigger_id) query = RepositoryBuildTrigger.select().where(RepositoryBuildTrigger.id == trigger_id)
trigger = db_for_update(query).get() trigger = db_for_update(query).get()
except RepositoryBuildTrigger.DoesNotExist: except RepositoryBuildTrigger.DoesNotExist:
logger.debug('Could not find build trigger %s', trigger_id) logger.debug('Could not find build trigger %s', trigger_id)
continue continue
trigger.used_legacy_github = True
trigger.save()
handler = BuildTriggerHandler.get_handler(trigger) handler = BuildTriggerHandler.get_handler(trigger)
config = handler.config config = handler.config
if not 'build_source' in config:
logger.debug('Could not find build source for trigger %s', trigger_id)
continue
build_source = config['build_source'] build_source = config['build_source']
gh_client = handler._get_client() gh_client = handler._get_client()
@ -83,5 +91,8 @@ def backfill_github_deploykeys():
if __name__ == "__main__": if __name__ == "__main__":
logging.getLogger('boto').setLevel(logging.CRITICAL)
logging.getLogger('github').setLevel(logging.CRITICAL)
logging.config.fileConfig('conf/logging_debug.conf', disable_existing_loggers=False) logging.config.fileConfig('conf/logging_debug.conf', disable_existing_loggers=False)
backfill_github_deploykeys() backfill_github_deploykeys()