[py]flask实现用户cmdb

人最难做到的就是跟自己学习, 总是学习别人的, 沉淀, 过一段时间忘了刚学到的, 在向别人学习, 而非看看自己的沉淀, 殊不知, 世界上最亲近,最默契的人是自己.

用户cmdb功能概述

之前学flask做了个cmdb的东西.当初作为学习的一个思路.

久而久之就忘掉了, 整理一下放上来. 可见时间是把杀猪刀, 稍不留神就让你背负着忘掉的东西,回顾它, 然后反过来浪费时间. 这就是搞这些的烦恼.

根据后端存储可分为3个节点, 这点和那个txl项目: python写一个通讯录step by step V3.0有点类似,不过那个没flask, 这个是简化前端版.

• 支持登录
• crud数据库来实现用户的增删改查

功能如下

• 打开首页让登录,输入正确用户名密码后, 出现欢迎界面
• 界面可以增加用户
• 可以删除用户
• 可以更新用户密码
• 可以退出

第一阶段, 后端用文件存储

fileutils.py

file_dict = {}

# file -> dict
def read_file():
    with open("users.txt") as f:
        for line in f.readlines():
            user, pwd = line.split()
            file_dict[user] = pwd
    return file_dict

# dict -> file
def write_file():
    tmp = []
    for user, pwd in file_dict.items():
        tmp.append("%s %s" % (user, pwd))
    with open("users.txt", "w") as f:
        f.write("\n".join(tmp))

因为用户名密码是要对应起来, 因此这里考虑用了dict. 所以解决dict到文件之间的存储. 注意: 只有字符串才能写入文件.

app.py

#!/usr/bin/env python
# coding=utf-8

from flask import Flask, render_template, request, redirect, flash, session, url_for

import fileutils

app = Flask(__name__)
app.secret_key = "1992"

@app.route("/head")
def get_header():
    user_agent = request.headers.get("User-Agent")
    Cookie = request.headers.get("Cookie")
    return "{{ session.username }}------- %s" % Cookie

@app.route("/")
def login():
    if "username" in session:
        return redirect("/list")
    else:
        return render_template("login.html")

@app.route("/loginaction")
def login_action():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    fileutils.read_file()
    if fileutils.file_dict.has_key(user) and pwd == fileutils.file_dict[user]:
        session['username'] = user
        session['password'] = pwd
        return redirect("/list")
    else:
        flash("user or pwd is error")
        return render_template("login.html")

@app.route('/logout')
def logout():
    session.pop('username', None)
    return redirect("/")

@app.route("/list")
def show_user():
    print session
    if "username" in session:
        user_list = fileutils.read_file()
        return render_template("list.html", user_list=user_list.items())
    else:
        return redirect("/")

@app.route("/adduser")
def add_user():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    fileutils.read_file()
    if user in fileutils.file_dict:
        return redirect("/list")
    else:
        fileutils.file_dict[user] = pwd
        fileutils.write_file()
        return redirect("/list")

@app.route("/del")
def del_user():
    user = request.args.get("user")
    fileutils.read_file()
    fileutils.file_dict.pop(user)
    fileutils.write_file()
    return redirect("/list")

@app.route("/update")
def update_user():
    user = request.args.get("user")
    fileutils.read_file()
    pwd = fileutils.file_dict[user]
    return render_template("update.html", user=user, pwd=pwd)

@app.route("/updateaction")
def update_action():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    print user, pwd
    fileutils.read_file()
    print fileutils.file_dict
    fileutils.file_dict[user] = pwd
    print fileutils.file_dict
    fileutils.write_file()
    return redirect("/list")

app.run(debug=True)

第二阶段, 后端用pickcle模块

pickle可以直接将dict处理后写入文件, 也可以直接从文件读成dict, 写入方便, 读出来也不需要做额外处理就可以用了.

第三阶段, 后端用pickcle模块

后端使用mysql

