o
    ;iG6                     @   s`   d Z ddlZddlmZ ddlmZ G dd deZ	G dd deZ
d	d
 ZG dd deZdS )zEClient and server classes corresponding to protobuf-defined services.    N)	empty_pb2)task_command_router_pb2c                   @   s   e Zd ZdZdd ZdS )TaskCommandRouterStub8Missing associated documentation comment in .proto file.c                 C   s   |j dtjjtjjd| _|j dtjjtjjd| _	|j dtj
jtjjd| _|jdtjjtjjd| _|j dtjjtjjd| _|j dtjjtjjd| _|j dtjjtjjd| _d	S )
zIConstructor.

        Args:
            channel: A grpc.Channel.
        9/modal.task_command_router.TaskCommandRouter/TaskExecPoll)request_serializerresponse_deserializer:/modal.task_command_router.TaskCommandRouter/TaskExecStart?/modal.task_command_router.TaskCommandRouter/TaskExecStdinWrite>/modal.task_command_router.TaskCommandRouter/TaskExecStdioRead9/modal.task_command_router.TaskCommandRouter/TaskExecWait?/modal.task_command_router.TaskCommandRouter/TaskMountDirectoryB/modal.task_command_router.TaskCommandRouter/TaskSnapshotDirectoryN)unary_unary+modal__proto_dot_task__command__router__pb2TaskExecPollRequestSerializeToStringTaskExecPollResponse
FromStringTaskExecPollTaskExecStartRequestTaskExecStartResponseTaskExecStartTaskExecStdinWriteRequestTaskExecStdinWriteResponseTaskExecStdinWriteunary_streamTaskExecStdioReadRequestTaskExecStdioReadResponseTaskExecStdioReadTaskExecWaitRequestTaskExecWaitResponseTaskExecWaitTaskMountDirectoryRequest"google_dot_protobuf_dot_empty__pb2EmptyTaskMountDirectoryTaskSnapshotDirectoryRequestTaskSnapshotDirectoryResponseTaskSnapshotDirectory)selfchannel r,   \/home/ubuntu/.local/lib/python3.10/site-packages/modal_proto/task_command_router_pb2_grpc.py__init__   sF   zTaskCommandRouterStub.__init__N)__name__
__module____qualname____doc__r.   r,   r,   r,   r-   r   	   s    r   c                   @   sH   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dS )TaskCommandRouterServicerr   c                 C       | tjj |d td)z7Poll for the exit status of an exec'd command.
        Method not implemented!set_codegrpc
StatusCodeUNIMPLEMENTEDset_detailsNotImplementedErrorr*   requestcontextr,   r,   r-   r   :      
z&TaskCommandRouterServicer.TaskExecPollc                 C   r4   )z'Execute a command in the task.
        r5   r6   r=   r,   r,   r-   r   A   r@   z'TaskCommandRouterServicer.TaskExecStartc                 C   r4   )z8Write to the stdin stream of an exec'd command.
        r5   r6   r=   r,   r,   r-   r   H   r@   z,TaskCommandRouterServicer.TaskExecStdinWritec                 C   r4   )zVGet a stream of output from the stdout or stderr stream of an exec'd command.
        r5   r6   r=   r,   r,   r-   r   O   r@   z+TaskCommandRouterServicer.TaskExecStdioReadc                 C   r4   )zEWait for an exec'd command to exit and return the exit code.
        r5   r6   r=   r,   r,   r-   r"   V   r@   z&TaskCommandRouterServicer.TaskExecWaitc                 C   r4   )z8Mount an image at a directory in the container.
        r5   r6   r=   r,   r,   r-   r&   ]   r@   z,TaskCommandRouterServicer.TaskMountDirectoryc                 C   r4   )zbSnapshot a directory with a mounted image, including any local changes, into a new image.
        r5   r6   r=   r,   r,   r-   r)   d   r@   z/TaskCommandRouterServicer.TaskSnapshotDirectoryN)r/   r0   r1   r2   r   r   r   r   r"   r&   r)   r,   r,   r,   r-   r3   7   s    r3   c                 C   s   t j| jtjjtjjdt j| jtj	jtj
jdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdd}t d|}||f d S )N)request_deserializerresponse_serializer)r   r   r   r   r"   r&   r)   z+modal.task_command_router.TaskCommandRouter)r8   unary_unary_rpc_method_handlerr   r   r   r   r   r   r   r   r   r   r   r   unary_stream_rpc_method_handlerr   r   r   r"   r    r!   r&   r#   r$   r%   r)   r'   r(   method_handlers_generic_handleradd_generic_rpc_handlers)servicerserverrpc_method_handlersgeneric_handlerr,   r,   r-   'add_TaskCommandRouterServicer_to_serverl   sP   %rK   c                   @   s   e Zd ZdZe								dddZe								dddZe								dd	d
Ze								dddZe								dddZ	e								dddZ
e								dddZdS )TaskCommandRouterr   r,   NFc
           
      C   ,   t j| |dtjjtjj||||||||	S )Nr   )r8   experimentalr   r   r   r   r   r   
r>   targetoptionschannel_credentialscall_credentialsinsecurecompressionwait_for_readytimeoutmetadatar,   r,   r-   r         zTaskCommandRouter.TaskExecPollc
           
      C   rM   )Nr	   )r8   rN   r   r   r   r   r   r   rO   r,   r,   r-   r      rY   zTaskCommandRouter.TaskExecStartc
           
      C   rM   )Nr
   )r8   rN   r   r   r   r   r   r   rO   r,   r,   r-   r      rY   z$TaskCommandRouter.TaskExecStdinWritec
           
      C   rM   )Nr   )r8   rN   r   r   r   r   r   r   rO   r,   r,   r-   r      rY   z#TaskCommandRouter.TaskExecStdioReadc
           
      C   rM   )Nr   )r8   rN   r   r   r    r   r!   r   rO   r,   r,   r-   r"      rY   zTaskCommandRouter.TaskExecWaitc
           
      C   s,   t j| |dtjjtjj||||||||	S )Nr   )	r8   rN   r   r   r#   r   r$   r%   r   rO   r,   r,   r-   r&      rY   z$TaskCommandRouter.TaskMountDirectoryc
           
      C   rM   )Nr   )r8   rN   r   r   r'   r   r(   r   rO   r,   r,   r-   r)     rY   z'TaskCommandRouter.TaskSnapshotDirectory)r,   NNFNNNN)r/   r0   r1   r2   staticmethodr   r   r   r   r"   r&   r)   r,   r,   r,   r-   rL      s    rL   )r2   r8   google.protobufr   r$   modal_protor   r   objectr   r3   rK   rL   r,   r,   r,   r-   <module>   s   .5,