mirror of https://github.com/craigerl/aprsd.git
15 lines
343 B
Python
15 lines
343 B
Python
import rpyc
|
|
|
|
|
|
class AuthSocketStream(rpyc.SocketStream):
|
|
"""Used to authenitcate the RPC stream to remote."""
|
|
|
|
@classmethod
|
|
def connect(cls, *args, authorizer=None, **kwargs):
|
|
stream_obj = super().connect(*args, **kwargs)
|
|
|
|
if callable(authorizer):
|
|
authorizer(stream_obj.sock)
|
|
|
|
return stream_obj
|