9975f38990
Log both the entrypoint and the return value from adding a task to Koji's database. We can measure both to ensure a task has been successfully added to the database as a service level indicator.
261 lines
7.7 KiB
Python
261 lines
7.7 KiB
Python
"""Koji osbuild integration for Koji Hub"""
|
|
import sys
|
|
import logging
|
|
import jsonschema
|
|
|
|
import koji
|
|
from koji.context import context
|
|
|
|
sys.path.insert(0, "/usr/share/koji-hub/")
|
|
import kojihub # pylint: disable=import-error, wrong-import-position
|
|
|
|
|
|
logger = logging.getLogger('koji.plugin.osbuild')
|
|
|
|
|
|
OSBUILD_IMAGE_SCHEMA = {
|
|
"$schema": "http://json-schema.org/draft-04/schema#",
|
|
"title": "osbuildImage arguments",
|
|
"type": "array",
|
|
"minItems": 7,
|
|
"items": [
|
|
{
|
|
"type": "string",
|
|
"description": "Name"
|
|
},
|
|
{
|
|
"type": "string",
|
|
"description": "Version"
|
|
},
|
|
{
|
|
"type": "string",
|
|
"description": "Distribution"
|
|
},
|
|
{
|
|
"oneOf": [
|
|
{
|
|
"type": "string",
|
|
"description": "Image Type"
|
|
},
|
|
{
|
|
"type": "array",
|
|
"description": "Image Types (this option is deprecated)",
|
|
"minItems": 1,
|
|
"maxItems": 1,
|
|
"items": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"type": "string",
|
|
"description": "Target"
|
|
},
|
|
{
|
|
"type": "array",
|
|
"description": "Architectures",
|
|
"minItems": 1,
|
|
"items": {
|
|
"type": "string"
|
|
}
|
|
},
|
|
{
|
|
"type": "object",
|
|
"$ref": "#/definitions/options"
|
|
}],
|
|
"definitions": {
|
|
"repo": {
|
|
"title": "Repository options",
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"properties": {
|
|
"baseurl": {
|
|
"type": "string"
|
|
},
|
|
"package_sets": {
|
|
"type": "array",
|
|
"description": "Repositories",
|
|
"items": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"ostree": {
|
|
"title": "OSTree specific options",
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"properties": {
|
|
"parent": {
|
|
"type": "string"
|
|
},
|
|
"ref": {
|
|
"type": "string"
|
|
},
|
|
"url": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
},
|
|
"options": {
|
|
"title": "Optional arguments",
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"properties": {
|
|
"customizations": {
|
|
"type": "object",
|
|
"additionalProperties": True
|
|
},
|
|
"ostree": {
|
|
"type": "object",
|
|
"$ref": "#/definitions/ostree"
|
|
},
|
|
"upload_options": {
|
|
"oneOf": [
|
|
{"$ref": "#/definitions/AWSEC2UploadOptions"},
|
|
{"$ref": "#/definitions/AWSS3UploadOptions"},
|
|
{"$ref": "#/definitions/GCPUploadOptions"},
|
|
{"$ref": "#/definitions/AzureUploadOptions"},
|
|
{"$ref": "#/definitions/ContainerUploadOptions"}
|
|
],
|
|
},
|
|
"repo": {
|
|
"type": "array",
|
|
"description": "Repositories",
|
|
"items": {
|
|
"oneOf": [
|
|
{"type": "string"},
|
|
{"$ref": "#/definitions/repo"}
|
|
]
|
|
}
|
|
},
|
|
"release": {
|
|
"type": "string",
|
|
"description": "Release override"
|
|
},
|
|
"skip_tag": {
|
|
"type": "boolean",
|
|
"description": "Omit tagging the result"
|
|
}
|
|
}
|
|
},
|
|
"AWSEC2UploadOptions": {
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"required": ["region", "share_with_accounts"],
|
|
"properties": {
|
|
"region": {
|
|
"type": "string",
|
|
},
|
|
"snapshot_name": {
|
|
"type": "string",
|
|
},
|
|
"share_with_accounts": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"AWSS3UploadOptions": {
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"required": ["region"],
|
|
"properties": {
|
|
"region": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
},
|
|
"AzureUploadOptions": {
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"required": ["tenant_id", "subscription_id", "resource_group", "location"],
|
|
"properties": {
|
|
"tenant_id": {
|
|
"type": "string"
|
|
},
|
|
"subscription_id": {
|
|
"type": "string"
|
|
},
|
|
"resource_group": {
|
|
"type": "string"
|
|
},
|
|
"location": {
|
|
"type": "string"
|
|
},
|
|
"image_name": {
|
|
"type": "string",
|
|
}
|
|
}
|
|
},
|
|
"GCPUploadOptions": {
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"required": ["region", "bucket"],
|
|
"properties": {
|
|
"region": {
|
|
"type": "string"
|
|
},
|
|
"bucket": {
|
|
"type": "string"
|
|
},
|
|
"image_name": {
|
|
"type": "string",
|
|
},
|
|
"share_with_accounts": {
|
|
"type": "array",
|
|
"items": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"ContainerUploadOptions": {
|
|
"type": "object",
|
|
"additionalProperties": False,
|
|
"properties": {
|
|
"name": {
|
|
"type": "string"
|
|
},
|
|
"tag": {
|
|
"type": "string"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
@koji.plugin.export
|
|
def osbuildImage(name, version, distro, image_type, target, arches, opts=None, priority=None):
|
|
"""Create an image via osbuild"""
|
|
context.session.assertPerm("image")
|
|
args = [name, version, distro, image_type, target, arches, opts]
|
|
task = {"channel": "image"}
|
|
|
|
logger.info("Create osbuildImage task")
|
|
|
|
try:
|
|
jsonschema.validate(args, OSBUILD_IMAGE_SCHEMA)
|
|
except jsonschema.exceptions.ValidationError as err:
|
|
raise koji.ParameterError(str(err)) from None
|
|
|
|
# Support array for backwards compatibility
|
|
# This check must be done after the schema validation
|
|
if isinstance(image_type, list):
|
|
image_type = image_type[0]
|
|
args = [name, version, distro, image_type, target, arches, opts]
|
|
|
|
if priority and priority < 0 and not context.session.hasPerm('admin'):
|
|
raise koji.ActionNotAllowed('only admins may create high-priority tasks')
|
|
|
|
# If task_id is returned from Koji Hub we assume
|
|
# that the task has been added to the database
|
|
task_id = kojihub.make_task('osbuildImage', args, **task)
|
|
if task_id:
|
|
logger.info("osbuildImage task %i added to database", task_id)
|
|
|
|
return task_id
|