Django网站单页面流量统计通用方式分享 ​

# 单页面浏览量记录模型，记录一些单页面的浏览量
class PageView(models.Model):
    url = models.CharField('页面地址', max_length=255, unique=True)  # 唯一性
    name = models.CharField('页面名称', max_length=255, blank=True, null=True)
    views = models.IntegerField('浏览量', default=0)
    is_compute = models.BooleanField('是否计算到访问量', default=True)  # 有的页面只记录，不计算
    create_date = models.DateTimeField(verbose_name='录入时间', auto_now_add=True)
    update_date = models.DateTimeField(verbose_name='更新时间', auto_now=True)

    def __str__(self):
        return self.url

    class Meta:
        verbose_name = "单页面浏览量"
        verbose_name_plural = verbose_name
        ordering = ['url']

    def update_views(self):
        self.views += 1
        self.save(update_fields=['views', 'update_date'])

def add_views(url, name=None, is_cache=True):
    """
    单页面访问量统计的视图函数装饰器
    @param is_cache: 是否使用缓存判断访问，跟文章的逻辑一样
    @param name: 页面名称，也可以是描述，方便辨认，没有实际作用
    @param url: 全局唯一，tool:ip这种格式，可以被解析成URL
    @return:
    """

    def decorator(func):
        def wrapper(request, *args, **kwargs):

            result = func(request, *args, **kwargs)
            # ******* 浏览量增加的逻辑 *******
            # 仅访问页面的时候才进行计算，接口调用不计算，管理员访问也不计算
            if request.method == "GET" and not request.is_ajax() and not request.user.is_superuser:
                # 获取或者创建一个实例
                page_views = PageView.objects.filter(url=url)
                if page_views:
                    obj = page_views.first()
                else:
                    obj = PageView(url=url, name=name, views=0)
                    obj.save()

                if is_cache:  # 要判断缓存，则存状态
                    cache_key = f'page_views:read:{url}'
                    is_read_time = request.session.get(cache_key)
                    if not is_read_time:
                        obj.update_views()
                        request.session[cache_key] = time.time()
                    else:
                        t = time.time() - is_read_time
                        if t > 60 * 30:
                            obj.update_views()
                            request.session[cache_key] = time.time()
                else:
                    obj.update_views()
            # ******* 浏览量增加的逻辑 *******

            return result

        return wrapper

    return decorator

@add_views('tool:tax', '综合所得年度汇算')
def tax(request):
    return render(request, 'tool/tax.html')

@method_decorator(add_views('blog:timeline', '时间线'), name='get')
class TimelineView(generic.ListView):
    model = Timeline
    template_name = 'blog/timeline.html'
    context_object_name = 'timeline_list'

    def get_ordering(self):
        return '-update_date',

def check_request_headers(headers_obj):
    """
    校验请求头信息，比如识别User-Agent，从而过滤掉该请求
    @param headers_obj: request.headers对象
    @return:
    use: flag = check_request_headers(request.headers)
    """
    # 常见的搜索引擎爬虫的请求头，还有Python的
    # 无请求头或者请求头里面包含爬虫信息则返回False，否则返回True
    user_agent_black_keys = ['spider', 'bot', 'python']
    if not headers_obj.get('user-agent'):
        return False
    else:
        user_agent = str(headers_obj.get('user-agent')).lower()
        for key in user_agent_black_keys:
            if key in user_agent:
                logger.warning(f'Bot/Spider request user-agent：{user_agent}')
                return False
    return True