velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

make JobManager.spawn copy current app context

This commit is contained in:
Luna 2019-10-25 09:38:48 -03:00
parent 4286b5d283
commit 7c878515e9
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
litecord/jobs.py
View File

@ -18,7 +18,9 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import asyncio import asyncio
from typing import Any
from quart.ctx import copy_current_app_context
from logbook import Logger from logbook import Logger
log = Logger(__name__) log = Logger(__name__)
@ -47,9 +49,14 @@ class JobManager:
def spawn(self, coro): def spawn(self, coro):
"""Spawn a given future or coroutine in the background.""" """Spawn a given future or coroutine in the background."""
task = self.loop.create_task(self._wrapper(coro))
@copy_current_app_context
async def _ctx_wrapper_bg() -> Any:
return await coro
task = self.loop.create_task(self._wrapper(_ctx_wrapper_bg()))
self.jobs.append(task) self.jobs.append(task)
return task
def close(self): def close(self):
"""Close the job manager, cancelling all existing jobs. """Close the job manager, cancelling all existing jobs.