mirror of https://github.com/akuker/RASCSI.git
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""
|
|
Module for sending and receiving data over a socket connection with the PiSCSI backend
|
|
"""
|
|
|
|
from flask import abort
|
|
from flask_babel import _
|
|
|
|
from piscsi.exceptions import (
|
|
EmptySocketChunkException,
|
|
InvalidProtobufResponse,
|
|
FailedSocketConnectionException,
|
|
)
|
|
from piscsi.socket_cmds import SocketCmds
|
|
|
|
|
|
class SocketCmdsFlask(SocketCmds):
|
|
"""
|
|
Class for sending and receiving data over a socket connection with the PiSCSI backend
|
|
"""
|
|
|
|
# pylint: disable=useless-super-delegation
|
|
def __init__(self, host="localhost", port=6868):
|
|
super().__init__(host, port)
|
|
|
|
def send_pb_command(self, payload):
|
|
"""
|
|
Takes a (str) containing a serialized protobuf as argument.
|
|
Establishes a socket connection with PiSCSI.
|
|
"""
|
|
try:
|
|
return super().send_pb_command(payload)
|
|
except FailedSocketConnectionException as err:
|
|
# After failing all attempts, throw a 404 error
|
|
abort(
|
|
404,
|
|
_(
|
|
"The PiSCSI Web Interface failed to connect to PiSCSI at "
|
|
"%(host)s:%(port)s with error: %(error_msg)s. The PiSCSI "
|
|
"process is not running or may have crashed.",
|
|
host=self.host,
|
|
port=self.port,
|
|
error_msg=str(err),
|
|
),
|
|
)
|
|
return None
|
|
|
|
def send_over_socket(self, sock, payload):
|
|
"""Sends a payload over a given socket"""
|
|
try:
|
|
return super().send_over_socket(sock, payload)
|
|
except EmptySocketChunkException:
|
|
abort(
|
|
503,
|
|
_(
|
|
"The PiSCSI Web Interface lost connection to PiSCSI. "
|
|
"Please go back and try again. "
|
|
"If the issue persists, please report a bug."
|
|
),
|
|
)
|
|
return None
|
|
except InvalidProtobufResponse:
|
|
abort(
|
|
500,
|
|
_(
|
|
"The PiSCSI Web Interface did not get a valid response from PiSCSI. "
|
|
"Please go back and try again. "
|
|
"If the issue persists, please report a bug."
|
|
),
|
|
)
|
|
return None
|