create database bbs;
use bbs;
create table users(user varchar(40), pwd varchar(80));
insert into users values('admin','123456');

app.py

#!/usr/bin/env python
# coding=utf-8

from flask import Flask, render_template, request, redirect, flash, session,url_for
import fileutils
import pymysql as mysql

con = mysql.connect(host='127.0.0.1',user='root',passwd='',db='bbs')
cur = con.cursor()
sql = "select * from users"
cur.execute(sql)
con.autocommit(True)
print cur.fetchall()

app = Flask(__name__)
app.secret_key = "1992"

@app.route("/")
def login():
    if "username" in session:
        return redirect("/list")
    else:
        return render_template("login.html")

@app.route("/loginaction")
def login_action():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    cur.execute("select * from users")
    for line in cur.fetchall():
        suser,spwd = line
        print suser,spwd
        if suser == user and spwd == pwd:
            session['username']=user
            return redirect("/list")

    flash("user or pwd is error")
    return render_template("login.html")

@app.route('/logout')
def logout():
    session.pop('username', None)
    return redirect("/")

@app.route("/list")
def show_user():
    cur.execute("select * from users")
    if "username" in session:
        user_list = fileutils.read_file()
        return render_template("list.html", user_list=cur.fetchall())
    else:
        return redirect("/")

@app.route("/adduser")
def add_user():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    if user and pwd:
        cur.execute("select * from users")
        for line in cur.fetchall():
            suser,spwd = line
            if suser == user:
                return redirect("/list")
        else:
            cur.execute("insert into users values('%s','%s')"%(user,pwd))
            return redirect("/list")
    else:
        return redirect("/list")

@app.route("/del")
def del_user():
    user = request.args.get("user")
    cur.execute("delete from users where user ='%s';"% user)
    return redirect("/list")

@app.route("/update")
def update_user():
    user = request.args.get("user")
    cur.execute("select pwd from users where user='%s'"%user)
    pwd = cur.fetchall()[0][0]
    return render_template("update.html", user=user, pwd=pwd)

@app.route("/updateaction")
def update_action():
    user = request.args.get("user")
    pwd = request.args.get("pwd")
    print user, pwd
    cur.execute("update users set pwd='%s' where user = '%s'" % (pwd,user))
    return redirect("/list")

app.run(debug=True)

login.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>login</title>
</head>
<body>
<form action="/loginaction">
    user: <input type="text" name="user">
    pwd:  <input type="text" name="pwd">
    <input type="submit"> <br>
    {{ get_flashed_messages()[0] }}<br>
</form>
</body>
</html>

list.html

<!doctype html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport"
          content="width=device-width, user-scalable=no, initial-scale=1.0, maximum-scale=1.0, minimum-scale=1.0">
    <meta http-equiv="X-UA-Compatible" content="ie=edge">
    <title>list</title>
</head>
<body>
<h2>welcome <span style="color: pink">{{ session.username }}</span>!</h2>

<h1><a href="/logout">logout</a></h1>
<form action="/adduser">
    user: <input type="text" name="user">
    pwd:  <input type="text" name="pwd">
    <input type="submit" value="添加用户">
</form>
<hr>
<table border="1">
    <tr>
        <th>user</th>
        <th>pwd</th>
        <th>action</th>
    </tr>
    {% for user in user_list %}
    <tr>
        <td>{{ user[0] }}</td>
        <td>{{ user[1] }}</td>
        <td>
            <a href="/update?user={{ user[0] }}">update</a>
            <a href="/del?user={{ user[0] }}">del</a>
        </td>
    </tr>

    {% endfor %}
</table>

</body>
</html>

update.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>update user </title>
</head>
<body>
<form action="/updateaction">
    user: {{ user }} <input type="text" value="{{ user }}" name="user" hidden="hidden"><br>
    pwd: <input type="text" value="{{ pwd }}" name="pwd">
    <input type="submit">
</form>
</body>
</html>

源码