def create_asgi_client(instance):
yaml_paths = [file_relative_path(__file__, "./workspace.yaml")]
with WorkspaceProcessContext(
instance=instance,
workspace_load_target=WorkspaceFileTarget(paths=yaml_paths),
version="",
read_only=True,
) as process_context:
yield TestClient(DagitWebserver(process_context).create_asgi_app())
def _mgr_fn(instance, read_only):
"""Goes out of process but same process as host process"""
with WorkspaceProcessContext(
instance,
WorkspaceFileTarget(paths=[
file_relative_path(__file__, "multi_location.yaml")
]),
version="",
read_only=read_only,
) as workspace:
yield workspace
def create_subscription_context(instance):
ws = mock.Mock()
yaml_paths = [file_relative_path(__file__, "./workspace.yaml")]
with WorkspaceProcessContext(
instance=instance,
workspace_load_target=WorkspaceFileTarget(paths=yaml_paths),
version="",
read_only=True,
) as process_context:
yield GeventConnectionContext(ws, process_context)
def _mgr_fn(recon_repo, instance, read_only):
"""Goes out of process but same process as host process"""
check.inst_param(recon_repo, "recon_repo", ReconstructableRepository)
with WorkspaceProcessContext(
instance,
WorkspaceFileTarget(paths=[file_relative_path(__file__, "multi_location.yaml")]),
version="",
read_only=read_only,
) as workspace:
yield workspace
def created_workspace_load_target(kwargs):
check.dict_param(kwargs, "kwargs")
if are_all_keys_empty(kwargs, WORKSPACE_CLI_ARGS):
if kwargs.get("empty_workspace"):
return EmptyWorkspaceTarget()
if os.path.exists("workspace.yaml"):
return WorkspaceFileTarget(paths=["workspace.yaml"])
raise click.UsageError("No arguments given and workspace.yaml not found.")
if kwargs.get("workspace"):
_check_cli_arguments_none(
kwargs,
"python_file",
"working_directory",
"empty_working_directory",
"module_name",
"package_name",
"attribute",
"grpc_host",
"grpc_port",
"grpc_socket",
)
return WorkspaceFileTarget(paths=list(kwargs["workspace"]))
if kwargs.get("python_file"):
_check_cli_arguments_none(
kwargs,
"module_name",
"package_name",
"grpc_host",
"grpc_port",
"grpc_socket",
)
working_directory = get_working_directory_from_kwargs(kwargs)
return PythonFileTarget(
python_file=kwargs.get("python_file"),
attribute=kwargs.get("attribute"),
working_directory=working_directory,
location_name=None,
)
if kwargs.get("module_name"):
_check_cli_arguments_none(
kwargs,
"package_name",
"working_directory",
"empty_working_directory",
"grpc_host",
"grpc_port",
"grpc_socket",
)
return ModuleTarget(
module_name=kwargs.get("module_name"),
attribute=kwargs.get("attribute"),
location_name=None,
)
if kwargs.get("package_name"):
_check_cli_arguments_none(
kwargs,
"working_directory",
"empty_working_directory",
"grpc_host",
"grpc_port",
"grpc_socket",
)
return PackageTarget(
package_name=kwargs.get("package_name"),
attribute=kwargs.get("attribute"),
location_name=None,
)
if kwargs.get("grpc_port"):
_check_cli_arguments_none(
kwargs,
"attribute",
"working_directory",
"empty_working_directory",
"grpc_socket",
)
return GrpcServerTarget(
port=kwargs.get("grpc_port"),
socket=None,
host=(kwargs.get("grpc_host") if kwargs.get("grpc_host") else "localhost"),
location_name=None,
)
elif kwargs.get("grpc_socket"):
_check_cli_arguments_none(
kwargs,
"attribute",
"working_directory",
"empty_working_directory",
)
return GrpcServerTarget(
port=None,
socket=kwargs.get("grpc_socket"),
host=(kwargs.get("grpc_host") if kwargs.get("grpc_host") else "localhost"),
location_name=None,
)
else:
_cli_load_invariant(False)
def get_workspace_load_target(kwargs: Dict[str, str]):
check.dict_param(kwargs, "kwargs")
if are_all_keys_empty(kwargs, WORKSPACE_CLI_ARGS):
if kwargs.get("empty_workspace"):
return EmptyWorkspaceTarget()
if os.path.exists("workspace.yaml"):
return WorkspaceFileTarget(paths=["workspace.yaml"])
raise click.UsageError(
"No arguments given and workspace.yaml not found.")
if kwargs.get("workspace"):
_check_cli_arguments_none(
kwargs,
"python_file",
"working_directory",
"module_name",
"package_name",
"attribute",
"grpc_host",
"grpc_port",
"grpc_socket",
)
return WorkspaceFileTarget(
paths=list(cast(Union[List, Tuple], kwargs.get("workspace"))))
if kwargs.get("python_file"):
_check_cli_arguments_none(
kwargs,
"module_name",
"package_name",
"grpc_host",
"grpc_port",
"grpc_socket",
)
working_directory = get_working_directory_from_kwargs(kwargs)
return PythonFileTarget(
python_file=check.str_elem(kwargs, "python_file"),
attribute=check.opt_str_elem(kwargs, "attribute"),
working_directory=working_directory,
location_name=None,
)
if kwargs.get("module_name"):
_check_cli_arguments_none(
kwargs,
"package_name",
"grpc_host",
"grpc_port",
"grpc_socket",
)
working_directory = get_working_directory_from_kwargs(kwargs)
return ModuleTarget(
module_name=check.str_elem(kwargs, "module_name"),
attribute=check.opt_str_elem(kwargs, "attribute"),
working_directory=working_directory,
location_name=None,
)
if kwargs.get("package_name"):
_check_cli_arguments_none(
kwargs,
"grpc_host",
"grpc_port",
"grpc_socket",
)
working_directory = get_working_directory_from_kwargs(kwargs)
return PackageTarget(
package_name=check.str_elem(kwargs, "package_name"),
attribute=check.opt_str_elem(kwargs, "attribute"),
working_directory=working_directory,
location_name=None,
)
if kwargs.get("grpc_port"):
_check_cli_arguments_none(
kwargs,
"attribute",
"working_directory",
"grpc_socket",
)
return GrpcServerTarget(
port=check.int_elem(kwargs, "grpc_port"),
socket=None,
host=check.opt_str_elem(kwargs, "grpc_host") or "localhost",
location_name=None,
)
elif kwargs.get("grpc_socket"):
_check_cli_arguments_none(
kwargs,
"attribute",
"working_directory",
)
return GrpcServerTarget(
port=None,
socket=check.str_elem(kwargs, "grpc_socket"),
host=check.opt_str_elem(kwargs, "grpc_host") or "localhost",
location_name=None,
)
else:
_cli_load_invariant(False)
# necessary for pyright, does not understand _cli_load_invariant(False) never returns
assert False