mirror
/
socketify.py
kopia lustrzana https://github.com/cirospaciari/socketify.py
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

fix middleware with mixed async and sync

pull/39/head
Ciro 2022-11-15 09:11:33 -03:00
rodzic 4bf88f436d
commit 7ed7325d25
1 zmienionych plików z 2 dodań i 2 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

4
examples/middleware_auto.py
Wyświetl plik

@ -9,7 +9,6 @@ def middleware(*functions):
for function in functions: for function in functions:
#detect if is coroutine or not #detect if is coroutine or not
if inspect.iscoroutinefunction(function): if inspect.iscoroutinefunction(function):
data = await function(res, req, data)
#in async query string, arguments and headers are only valid until the first await #in async query string, arguments and headers are only valid until the first await
if not some_async_as_run: if not some_async_as_run:
#get_headers will preserve headers (and cookies) inside req, after await #get_headers will preserve headers (and cookies) inside req, after await
@ -20,6 +19,7 @@ def middleware(*functions):
queries = req.get_queries() queries = req.get_queries()
#mark to only grab header, params and queries one time #mark to only grab header, params and queries one time
some_async_as_run = True some_async_as_run = True
data = await function(res, req, data)
else: else:
#call middlewares #call middlewares
data = function(res, req, data) data = function(res, req, data)
@ -55,6 +55,6 @@ def home(res, req, user=None):
res.end(user.get('greeting', None)) res.end(user.get('greeting', None))
app = App() app = App()
app.get("/", middleware(auth, another_midie, home)) app.get("/", middleware(auth, another_middie, home))
app.listen(3000, lambda config: print("Listening on port http://localhost:%d now\n" % config.port)) app.listen(3000, lambda config: print("Listening on port http://localhost:%d now\n" % config.port))
app.run() app.run()