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

copy background task marker (#3255)

Browse Source
This commit is contained in:
benedikt-bartscher 2024-05-08 22:31:20 +02:00 committed by GitHub
parent bdcbe006a2
commit 38c46ec4d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
reflex/state.py
View File

@ -39,6 +39,7 @@ from redis.asyncio import Redis
from reflex import constants from reflex import constants
from reflex.base import Base from reflex.base import Base
from reflex.event import ( from reflex.event import (
BACKGROUND_TASK_MARKER,
Event, Event,
EventHandler, EventHandler,
EventSpec, EventSpec,
@ -579,6 +580,8 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
closure=fn.__closure__, closure=fn.__closure__,
) )
newfn.__annotations__ = fn.__annotations__ newfn.__annotations__ = fn.__annotations__
if mark := getattr(fn, BACKGROUND_TASK_MARKER, None):
setattr(newfn, BACKGROUND_TASK_MARKER, mark)
return newfn return newfn
@staticmethod @staticmethod