1 # Licensed to the Apache Software Foundation (ASF) under one or more
2 # contributor license agreements. See the NOTICE file distributed with
3 # this work for additional information regarding copyright ownership.
4 # The ASF licenses this file to You under the Apache License, Version 2.0
5 # (the "License"); you may not use this file except in compliance with
6 # the License. You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
17 Entry point functions.
20 from aria.orchestrator import operation
21 from . import local as local_operations
25 def run_script_locally(ctx,
29 return local_operations.run_script(
31 script_path=script_path,
37 def run_script_with_ssh(ctx,
44 return _try_import_ssh().run_script(
46 script_path=script_path,
47 fabric_env=fabric_env,
50 hide_output=hide_output,
55 def run_commands_with_ssh(ctx,
61 return _try_import_ssh().run_commands(
64 fabric_env=fabric_env,
66 hide_output=hide_output)
69 def _try_import_ssh():
71 from .ssh import operations as ssh_operations
73 except Exception as e:
75 raise RuntimeError('Failed to import SSH modules; Have you installed the ARIA SSH extra?')