import logging
from functools import wraps
from apps.utils.tasks import send_mail_task
import traceback
import hashlib
import json
from django.core.cache import cache
from rest_framework.exceptions import ParseError
from rest_framework.response import Response

myLogger = logging.getLogger('log')


def auto_log(name='', raise_exception=True, send_mail=False):
    def decorate(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            try:
                real_func = func(*args, **kwargs)
                return real_func
            except Exception:
                myLogger.error(name, exc_info=True)
                if send_mail:
                    send_mail_task.delay(message=traceback.format_exc())
                if raise_exception:
                    raise
        return wrapper
    return decorate


def idempotent(seconds=4):
    def decorate(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            request = args[1]
            rdata = dict(request.data)
            rdata['request_userid'] = getattr(request, 'user').id
            rdata['request_path'] = getattr(request, 'path')
            payload = json.dumps(rdata, sort_keys=True, ensure_ascii=False, default=str)
            hash_k = hashlib.sha256(payload.encode('utf-8')).hexdigest()
            hash_v_e = cache.get(hash_k, None)
            if hash_v_e is None:
                cache.set(hash_k, 'o', seconds)
                real_func = func(*args, **kwargs)
                if hasattr(real_func, 'data'):
                    cache.set(hash_k, real_func.data, seconds)
                else:
                    cache.delete(hash_k)
                return real_func
            elif hash_v_e == 'o':  # 说明请求正在处理
                raise ParseError(f'请求忽略,请{seconds}秒后重试')
            return Response(hash_v_e)
        return wrapper
    return decorate