diff --git a/src/nucleus/cli/server.py b/src/nucleus/cli/server.py index 8b240e4..dece0af 100644 --- a/src/nucleus/cli/server.py +++ b/src/nucleus/cli/server.py @@ -39,11 +39,12 @@ async def server_run(): host, port ) - async with server: - await server.serve_forever() print(f"======= Serving on http://{host}:{port}/ ======") + async with server: + await server.serve_forever() + def handler_server(args): loop.add_signal_handler( diff --git a/src/nucleus/domains/http/models/preset.py b/src/nucleus/domains/http/models/preset.py new file mode 100644 index 0000000..9cde448 --- /dev/null +++ b/src/nucleus/domains/http/models/preset.py @@ -0,0 +1,12 @@ +__author__ = 'RemiZOffAlex' +__email__ = 'remizoffalex@mail.ru' + +from .response import Response + + +def abort(code, message=''): + return Response( + message, + mimetype='text/plain', + code=404 + ) diff --git a/src/nucleus/domains/http/models/response.py b/src/nucleus/domains/http/models/response.py index c77e2c8..f3722de 100644 --- a/src/nucleus/domains/http/models/response.py +++ b/src/nucleus/domains/http/models/response.py @@ -7,7 +7,12 @@ class Response: self.code = 200 self.phrase = 'Success' self.version = 'HTTP/1.1' + self.headers = {} + self.content_type = "text/plain" self.__data = data + for key in kwargs: + if key in self.__dict__: + setattr(self, key, kwargs[key]) def __call__(self): status = ' '.join([self.version, str(self.code), self.phrase])