diff --git a/config.go b/config.go index 691cac8..870d8cf 100644 --- a/config.go +++ b/config.go @@ -10,4 +10,5 @@ type Config struct { type Streamer struct { User string `yaml:"name"` Quality string `yaml:"quality"` + VOD bool `yaml:"vod"` } diff --git a/stream_pb2_grpc.py b/stream_pb2_grpc.py index cab87f5..60bb466 100644 --- a/stream_pb2_grpc.py +++ b/stream_pb2_grpc.py @@ -15,10 +15,10 @@ def __init__(self, channel): channel: A grpc.Channel. """ self.GetStream = channel.unary_unary( - '/protos.Stream/GetStream', - request_serializer=stream__pb2.StreamInfo.SerializeToString, - response_deserializer=stream__pb2.StreamResponse.FromString, - ) + "/protos.Stream/GetStream", + request_serializer=stream__pb2.StreamInfo.SerializeToString, + response_deserializer=stream__pb2.StreamResponse.FromString, + ) class StreamServicer(object): @@ -27,40 +27,53 @@ class StreamServicer(object): def GetStream(self, request, context): """Missing associated documentation comment in .proto file.""" context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') + context.set_details("Method not implemented!") + raise NotImplementedError("Method not implemented!") def add_StreamServicer_to_server(servicer, server): rpc_method_handlers = { - 'GetStream': grpc.unary_unary_rpc_method_handler( - servicer.GetStream, - request_deserializer=stream__pb2.StreamInfo.FromString, - response_serializer=stream__pb2.StreamResponse.SerializeToString, - ), + "GetStream": grpc.unary_unary_rpc_method_handler( + servicer.GetStream, + request_deserializer=stream__pb2.StreamInfo.FromString, + response_serializer=stream__pb2.StreamResponse.SerializeToString, + ), } generic_handler = grpc.method_handlers_generic_handler( - 'protos.Stream', rpc_method_handlers) + "protos.Stream", rpc_method_handlers + ) server.add_generic_rpc_handlers((generic_handler,)) - # This class is part of an EXPERIMENTAL API. +# This class is part of an EXPERIMENTAL API. class Stream(object): """Missing associated documentation comment in .proto file.""" @staticmethod - def GetStream(request, + def GetStream( + request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None, + ): + return grpc.experimental.unary_unary( + request, target, - options=(), - channel_credentials=None, - call_credentials=None, - insecure=False, - compression=None, - wait_for_ready=None, - timeout=None, - metadata=None): - return grpc.experimental.unary_unary(request, target, '/protos.Stream/GetStream', + "/protos.Stream/GetStream", stream__pb2.StreamInfo.SerializeToString, stream__pb2.StreamResponse.FromString, - options, channel_credentials, - insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + options, + channel_credentials, + insecure, + call_credentials, + compression, + wait_for_ready, + timeout, + metadata, + ) diff --git a/streamdl_proto_srv.py b/streamdl_proto_srv.py index aa5e058..0dcecbd 100644 --- a/streamdl_proto_srv.py +++ b/streamdl_proto_srv.py @@ -124,5 +124,5 @@ def get_stream(r): if __name__ == "__main__": try: serve() - except KeyboardInterrupt as e: + except KeyboardInterrupt: print("\nClosing Due To Keyboard Interrupt...")