class UserSigninLog(db.Model): """登录日志表""" bind_key = 'db2' tablename = 'user_signin_log' id = mapped_column(Integer, primary_key=True) userid = mapped_column(String(16), index=True) username = mapped_column(String(32)) ip_addr = mapped_column(String(32)) request_user_agent = mapped_column(String(256)) c_date = mapped_column(DateTime) @staticmethod def add_user_signin_log(userid: str, username: str): """新增登录日志""" user_signin_log = UserSigninLog() user_signin_log.userid = userid user_signin_log.username = username user_signin_log.ip_addr = request.access_route[0] user_signin_log.request_user_agent = request.headers.get("User-Agent") user_signin_log.c_date = datetime.now() db.session.add(user_signin_log) db.session.commit() return True @staticmethod def check_ip_and_header_within_x_days(userid: str, days=7): """检查x天内登录过的ip地址和请求头""" sql_query = select(UserSigninLog) \ .where(UserSigninLog.userid == userid, UserSigninLog.ip_addr == request.access_route[0], UserSigninLog.request_user_agent == request.headers.get("User-Agent"), UserSigninLog.c_date >= datetime.now() - timedelta(days=days)).limit(1) user_ip_addr_log = db.session.execute(sql_query).scalar_one_or_none() if user_ip_addr_log: return True return False 请帮我优化并建议

时间: 2024-02-17 22:00:02 浏览: 24
这段代码中的add_user_signin_log和check_ip_and_header_within_x_days方法都是静态方法,这意味着它们不依赖于类的实例,因此可以将它们提取到一个单独的辅助类中,以便更好地组织代码。同时,建议将数据库访问逻辑从模型类中分离出来,以便更好地实现单一职责原则。 下面是优化后的代码: ``` class UserSigninLog(db.Model): """登录日志表""" __bind_key__ = 'db2' __tablename__ = 'user_signin_log' id = db.Column(db.Integer, primary_key=True) userid = db.Column(db.String(16), index=True) username = db.Column(db.String(32)) ip_addr = db.Column(db.String(32)) request_user_agent = db.Column(db.String(256)) c_date = db.Column(db.DateTime) class UserSigninLogService: @staticmethod def add_user_signin_log(userid: str, username: str, ip_addr: str, user_agent: str): """新增登录日志""" user_signin_log = UserSigninLog() user_signin_log.userid = userid user_signin_log.username = username user_signin_log.ip_addr = ip_addr user_signin_log.request_user_agent = user_agent user_signin_log.c_date = datetime.now() db.session.add(user_signin_log) db.session.commit() @staticmethod def check_ip_and_header_within_x_days(userid: str, ip_addr: str, user_agent: str, days=7): """检查x天内登录过的ip地址和请求头""" user_ip_addr_log = UserSigninLog.query.filter_by(userid=userid, ip_addr=ip_addr, request_user_agent=user_agent).\ filter(UserSigninLog.c_date >= datetime.now() - timedelta(days=days)).first() return user_ip_addr_log is not None ``` 在优化后的代码中,将add_user_signin_log和check_ip_and_header_within_x_days方法提取到了UserSigninLogService类中。同时,将数据库访问逻辑从模型类中分离出来,使用query对象进行查询。在check_ip_and_header_within_x_days方法中使用了first方法而不是scalar_one_or_none方法,它们的效果是相同的,但是前者更加常用和易读。 建议对ip_addr和user_agent参数进行参数校验,确保它们的值是合法的。

相关推荐

