Python製作簡易註冊登入系統_python

這次我主要講解如何用Python基於Flask的登入和註冊,驗證方式採用Basic Auth

主要用以下庫

import os#Flask的基礎庫from flask import Flask, abort, request, jsonify, g, url_for#Flaks的資料庫操作的庫from flask.ext.sqlalchemy import SQLAlchemy#Flask登入註冊的庫from flask.ext.httpauth import HTTPBasicAuth#加密解密密碼的庫from passlib.apps import custom_app_context as pwd_context#URL安全序列化工具from itsdangerous import (TimedJSONWebSignatureSerializer             as Serializer, BadSignature, SignatureExpired)

首先當然是初始化

app = Flask(__name__)# 設定密鑰app.config['SECRET_KEY'] = 'the quick brown fox jumps over the lazy dog'# 資料庫的配置app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True#資料庫初始化db = SQLAlchemy(app)# 驗證的初始化auth = HTTPBasicAuth()

然後是建模

SQLAlchemy是ORM模型操作資料庫的，所以是非常的方便
除了基本的屬性之後我們我定義了一些必要的方法

class User(db.Model):  __tablename__ = 'users'  id = db.Column(db.Integer, primary_key=True)  username = db.Column(db.String(32), index=True)  password_hash = db.Column(db.String(64))  # 加密密碼  def hash_password(self, password):    self.password_hash = pwd_context.encrypt(password)  # 驗證密碼  def verify_password(self, password):    return pwd_context.verify(password, self.password_hash)  # 產生token，並設定到期時間  def generate_auth_token(self, expiration=600):    s = Serializer(app.config['SECRET_KEY'], expires_in=expiration)    return s.dumps({'id': self.id})  # 靜態驗證token的方法  @staticmethod  def verify_auth_token(token):    s = Serializer(app.config['SECRET_KEY'])    try:      data = s.loads(token)    except SignatureExpired:      return None  # token到期    except BadSignature:      return None  # token無效    user = User.query.get(data['id'])    return user

註冊功能

@app.route('/api/users', methods=['POST'])def new_user():  username = request.json.get('username')  password = request.json.get('password')  if username is None or password is None:    abort(400)  # 使用者名稱或者密碼為空白  if User.query.filter_by(username=username).first() is not None:    abort(400)  # 使用者已存在  user = User(username=username)  # 加密密碼  user.hash_password(password)  # 儲存進資料庫  db.session.add(user)  db.session.commit()  # 成功註冊後返回使用者名稱，Location後面接著的是跳轉的地址  return (jsonify({'username': user.username}), 201,      {'Location': url_for('get_user', id=user.id, _external=True)})

登入功能

# 登入後擷取token@app.route('/api/token')@auth.login_requireddef get_auth_token():  # 設定token到期時間  token = g.user.generate_auth_token(600)  return jsonify({'token': token.decode('ascii'), 'duration': 600})

擷取token後之後，每次請求只需傳token就好了

我們可以通過一個方法驗證token是否有效

# 可以通過token或者帳號密碼登入@app.route('/api/resource')@auth.login_requireddef get_resource():  # 如果token有效話就返回username  return jsonify({'data': 'Hello, %s!' % g.user.username})

細心的人會發現上面這兩個方法前都帶有@auth.login_required，這其實就是奧妙之處

# 有@auth.login_required標誌的都要調用這個方法,傳token或者傳帳號和密碼@auth.verify_passworddef verify_password(username_or_token, password):  # 首先驗證token  user = User.verify_auth_token(username_or_token)  if not user:    # 然後再驗證使用者名稱和密碼    user = User.query.filter_by(username=username_or_token).first()    if not user or not user.verify_password(password):      return False  g.user = user  return True

最後寫一個入口方法

if __name__ == '__main__':  # 如果這個資料庫不存在就建立  if not os.path.exists('db.sqlite'):    db.create_all()  app.run(debug=True)

這樣就大功告成了

效果圖

註冊

登入

驗證token

以上就是本文的全部內容，希望對大家的學習有所協助，也希望大家多多支援雲棲社區。