laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Handle non-async middleware (#790)

Browse Source
This commit is contained in:
Nikhil Rao 2023-04-08 22:47:42 -07:00 committed by GitHub
parent e96f1c4d39
commit ae49759f1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pynecone/app.py
View File

@ -1,5 +1,6 @@
"""The main Pynecone app."""
import asyncio
import inspect
from typing import Any, Callable, Coroutine, Dict, List, Optional, Tuple, Type, Union
@ -167,9 +168,12 @@ class App(Base):
An optional state to return.
"""
for middleware in self.middleware:
out = await middleware.preprocess(app=self, state=state, event=event)
if asyncio.iscoroutinefunction(middleware.preprocess):
out = await middleware.preprocess(app=self, state=state, event=event)
else:
out = middleware.preprocess(app=self, state=state, event=event)
if out is not None:
return out
return out # type: ignore
async def postprocess(
self, state: State, event: Event, delta: Delta
@ -191,11 +195,16 @@ class App(Base):
An optional state to return.
"""
for middleware in self.middleware:
out = await middleware.postprocess(
app=self, state=state, event=event, delta=delta
)
if asyncio.iscoroutinefunction(middleware.postprocess):
out = await middleware.postprocess(
app=self, state=state, event=event, delta=delta
)
else:
out = middleware.postprocess(
app=self, state=state, event=event, delta=delta
)
if out is not None:
return out
return out # type: ignore
def add_middleware(self, middleware: Middleware, index: Optional[int] = None):
"""Add middleware to the app.