2021-07-26 06:52:13 +02:00
|
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
|
|
|
|
|
|
import anyio
|
2021-07-26 08:29:20 +02:00
|
|
|
from functools import wraps
|
2021-07-26 06:52:13 +02:00
|
|
|
|
|
|
|
|
def shield(f):
|
2021-07-26 08:29:20 +02:00
|
|
|
@wraps(f)
|
2021-07-26 06:52:13 +02:00
|
|
|
async def shielded(*args, **kwargs):
|
2021-07-26 08:29:20 +02:00
|
|
|
with anyio.CancelScope(shield=True):
|
2021-07-26 06:52:13 +02:00
|
|
|
return await f(*args, **kwargs)
|
|
|
|
|
return shielded
|
2021-08-19 12:40:57 +02:00
|
|
|
|
|
|
|
|
def removeprefix(s, prefix):
|
|
|
|
|
try:
|
|
|
|
|
return s.removeprefix(prefix)
|
|
|
|
|
except AttributeError:
|
|
|
|
|
# compatibility for pre-3.9
|
|
|
|
|
return s[len(prefix):] if s.startswith(prefix) else s
|
