Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

Make decorators used in packaging preserve docstrings

This commit is contained in:
Éric Araujo 2011-06-17 21:10:21 +02:00
parent 980d344ff3
commit 3c8ca089b1
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Lib/packaging/pypi/simple.py
View File

@ -15,8 +15,8 @@ import urllib.parse
import urllib.error import urllib.error
import os import os
from fnmatch import translate from fnmatch import translate
from functools import wraps
from packaging import logger from packaging import logger
from packaging.metadata import Metadata from packaging.metadata import Metadata
from packaging.version import get_version_predicate from packaging.version import get_version_predicate
@ -53,8 +53,9 @@ REL = re.compile("""<([^>]*\srel\s*=\s*['"]?([^'">]+)[^>]*)>""", re.I)
def socket_timeout(timeout=SOCKET_TIMEOUT): def socket_timeout(timeout=SOCKET_TIMEOUT):
"""Decorator to add a socket timeout when requesting pages on PyPI. """Decorator to add a socket timeout when requesting pages on PyPI.
""" """
def _socket_timeout(func): def wrapper(func):
def _socket_timeout(self, *args, **kwargs): @wraps(func)
def wrapped(self, *args, **kwargs):
old_timeout = socket.getdefaulttimeout() old_timeout = socket.getdefaulttimeout()
if hasattr(self, "_timeout"): if hasattr(self, "_timeout"):
timeout = self._timeout timeout = self._timeout
@ -63,13 +64,14 @@ def socket_timeout(timeout=SOCKET_TIMEOUT):
return func(self, *args, **kwargs) return func(self, *args, **kwargs)
finally: finally:
socket.setdefaulttimeout(old_timeout) socket.setdefaulttimeout(old_timeout)
return _socket_timeout return wrapped
return _socket_timeout return wrapper
def with_mirror_support(): def with_mirror_support():
"""Decorator that makes the mirroring support easier""" """Decorator that makes the mirroring support easier"""
def wrapper(func): def wrapper(func):
@wraps(func)
def wrapped(self, *args, **kwargs): def wrapped(self, *args, **kwargs):
try: try:
return func(self, *args, **kwargs) return func(self, *args, **kwargs)