socketify.py/examples/middleware_sync.py

47 wiersze
1.1 KiB
Python
Czysty Zwykły widok Historia

from socketify import App
2022-11-16 19:28:46 +00:00
# this is just an example of implementation you can just import using from socketify import middleware for an more complete version
def middleware(*functions):
def middleware_route(res, req):
data = None
2022-11-16 19:28:46 +00:00
# cicle to all middlewares
for function in functions:
2022-11-16 19:28:46 +00:00
# call middlewares
data = function(res, req, data)
2022-11-16 19:28:46 +00:00
# stops if returns Falsy
if not data:
break
return middleware_route
2022-11-16 19:28:46 +00:00
def get_user(authorization_header):
if authorization_header:
2022-11-16 19:28:46 +00:00
return {"greeting": "Hello, World"}
return None
2022-11-16 19:28:46 +00:00
def auth(res, req, data=None):
2022-11-16 19:28:46 +00:00
user = get_user(req.get_header("authorization"))
if not user:
res.write_status(403).end("not authorized")
return False
2022-11-16 19:28:46 +00:00
# returns extra data
return user
2022-11-16 19:28:46 +00:00
def home(res, req, user=None):
2022-11-16 19:28:46 +00:00
res.end(user.get("greeting", None))
app = App()
app.get("/", middleware(auth, home))
2022-11-16 19:28:46 +00:00
app.listen(
3000,
lambda config: print("Listening on port http://localhost:%d now\n" % config.port),
)
app.run()