import re import json import time import requests import datetime import pymysql import selenium from bs4 import BeautifulSoup from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from sqlalchemy import create_engine, Column, Integer, String, Text, DateTime from sqlalchemy.orm import sessionmaker from sqlalchemy.ext.declarative import declarative_base from selenium.webdriver import Edge, EdgeOptions # 创建浏览器对象 options = EdgeOptions() options.use_chromium = True options.binary_location = r'C:\Users\邓枫林\PycharmProjects\pythonProject\edgedriver_win64\msedgedriver.exe' browser = Edge(options=options) wait = WebDriverWait(browser, 10) # 打开微博话题页面 url = 'https://weibo.com/n/%E4%B8%AD%E5%9B%BD%E9%A3%9F%E5%93%81%E5%8D%AB%E7%94%9F?from=feed&loc=at&nick=%E4%B8%AD%E5%9B%BD%E9%A3%9F%E5%93%81%E5%8D%AB%E7%94%9F&order=hot' browser.get(url) # 等待页面加载完成 wait.until(lambda driver: driver.execute_script("return document.readyState") == "complete") browser = selenium.webdriver.Edge(executable_path='C:/Users/邓枫林/PycharmProjects/pythonProject/edgedriver_win64/msedgedriver.exe') wait = selenium.webdriver.support.ui.WebDriverWait(browser, 10) # 监测页面是否包含“高校类”敏感词汇 if '高校类' in browser.page_source: # 获取原始微博 weibo = browser.find_element_by_css_selector('.WB_feed_detail .WB_text.W_f14').text # 获取转发该微博的用户昵称和转发内容 reposts = [] repost_items = browser.find_elements_by_css_selector('.list_ul .list_li') for item in repost_items: nickname = item.find_element_by_css_selector('.WB_text.W_f14').text content = item.find_element_by_css_selector('.WB_text.W_f14 + .comment_txt').text reposts.append({'nickname': nickname, 'content': content}) # 关闭浏览器 browser.quit() # 将微博和转发内容存入MySQL数据库中 Base = declarative_base() class Weibo(Base): __tablename__ = 'weibo_user' id = Column(Integer, primary_key=True) content = Column(Text) create_time = Column(DateTime) class Repost(Base): __tablename__ = 'weibo_repost' id = Column(Integer, primary_key=True) weibo_id = Column(Integer) nickname = Column(String(50)) content = Column(Text) engine = create_engine('mysql+pymysql://root:root@hostname:port/weibo?charset=utf8mb4') Session = sessionmaker(bind=engine) session = Session() now = datetime.datetime.now() weibo_obj = Weibo(content=weibo, create_time=now) session.add(weibo_obj) session.commit() for repost in reposts: repost_obj = Repost(weibo_id=weibo_obj.id, nickname=repost['nickname'], content=repost['content']) session.add(repost_obj) session.commit() session.close() else: # 关闭浏览器 browser.quit()

最新推荐

recommend-type

MongoDB 3.6版本中bind_ip设置详解

主要给大家介绍了关于MongoDB 3.6版本中bind_ip设置的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

jdk10+ java.lang.NoClassDefFoundError: javax/xml/ws/Service 错误

jdk升级之后出现异常:java.lang.NoClassDefFoundError: javax/xml/ws/Service 1、解决方案jdk降级至1.8之前 2、由于JDK9以后,去除了javax... <groupId>javax.xml.bind <artifactId>jaxb-api <version>2.3.1 .....
recommend-type

OSEK标准_ISO 17356-2-2005_Part 2 OSEKVDX specifications for bindin

ISO 17356-2:2005 Part 2:OSEK/VDX specifications for binding OS,COM and NM 该标准最后一次审查和确认是在2020年。因此,该版本仍然是最新的。 摘要 ISO 17356-2:2005给出了OSEK/VDX规范,用于绑定嵌入式汽车...
recommend-type

网络编程网络编程网络编程

网络编程网络编程网络编程网络编程
recommend-type

setuptools-5.4.zip

Python库是一组预先编写的代码模块,旨在帮助开发者实现特定的编程任务,无需从零开始编写代码。这些库可以包括各种功能,如数学运算、文件操作、数据分析和网络编程等。Python社区提供了大量的第三方库,如NumPy、Pandas和Requests,极大地丰富了Python的应用领域,从数据科学到Web开发。Python库的丰富性是Python成为最受欢迎的编程语言之一的关键原因之一。这些库不仅为初学者提供了快速入门的途径,而且为经验丰富的开发者提供了强大的工具,以高效率、高质量地完成复杂任务。例如,Matplotlib和Seaborn库在数据可视化领域内非常受欢迎,它们提供了广泛的工具和技术,可以创建高度定制化的图表和图形,帮助数据科学家和分析师在数据探索和结果展示中更有效地传达信息。
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

用 Python 画一个可以动的爱心

好的,我可以帮助您使用Python来绘制一个动态的爱心。您可以使用turtle库来实现。以下是实现代码: ```python import turtle import math # 设置画布和画笔 canvas = turtle.Screen() canvas.bgcolor("black") pencil = turtle.Turtle() pencil.speed(0) pencil.color("red", "pink") pencil.pensize(3) # 定义爱心函数 def draw_love(heart_size, x_offset=0, y_offset=0):
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。