def acr_task_create(cmd, # pylint: disable=too-many-locals
client,
task_name,
registry_name,
context_path,
file,
git_access_token=None,
image_names=None,
status='Enabled',
os_type=OS.linux.value,
cpu=DEFAULT_CPU,
timeout=DEFAULT_TIMEOUT_IN_SEC,
values=None,
source_trigger_name='defaultSourceTriggerName',
commit_trigger_enabled=True,
pull_request_trigger_enabled=True,
branch='master',
no_push=False,
no_cache=False,
arg=None,
secret_arg=None,
set_value=None,
set_secret=None,
base_image_trigger_name='defaultBaseimageTriggerName',
base_image_trigger_enabled=True,
base_image_trigger_type='Runtime',
resource_group_name=None):
if (commit_trigger_enabled or pull_request_trigger_enabled) and not git_access_token:
raise CLIError("If source control trigger is enabled [--commit-trigger-enabled] or "
"[--pull-request-trigger-enabled] --git-access-token must be provided.")
if file.endswith(ALLOWED_TASK_FILE_TYPES):
step = FileTaskStep(
task_file_path=file,
values_file_path=values,
context_path=context_path,
context_access_token=git_access_token,
values=(set_value if set_value else []) + (set_secret if set_secret else [])
)
else:
step = DockerBuildStep(
image_names=image_names,
is_push_enabled=not no_push,
no_cache=no_cache,
docker_file_path=file,
arguments=(arg if arg else []) + (secret_arg if secret_arg else []),
context_path=context_path,
context_access_token=git_access_token
)
registry, resource_group_name = validate_managed_registry(
cmd.cli_ctx, registry_name, resource_group_name, TASK_NOT_SUPPORTED)
source_control_type = SourceControlType.visual_studio_team_service.value
if context_path is not None and 'GITHUB.COM' in context_path.upper():
source_control_type = SourceControlType.github.value
source_triggers = None
source_trigger_events = []
if commit_trigger_enabled:
source_trigger_events.append(SourceTriggerEvent.commit.value)
if pull_request_trigger_enabled:
source_trigger_events.append(SourceTriggerEvent.pullrequest.value)
# if source_trigger_events contains any event types we assume they are enabled
if source_trigger_events:
source_triggers = [
SourceTrigger(
source_repository=SourceProperties(
source_control_type=source_control_type,
repository_url=context_path,
branch=branch,
source_control_auth_properties=AuthInfo(
token=git_access_token,
token_type=DEFAULT_TOKEN_TYPE,
scope='repo'
)
),
source_trigger_events=source_trigger_events,
status=TriggerStatus.enabled.value,
name=source_trigger_name
)
]
base_image_trigger = None
if base_image_trigger_enabled:
base_image_trigger = BaseImageTrigger(
base_image_trigger_type=base_image_trigger_type,
status=TriggerStatus.enabled.value if base_image_trigger_enabled else TriggerStatus.disabled.value,
name=base_image_trigger_name
)
task_create_parameters = Task(
location=registry.location,
step=step,
platform=PlatformProperties(
os=os_type,
architecture=Architecture.amd64.value
),
status=status,
timeout=timeout,
agent_configuration=AgentProperties(
cpu=cpu
),
trigger=TriggerProperties(
source_triggers=source_triggers,
base_image_trigger=base_image_trigger
)
)
try:
return client.create(resource_group_name=resource_group_name,
registry_name=registry_name,
task_name=task_name,
task_create_parameters=task_create_parameters)
except ValidationError as e:
raise CLIError(e)
def acr_task_update(cmd, # pylint: disable=too-many-locals
client,
task_name,
registry_name,
resource_group_name=None,
# task parameters
status=None,
os_type=None,
cpu=None,
timeout=None,
context_path=None,
commit_trigger_enabled=None,
pull_request_trigger_enabled=None,
git_access_token=None,
branch=None,
image_names=None,
no_push=None,
no_cache=None,
file=None,
values=None,
arg=None,
secret_arg=None,
set_value=None,
set_secret=None,
base_image_trigger_enabled=None,
base_image_trigger_type=None):
_, resource_group_name = validate_managed_registry(
cmd.cli_ctx, registry_name, resource_group_name, TASK_NOT_SUPPORTED)
task = client.get(resource_group_name, registry_name, task_name)
step = task.step
if arg is None and secret_arg is None:
arguments = None
else:
arguments = (arg if arg else []) + (secret_arg if secret_arg else [])
if set_value is None and set_secret is None:
set_values = None
else:
set_values = (set_value if set_value else []) + (set_secret if set_secret else [])
if file and file.endswith(ALLOWED_TASK_FILE_TYPES):
step = FileTaskStepUpdateParameters(
task_file_path=file,
values_file_path=values,
context_path=context_path,
context_access_token=git_access_token,
values=set_values
)
elif file and not file.endswith(ALLOWED_TASK_FILE_TYPES):
step = DockerBuildStepUpdateParameters(
image_names=image_names,
is_push_enabled=not no_push,
no_cache=no_cache,
docker_file_path=file,
arguments=arguments,
context_path=context_path,
context_access_token=git_access_token
)
elif step:
if isinstance(step, DockerBuildStep):
step = DockerBuildStepUpdateParameters(
image_names=image_names,
is_push_enabled=not no_push,
no_cache=no_cache,
docker_file_path=file,
arguments=arguments,
context_path=context_path,
context_access_token=git_access_token
)
elif isinstance(step, FileTaskStep):
step = FileTaskStepUpdateParameters(
task_file_path=file,
values_file_path=values,
context_path=context_path,
context_access_token=git_access_token,
values=set_values
)
source_control_type = None
if context_path:
if 'GITHUB.COM' in context_path.upper():
source_control_type = SourceControlType.github.value
else:
source_control_type = SourceControlType.visual_studio_team_service.value
# update trigger
source_trigger_update_params = None
base_image_trigger_update_params = None
if task.trigger:
source_triggers = task.trigger.source_triggers
base_image_trigger = task.trigger.base_image_trigger
if (commit_trigger_enabled or pull_request_trigger_enabled) or source_triggers:
source_trigger_events = _get_trigger_event_list(source_triggers,
commit_trigger_enabled,
pull_request_trigger_enabled)
source_trigger_update_params = [
SourceTriggerUpdateParameters(
source_repository=SourceUpdateParameters(
source_control_type=source_control_type,
repository_url=context_path,
branch=branch,
source_control_auth_properties=AuthInfoUpdateParameters(
token=git_access_token,
token_type=DEFAULT_TOKEN_TYPE
)
),
source_trigger_events=source_trigger_events,
status=TriggerStatus.enabled.value if source_trigger_events else TriggerStatus.disabled.value,
name=source_triggers[0].name if source_triggers else "defaultSourceTriggerName"
)
]
if base_image_trigger_enabled or base_image_trigger is not None:
status = None
if base_image_trigger_enabled is not None:
status = TriggerStatus.enabled.value if base_image_trigger_enabled else TriggerStatus.disabled.value
base_image_trigger_update_params = BaseImageTriggerUpdateParameters(
base_image_trigger_type=base_image_trigger_type,
status=status,
name=base_image_trigger.name if base_image_trigger else "defaultBaseimageTriggerName"
)
taskUpdateParameters = TaskUpdateParameters(
status=status,
platform=PlatformUpdateParameters(
os=os_type
),
agent_configuration=AgentProperties(
cpu=cpu
),
timeout=timeout,
step=step,
trigger=TriggerUpdateParameters(
source_triggers=source_trigger_update_params,
base_image_trigger=base_image_trigger_update_params
)
)
return client.update(resource_group_name, registry_name, task_name, taskUpdateParameters)