थोड़ा सा डेकोरेटर ट्रीटमेंट (शायद बहुत अच्छे से प्रेरित होकर मोनाड और लिफ्टिंग)। आप सुरक्षित रूप से पायथन 3.6 प्रकार के एनोटेशन को हटा सकते हैं और पुराने संदेश स्वरूपण शैली का उपयोग कर सकते हैं।
fallible.py
from functools import wraps
from typing import Callable, TypeVar, Optional
import logging
A = TypeVar('A')
def fallible(*exceptions, logger=None) \
-> Callable[[Callable[..., A]], Callable[..., Optional[A]]]:
"""
:param exceptions: a list of exceptions to catch
:param logger: pass a custom logger; None means the default logger,
False disables logging altogether.
"""
def fwrap(f: Callable[..., A]) -> Callable[..., Optional[A]]:
@wraps(f)
def wrapped(*args, **kwargs):
try:
return f(*args, **kwargs)
except exceptions:
message = f'called {f} with *args={args} and **kwargs={kwargs}'
if logger:
logger.exception(message)
if logger is None:
logging.exception(message)
return None
return wrapped
return fwrap
डेमो:
In [1] from fallible import fallible
In [2]: @fallible(ArithmeticError)
...: def div(a, b):
...: return a / b
...:
...:
In [3]: div(1, 2)
Out[3]: 0.5
In [4]: res = div(1, 0)
ERROR:root:called <function div at 0x10d3c6ae8> with *args=(1, 0) and **kwargs={}
Traceback (most recent call last):
File "/Users/user/fallible.py", line 17, in wrapped
return f(*args, **kwargs)
File "<ipython-input-17-e056bd886b5c>", line 3, in div
return a / b
In [5]: repr(res)
'None'
आप इस समाधान को संशोधित कर सकते हैं None
कि except
भाग की तुलना में कुछ अधिक वापस लौटाएं (या इस fallible
तर्क में इस रिटर्न वैल्यू को निर्दिष्ट करके समाधान को सामान्य बना दें )।
exception
विधि बस कॉलerror(message, exc_info=1)
। जैसे ही आपexc_info
किसी अपवाद संदर्भ से लॉगिंग विधियों में से किसी के पास जाते हैं , आपको एक ट्रेसबैक मिलेगा।