浏览代码

copy background task marker (#3255)

benedikt-bartscher 1 年之前
父节点
当前提交
38c46ec4d5
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      reflex/state.py

+ 3 - 0
reflex/state.py

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