pleroma-ebooks/utils.py

# SPDX-License-Identifier: AGPL-3.0-only

import anyio
from functools import wraps

def shield(f):
	@wraps(f)
	async def shielded(*args, **kwargs):
		with anyio.CancelScope(shield=True):
			return await f(*args, **kwargs)
	return shielded
license compliance stuff and other changes relicense to AGPLv3 comply with the MPL's file copyleft some other changes in the direction of supporting GPT-2 which were hard to untangle from the rest of the changes (sorry) 2021-07-25 22:52:13 -06:00			`# SPDX-License-Identifier: AGPL-3.0-only`

			`import anyio`
make third_party.utils.make_toot async 2021-07-26 00:29:20 -06:00			`from functools import wraps`
license compliance stuff and other changes relicense to AGPLv3 comply with the MPL's file copyleft some other changes in the direction of supporting GPT-2 which were hard to untangle from the rest of the changes (sorry) 2021-07-25 22:52:13 -06:00
			`def shield(f):`
make third_party.utils.make_toot async 2021-07-26 00:29:20 -06:00			`@wraps(f)`
license compliance stuff and other changes relicense to AGPLv3 comply with the MPL's file copyleft some other changes in the direction of supporting GPT-2 which were hard to untangle from the rest of the changes (sorry) 2021-07-25 22:52:13 -06:00			`async def shielded(args, *kwargs):`
make third_party.utils.make_toot async 2021-07-26 00:29:20 -06:00			`with anyio.CancelScope(shield=True):`
license compliance stuff and other changes relicense to AGPLv3 comply with the MPL's file copyleft some other changes in the direction of supporting GPT-2 which were hard to untangle from the rest of the changes (sorry) 2021-07-25 22:52:13 -06:00			`return await f(args, *kwargs)`
			`return shielded`