diff --git a/pwdselfservice/settings.py b/pwdselfservice/settings.py index f71aee6..c5a15fb 100644 --- a/pwdselfservice/settings.py +++ b/pwdselfservice/settings.py @@ -1,4 +1,6 @@ +import logging.config import os +from django.utils.log import DEFAULT_LOGGING APP_ENV = os.getenv('APP_ENV') if APP_ENV == 'dev': @@ -24,49 +26,49 @@ LOG_PATH = os.path.join(BASE_DIR, 'log') if not os.path.isdir(LOG_PATH): os.mkdir(LOG_PATH) -LOGGING = { +# Disable Django's logging setup +LOGGING_CONFIG = None + +LOGLEVEL = os.environ.get('LOGLEVEL', 'info').upper() + +logging.config.dictConfig({ 'version': 1, - # 此选项开启表示禁用部分日志，不建议设置为True 'disable_existing_loggers': False, 'formatters': { - 'verbose': { - 'format': '%(asctime)s %(levelname)s %(message)s' - }, - 'simple': { - 'format': '%(asctime)s %(levelname)s %(message)s' - }, - }, - 'filters': { - 'require_debug_true': { - # 过滤器，只有当setting的DEBUG = True时生效 - '()': 'django.utils.log.RequireDebugTrue', + 'default': { + # exact format is not important, this is the minimum information + "format": "%(asctime)s %(module)s %(levelname)s -%(thread)d- %(message)s" }, + 'django.server': DEFAULT_LOGGING['formatters']['django.server'], }, 'handlers': { + # console logs to stderr 'console': { - 'level': 'DEBUG', - 'filters': ['require_debug_true'], + "level": "DEBUG", 'class': 'logging.StreamHandler', - 'formatter': 'verbose' + 'formatter': 'default', }, 'file': { - 'level': 'DEBUG', - 'class': 'logging.FileHandler', - # 日志保存文件 - 'filename': '%s/all.log' % LOG_PATH, - # 日志格式，与上边的设置对应选择 - 'formatter': 'verbose' - } + 'level': "DEBUG", + 'class': 'logging.handlers.RotatingFileHandler', + 'filename': os.path.join(LOG_PATH, "all.log"), + 'formatter': 'default', + 'maxBytes': 1024 * 1024 * 50, # 日志大小 10M + 'backupCount': 24, # 备份数为 2# 简单格式 + 'encoding': 'utf-8', + }, + 'django.server': DEFAULT_LOGGING['handlers']['django.server'], }, 'loggers': { - 'django': { - # 日志记录器 + # default for all undefined Python modules + '': { + 'level': LOGLEVEL, 'handlers': ['file'], - 'level': 'DEBUG', - 'propagate': True, - } + }, + # Default runserver request logging + 'django.server': DEFAULT_LOGGING['loggers']['django.server'], }, -} +}) # SESSION diff --git a/resetpwd/utils.py b/resetpwd/utils.py index eb56ee7..4571efe 100644 --- a/resetpwd/utils.py +++ b/resetpwd/utils.py @@ -11,6 +11,7 @@ import logging from ldap3.core.exceptions import LDAPException from django.conf import settings import os +from utils.tracecalls import decorator_logger APP_ENV = os.getenv('APP_ENV') if APP_ENV == 'dev': @@ -18,9 +19,10 @@ if APP_ENV == 'dev': else: from conf.local_settings import * -logger = logging.getLogger('django') +logger = logging.getLogger(__name__) +@decorator_logger(logger, log_head='AccountOps', pretty=True, indent=2, verbose=1) def code_2_user_detail(ops, home_url, code): """ 临时授权码换取userinfo @@ -29,6 +31,7 @@ def code_2_user_detail(ops, home_url, code): return _, s, e +@decorator_logger(logger, log_head='AccountOps', pretty=True, indent=2, verbose=1) def code_2_user_info_with_oauth2(ops, request, msg_template, home_url, code): """ 临时授权码换取userinfo @@ -53,6 +56,7 @@ def code_2_user_info_with_oauth2(ops, request, msg_template, home_url, code): return True, user_id, user_info +@decorator_logger(logger, log_head='AccountOps', pretty=True, indent=2, verbose=1) def ops_account(ad_ops, request, msg_template, home_url, username, new_password): """ ad 账号操作，判断账号状态，重置密码或解锁账号 diff --git a/resetpwd/views.py b/resetpwd/views.py index 6da138c..85533d7 100644 --- a/resetpwd/views.py +++ b/resetpwd/views.py @@ -9,6 +9,7 @@ from .form import CheckForm from .utils import code_2_user_detail, ops_account from django.conf import settings from utils.logger_filter import decorator_request_logger +from utils.tracecalls import decorator_logger APP_ENV = os.getenv('APP_ENV') if APP_ENV == 'dev': @@ -19,7 +20,7 @@ else: DING_CORP_ID, TITLE msg_template = 'messages.html' -logger = logging.getLogger('django') +logger = logging.getLogger(__name__) class PARAMS(object): @@ -43,7 +44,7 @@ scan_params = PARAMS() _ops = scan_params.ops -@decorator_request_logger(logger) +@decorator_logger(logger, log_head='Request', pretty=True, indent=2, verbose=1) def auth(request): home_url = '%s://%s' % (request.scheme, HOME_URL) corp_id = scan_params.corp_id @@ -56,25 +57,13 @@ def auth(request): global_title = TITLE if request.method == 'GET': - code = request.GET.get('code', None) - username = request.GET.get('username', None) - # 如果满足，说明已经授权免密登录 - if username and code and request.session.get(username) == code: - context = {'global_title': TITLE, - 'username': username, - 'code': code, - } - return render(request, 'reset_password.html', context) return render(request, 'auth.html', locals()) else: logger.error('[异常] 请求方法：%s，请求路径%s' % (request.method, request.path)) -@decorator_request_logger(logger) +@decorator_logger(logger, log_head='Request', pretty=True, indent=2, verbose=1) def index(request): - """ - 用户自行修改密码/首页 - """ home_url = '%s://%s' % (request.scheme, HOME_URL) scan_app = scan_params.AUTH_APP global_title = TITLE @@ -93,41 +82,45 @@ def index(request): else: _msg = check_form logger.error('[异常] 请求方法：%s，请求路径：%s，错误信息：%s' % (request.method, request.path, _msg)) - context = {'global_title': TITLE, - 'msg': _msg, - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': _msg, + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) # 格式化用户名 _, username = format2username(username) if _ is False: - context = {'global_title': TITLE, - 'msg': username, - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': username, + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) # 检测账号状态 auth_status, auth_result = AdOps().ad_auth_user(username=username, password=old_password) if not auth_status: - context = {'global_title': TITLE, - 'msg': str(auth_result), - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': str(auth_result), + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) return ops_account(AdOps(), request, msg_template, home_url, username, new_password) else: - context = {'global_title': TITLE, - 'msg': "不被接受的认证信息，请重新尝试认证授权。", - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': "不被接受的认证信息，请重新尝试认证授权。", + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) -@decorator_request_logger(logger) +@decorator_logger(logger, log_head='Request', pretty=True, indent=2, verbose=1) def reset_password(request): """ 钉钉扫码并验证信息通过之后，在重置密码页面将用户账号进行绑定 @@ -140,21 +133,23 @@ def reset_password(request): username = request.GET.get('username', None) # 如果满足，说明已经授权免密登录 if username and code and request.session.get(username) == code: - context = {'global_title': TITLE, - 'username': username, - 'code': code, - } + context = { + 'global_title': TITLE, + 'username': username, + 'code': code, + } return render(request, 'reset_password.html', context) else: if code: logger.info('[成功] 请求方法：%s，请求路径：%s，Code：%s' % (request.method, request.path, code)) else: logger.error('[异常] 请求方法：%s，请求路径：%s，未能拿到Code。' % (request.method, request.path)) - context = {'global_title': TITLE, - 'msg': "错误，临时授权码己失效，请尝试重新认证授权..", - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': "错误，临时授权码己失效，请尝试重新认证授权..", + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) try: # 用code换取用户基本信息 @@ -164,86 +159,91 @@ def reset_password(request): # 账号在企业微信或钉钉中是否是激活的 _, res = get_user_is_active(user_info) if not _: - context = {'global_title': TITLE, - 'msg': '当前扫码的用户未激活或可能己离职，用户信息如下：%s' % user_info, - 'button_click': "window.location.href='%s'" % home_url, - 'button_display': "返回主页" - } + context = { + 'global_title': TITLE, + 'msg': '当前扫码的用户未激活或可能己离职，用户信息如下：%s' % user_info, + 'button_click': "window.location.href='%s'" % home_url, + 'button_display': "返回主页" + } return render(request, msg_template, context) except Exception as callback_e: - context = {'global_title': TITLE, - 'msg': "错误[%s]，请与管理员联系." % str(callback_e), - 'button_click': "window.location.href='%s'" % home_url, - 'button_display': "返回主页" - } + context = { + 'global_title': TITLE, + 'msg': "错误[%s]，请与管理员联系." % str(callback_e), + 'button_click': "window.location.href='%s'" % home_url, + 'button_display': "返回主页" + } logger.error('[异常] ：%s' % str(callback_e)) return render(request, msg_template, context) # 通过user_info拿到用户邮箱，并格式化为username _, email = get_email_from_userinfo(user_info) if not _: - context = {'global_title': TITLE, - 'msg': email, - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': email, + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) _, username = format2username(email) if _ is False: - context = {'global_title': TITLE, - 'msg': username, - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': username, + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) - # 如果邮箱能提取到，则格式化之后，提取出账号提交到前端绑定 if username: request.session[username] = code - context = {'global_title': TITLE, - 'username': username, - 'code': code, - } + context = { + 'global_title': TITLE, + 'username': username, + 'code': code, + } return render(request, 'reset_password.html', context) else: - context = {'global_title': TITLE, - 'msg': "{}，您好，企业{}中未能找到您账号的邮箱配置，请联系HR完善信息。".format( - user_info.get('name'), scan_params.AUTH_APP), - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': "{}，您好，企业{}中未能找到您账号的邮箱配置，请联系HR完善信息。".format( + user_info.get('name'), scan_params.AUTH_APP), + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) # 重置密码页面，输入新密码后点击提交 elif request.method == 'POST': username = request.POST.get('username') code = request.POST.get('code') - if code and request.session.get(username) == code: + if username and code and request.session.get(username) == code: _new_password = request.POST.get('new_password').strip() try: return ops_account(ad_ops=AdOps(), request=request, msg_template=msg_template, home_url=home_url, username=username, new_password=_new_password) except Exception as reset_e: - context = {'global_title': TITLE, - 'msg': "错误[%s]，请与管理员联系." % str(reset_e), - 'button_click': "window.location.href='%s'" % home_url, - 'button_display': "返回主页" - } + context = { + 'global_title': TITLE, + 'msg': "错误[%s]，请与管理员联系." % str(reset_e), + 'button_click': "window.location.href='%s'" % home_url, + 'button_display': "返回主页" + } logger.error('[异常] ：%s' % str(reset_e)) return render(request, msg_template, context) - finally: - del request.session[username] - else: - context = {'global_title': TITLE, - 'msg': "认证已经失效，可尝试从重新认证授权。", - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } - return render(request, msg_template, context) + else: + context = { + 'global_title': TITLE, + 'msg': "认证已经失效，可尝试从重新认证授权。", + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } + return render(request, msg_template, context) -@decorator_request_logger(logger) +@decorator_logger(logger, log_head='Request', pretty=True, indent=2, verbose=1) def unlock_account(request): """ 解锁账号 @@ -255,53 +255,56 @@ def unlock_account(request): if request.method == 'GET': code = request.GET.get('code') username = request.GET.get('username') - if code and request.session.get(username) == code: - context = {'global_title': TITLE, - 'username': username, - 'code': code, - } + if username and code and request.session.get(username) == code: + context = { + 'global_title': TITLE, + 'username': username, + 'code': code, + } return render(request, 'unlock.html', context) else: - context = {'global_title': TITLE, - 'msg': "{}，您好，当前会话可能已经过期，请再试一次吧。".format(username), - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } + context = { + 'global_title': TITLE, + 'msg': "{}，您好，当前会话可能已经过期，请再试一次吧。".format(username), + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } return render(request, msg_template, context) if request.method == 'POST': username = request.POST.get('username') code = request.POST.get('code') - if request.session.get(username) and request.session.get(username) == code: + if username and code and request.session.get(username) == code: try: return ops_account(AdOps(), request, msg_template, home_url, username, None) except Exception as reset_e: - context = {'global_title': TITLE, - 'msg': "错误[%s]，请与管理员联系." % str(reset_e), - 'button_click': "window.location.href='%s'" % home_url, - 'button_display': "返回主页" - } + context = { + 'global_title': TITLE, + 'msg': "错误[%s]，请与管理员联系." % str(reset_e), + 'button_click': "window.location.href='%s'" % home_url, + 'button_display': "返回主页" + } logger.error('[异常] ：%s' % str(reset_e)) return render(request, msg_template, context) - finally: - del request.session[username] - else: - context = {'global_title': TITLE, - 'msg': "认证已经失效，请尝试从重新进行认证授权。", - 'button_click': "window.location.href='%s'" % '/auth', - 'button_display': "重新认证授权" - } - return render(request, msg_template, context) + else: + context = { + 'global_title': TITLE, + 'msg': "认证已经失效，请尝试从重新进行认证授权。", + 'button_click': "window.location.href='%s'" % '/auth', + 'button_display': "重新认证授权" + } + return render(request, msg_template, context) -@decorator_request_logger(logger) +@decorator_logger(logger, log_head='Request', pretty=True, indent=2, verbose=1) def messages(request): _msg = request.GET.get('msg') button_click = request.GET.get('button_click') button_display = request.GET.get('button_display') - context = {'global_title': TITLE, - 'msg': _msg, - 'button_click': button_click, - 'button_display': button_display - } + context = { + 'global_title': TITLE, + 'msg': _msg, + 'button_click': button_click, + 'button_display': button_display + } return render(request, msg_template, context) diff --git a/templates/unlock.html b/templates/unlock.html index 49067fe..2a25353 100644 --- a/templates/unlock.html +++ b/templates/unlock.html @@ -3,7 +3,7 @@ {% block paneltitle %}解锁账号{% endblock %} {% block middleblock %}
-
{% csrf_token %} + {% csrf_token %}
diff --git a/utils/ad_ops.py b/utils/ad_ops.py index 316a5eb..a7482d9 100644 --- a/utils/ad_ops.py +++ b/utils/ad_ops.py @@ -4,6 +4,8 @@ from ldap3.core.exceptions import LDAPInvalidCredentialsResult, LDAPOperationRes from ldap3.core.results import * from ldap3.utils.dn import safe_dn import os +from utils.tracecalls import decorator_logger +import logging APP_ENV = os.getenv('APP_ENV') if APP_ENV == 'dev': @@ -11,6 +13,8 @@ if APP_ENV == 'dev': else: from conf.local_settings import * +logger = logging.getLogger(__name__) + """ 根据以下网站的说明： https://docs.microsoft.com/zh-cn/troubleshoot/windows/win32/change-windows-active-directory-user-password @@ -86,6 +90,7 @@ class AdOps(object): except LDAPException as l_e: return False, LDAPException("LDAPException: " + str(l_e)) + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_auth_user(self, username, password): """ 验证账号 @@ -141,6 +146,7 @@ class AdOps(object): except Exception as e: return False, "AdOps Exception: {}".format(e) + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_get_user_displayname_by_account(self, username): """ 通过username查询某个用户的显示名 @@ -156,6 +162,7 @@ class AdOps(object): except Exception as e: return False, "AdOps Exception: {}".format(e) + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_get_user_dn_by_account(self, username): """ 通过username查询某个用户的完整DN @@ -168,10 +175,14 @@ class AdOps(object): attributes=['distinguishedName']) return True, str(self.conn.entries[0]['distinguishedName']) except IndexError: + logger.error("AdOps Exception: Connect.search未能检索到任何信息，当前账号可能被排除在之外，请联系管理员处理。") + logger.error("self.conn.search(BASE_DN, {}, attributes=['distinguishedName'])".format(SEARCH_FILTER.format(username))) return False, "AdOps Exception: Connect.search未能检索到任何信息，当前账号可能被排除在之外，请联系管理员处理。" except Exception as e: + logger.error("AdOps Exception: {}".format(e)) return False, "AdOps Exception: {}".format(e) + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_get_user_status_by_account(self, username): """ 通过username查询某个用户的账号状态 @@ -180,14 +191,18 @@ class AdOps(object): """ try: self.__conn() - self.conn.search(BASE_DN, SEARCH_FILTER.format(username), - attributes=['userAccountControl']) + self.conn.search(BASE_DN, SEARCH_FILTER.format(username), attributes=['userAccountControl']) return True, self.conn.entries[0]['userAccountControl'] except IndexError: + logger.error("AdOps Exception: Connect.search未能检索到任何信息，当前账号可能被排除在之外，请联系管理员处理。") + logger.error("self.conn.search({}, {}, attributes=['userAccountControl'])".format(BASE_DN, SEARCH_FILTER.format(username))) + logger.info("self.conn.entries -- {}".format(self.conn.entries)) return False, "AdOps Exception: Connect.search未能检索到任何信息，当前账号可能被排除在之外，请联系管理员处理。" except Exception as e: + logger.error("AdOps Exception: {}".format(e)) return False, "AdOps Exception: {}".format(e) + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_unlock_user_by_account(self, username): """ 通过username解锁某个用户 @@ -201,10 +216,12 @@ class AdOps(object): except IndexError: return False, "AdOps Exception: Connect.search未能检索到任何信息，当前账号可能被排除在之外，请联系管理员处理。" except Exception as e: + logger.error("AdOps Exception: {}".format(e)) return False, "AdOps Exception: {}".format(e) else: return False, user_dn + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_reset_user_pwd_by_account(self, username, new_password): """ 重置某个用户的密码 @@ -242,6 +259,7 @@ class AdOps(object): else: return False, user_dn + @decorator_logger(logger, log_head='AdOps', pretty=True, indent=2, verbose=1) def ad_get_user_locked_status_by_account(self, username): """ 通过username获取某个用户账号是否被锁定 diff --git a/utils/tracecalls.py b/utils/tracecalls.py new file mode 100644 index 0000000..799d5e8 --- /dev/null +++ b/utils/tracecalls.py @@ -0,0 +1,149 @@ +# -*- coding: utf-8 -*- +import os.path +import re +import sys +from copy import deepcopy as dcopy +from functools import wraps +from traceback import format_exc +from pwdselfservice.settings import BASE_DIR +from pprint import pformat as cformat + + +NOT_CHECK_CALL_FUNC_NAME = [] + +DEBUG_FLAG_FILE_PATH = os.path.join(BASE_DIR, '/log') + + +class TraceFuncContext: + def __init__(self, func_name, logger, log_head='Run function', debug_flag_name=None, verbose=1, + pretty=False, indent=0, check_calls=None): + self.name = func_name + self.logger = logger + self.debug_flag_name = debug_flag_name + self.log_head = log_head + self.verbose = verbose + self.pretty = pretty + self.indent = indent + self.check_calls = check_calls + self.copied_verbose = dcopy(verbose) + self.is_exit = False + self.debug_flag_suffix = '.debug.flag' + if self.verbose is not None: + self.check_calls = None + + def __enter__(self): + sys.settrace(self.get_callbacks) + + def check_debug_flag(self): + if self.debug_flag_name: + if os.path.isfile(os.path.join(DEBUG_FLAG_FILE_PATH, '{}{}'.format( + self.debug_flag_name, self.debug_flag_suffix))): + self.verbose = 2 + else: + self.verbose = self.copied_verbose + + @staticmethod + def check_in_excludes(func_name): + if len(NOT_CHECK_CALL_FUNC_NAME) > 0: + regex_gen = r'{0}'.format('|'.join(NOT_CHECK_CALL_FUNC_NAME)) + re_c = re.compile(regex_gen, flags=re.IGNORECASE) + return re_c.search(func_name) + return False + + def get_callbacks(self, frame, event, arg=None): + self.check_debug_flag() + __co_name = frame.f_code.co_name + if event != 'call': # Only trace call + return + if self.verbose: + if self.verbose == 1 or self.verbose == 'v': + self.check_calls = [self.name] + if __co_name not in self.check_calls: + return + elif self.verbose == 2 or self.verbose == 'vv': + if self.check_calls is None: + self.check_calls = list(set(frame.f_code.co_names)) + self.check_calls.append(self.name) + if __co_name not in self.check_calls or self.check_in_excludes(__co_name): + return + else: + raise ValueError("UNKNOWN VERBOSE VALUE: support verbose is 1/2 or v/vv...") + else: + if self.check_calls is None: + self.check_calls = [self.name] + else: + if isinstance(self.check_calls, list): + self.check_calls.append(self.name) + else: + raise ValueError("CHECK CALLS TYPE ERROR: check_calls need a list or tuple with function name.") + if __co_name not in self.check_calls: + return + return self.get_code_line + + def get_code_line(self, frame, event, agr=None): + # 正常情况下只有line或return事件才做记录 + if event not in ['line', 'return']: + return + __code = frame.f_code + __func_name = __code.co_name + __line_num = frame.f_lineno + __locals = frame.f_locals + if __func_name == self.name: + self.logger.info( + "{4} [{0}] trace detail " + "--- {1} {2}, locals as following: {5}{3}{5}".format(__func_name, + event, + __line_num, + cformat(__locals, indent=self.indent) + if self.pretty else __locals, + self.log_head, + '\n' if self.pretty else '' + )) + else: + self.logger.info( + "{5} [{0}] call [{1}] trace detail " + "--- {2} {3}, locals as following: {6}{4}{6}".format(self.name, + __func_name, + event, + __line_num, + cformat(__locals, indent=self.indent) + if self.pretty else __locals, + self.log_head, + '\n' if self.pretty else '' + )) + + def __exit__(self, exc_type, exc_val, exc_tb): + sys.settrace(None) + + +def decorator_logger(logger, log_head='Run function', debug_flag_name=None, verbose=1, check_calls=None, + pretty=False, indent=0): + def decorator(func): + @wraps(func) + def wrapper(*args, **kwargs): + func_consts = func.__code__.co_consts + logger.debug("{4} [{0}] entering trace with --- consts-{3}, args-{1}, kwargs-{2}...".format( + func.__name__, args, kwargs, func_consts, log_head)) + + with TraceFuncContext(func.__name__, logger, + log_head=log_head, + debug_flag_name=debug_flag_name, + verbose=verbose, + check_calls=check_calls, + pretty=pretty, + indent=indent + ): + try: + func_res = func(*args, **kwargs) + logger.debug("{4} [{0}] exiting trace with --- consts-{3}, args-{1}, kwargs-{2}...".format( + func.__name__, args, kwargs, func_consts, log_head)) + return func_res + except Exception as e: + logger.error( + "{4} [{0}] has exception, trace with --- consts-{3}, args-{1}, kwargs-{2}. Trackback as " + "following: ".format( + func.__name__, args, kwargs, func_consts, log_head)) + logger.error(format_exc()) + raise e + return wrapper + return decorator