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

用户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>

巴特西

[py]flask实现用户cmdb

用户cmdb功能概述

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

fileutils.py

app.py

第二阶段, 后端用pickcle模块

第三阶段, 后端用pickcle模块

后端使用mysql

app.py

login.html

list.html

update.html

源码

最新文章

热门文章