Python SQLAlchemy 自动时间戳与密码哈希实现详解

时间戳处理与自动更新机制

一、时间戳

在实际开发场景中，通常期望数据模型中的 create_time 和 update_time 字段能够自动赋值。因此，在定义数据库模型时，需要对相关属性进行特殊配置处理。

from sqlalchemy.sql import func

class User(db.Model):
  __tablename__ = 'user'
  id = db.Column(db.Integer, primary_key = True)
  email = db.Column(db.String(128))
  password_hash = db.Column(db.String(128))
  create_time = db.Column(db.DateTime, server_default = func.now())
  update_time = db.Column(db.DateTime, server_default = func.now(), onupdate = func.now())

关于将 datetime 对象转换为时间戳的方法，例如从数据库读取的字符串格式为 '2020-07-09 14:15:31'，首先需要将其转换为 time.struct_time 对象：

import time timetuple = time.strptime('2020-07-09 14:15:31', '%Y-%m-%d %H:%M:%S')

如果已经获取到 datetime 对象，转换过程则更为简便：

now = datetime.datetime.now() timetuple = now.timetuple()

接下来，将得到的 timetuple 转换为时间戳：

timestamp = int(time.mktime(timetuple))

那么，时间戳又要如何转回datetime呢？方法如下：

last_time = datetime.datetime.fromtimestamp(timestamp)

密码安全存储与验证实现

二、密码管理

在数据库设计中，严禁明文存储用户密码，应使用散列值替代。以下是修改后的模型定义：

from sqlalchemy.sql import func
from werkzeug.security import generate_password_hash, check_password_hash

class User(db.Model):
  __tablename__ = 'user'
  id = db.Column(db.Integer, primary_key = True)
  email = db.Column(db.String(128))
  password_hash = db.Column(db.String(128))
  create_time = db.Column(db.DateTime, server_default = func.now())
  update_time = db.Column(db.DateTime, server_default = func.now(), onupdate = func.now())

  @property
  def password(self):
    raise AttributeError('password is not a readable attribute')

  @password.setter
  def password(self, password):
    self.password_hash = generate_password_hash(password)
  
  def verify_password(self, password):
    return check_password_hash(self.password_hash, password)

将 password 属性设置为只写，并通过 verify_password 方法对输入的密码进行校验。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。