Python+Flask设置ip白名单

一、背景

基于 Flask 开发了一些 API接口，接口的控制需要访问权限，防止外部用户随意调用。

二、方案

获取请求方ip，然后判断此 ip 是否在白名单列表里面，如果不在则拒绝请求，这个白名单信息最好是可以通过动态更新或者数据库来维护，不然每次新增都需要上线操作，比较麻烦，代码示例如下：

# -*- coding: utf-8 -*-
from flask import Flask, request
from logzero import logger

# 创建Flask的应用程序
app = Flask(__name__)

# ip白名单
ip_whitelist = ["10.13.12.136", "10.23.56.135"]

# 以装饰器形式绑定路由
@app.route('/')
def hello_world():
    # 获取请求来源ip
    ip = request.remote_addr
    logger.info("ip: {}".format(ip))

    if ip not in ip_whitelist:
        return 'ip is not in whitelist!'

    return 'Hello World!'

if __name__ == '__main__':
    # 启动flask程序
    app.run(host="0.0.0.0", debug=True, port=8765)

根据机器名字获取 ip 信息：

获取ip
import socket
host_ip = socket.gethostbyname