flask使用SQLAlchemy操做mysql

注意：看这篇博客须要先看我以前写的一篇博客，在 https://my.oschina.net/zhuangweihong/blog/780913 （flask获取远程服务器的硬件信息） 这篇博客中有提到，获取远程服务器的硬件信息能够入mysql数据库，如今来实现下。还实现了下分页的功能，以下截图：

 

首先应该了解下SQLAlchemy框架对mysql数据库的操做，使用SQLAlchemy，除了要安装SQLAlchemy外，还须要安装MySQL-python，安装完后看下代码：

from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://xxx:xxx@192.168.xxx.xxx/xxx'
db = SQLAlchemy(app)
#添加服务器
def server_add(server1):
    db.session.add(server1)
    db.session.commit()

#删除服务器
def server_delete(server1):
    db.session.delete(server1)
    db.session.commit()

#查询服务器(经过ip)
def server_query_from_ip(ip):
    server1 = server.Server.query.filter_by(ip = ip).first()
    return server1

#查询服务器(全部)
def server_query_all():
    server_all = server.Server.query.all()
    return server_all

#查询服务器(分页)
def server_query_page(page_now, page_size):
    server_page = server.Server.query.paginate(page_now, page_size, False)
    return server_page

db.session.add为插入操做，db.session.delete为删除操做，query相关的为查询操做。SQLAlchemy框架底层调用了MySQL-python，并进行封装，不用再使用sql语句对数据库进行操做了。上述代码有使用到一个server类，server类跟数据库的servers表相对应，server类的定义和表的定义以下：

class Server(db.Model):
    __tablename__ = 'servers'
    id = db.Column(db.Integer, primary_key=True)
    ip = db.Column(db.String(128), unique=True)
    hostname = db.Column(db.String(128))
    cpu = db.Column(db.Integer)
    memory = db.Column(db.Integer)
    load = db.Column(db.String(128))
    ulimit = db.Column(db.Integer)
    version = db.Column(db.String(128))
    selinux = db.Column(db.String(128))

CREATE TABLE `servers` (
  `id` int(6) NOT NULL AUTO_INCREMENT,
  `ip` varchar(128) DEFAULT NULL,
  `hostname` varchar(128) DEFAULT NULL,
  `cpu` int(3) DEFAULT NULL,
  `memory` int(8) DEFAULT NULL,
  `load` varchar(128) DEFAULT NULL,
  `ulimit` int(9) DEFAULT NULL,
  `version` varchar(128) DEFAULT NULL,
  `selinux` varchar(128) DEFAULT NULL,
  PRIMARY KEY (`id`),
  UNIQUE KEY `ip` (`ip`)
) ENGINE=InnoDB AUTO_INCREMENT=36 DEFAULT CHARSET=utf8;

如上定义，基本可完成对数据的操做，但还有不少可优化的地方。

 

前端代码以下，现主要先分为两部分，一个是提交ip用来取ip的数据而后入库的，另一个是分页展现服务器的硬件信息(分页的一些对象会在最后进行讲解)：

{% extends "basic.html" %}
{% block page_content %}
    <form role="form" class="form-inline" method="post" action="/db/server/add">
        <textarea class="form-control" rows="1" name="searchip" placeholder="请输入ip"></textarea>
        <button type="submit" class="btn btn-default">提交</button>
    </form>

    <div style="height: 260px;">
        <table class="table table-bordered">
            <thead>
            <tr>
                <th>服务器ip</th>
                <th>主机名</th>
                <th>cpu内存</th>
                <th>负载</th>
                <th>资源数限制</th>
                <th>系统版本</th>
                <th>selinux状态</th>
                <th>操做</th>
            </tr>
            </thead>
            <tbody>
            {% for server1 in object_list %}
                <tr>
                    <td>{{ server1.ip }}</td>
                    <td>{{ server1.hostname }}</td>
                    <td>{{ server1.cpu }}核{{ server1.memory }}M</td>
                    <td>{{ server1.load }}</td>
                    <td>{{ server1.ulimit }}</td>
                    <td>{{ server1.version }}</td>
                    <td>{{ server1.selinux }}</td>
                    <td><a href="/db/server/delete?ip={{ server1.ip }}"><span class="label label-warning">delete</span></a></td>
                </tr>
            {% endfor %}
            </tbody>
        </table>
    </div>
    <ul class="pagination">
        {% if server_page.has_prev %}
            <li><a href="/db/server/query/page?page_now={{ server_page.prev_num }}">&laquo;</a></li>
        {% else %}
            <li class="disabled"><a href="#">&laquo;</a></li>
        {% endif %}

        {% for i in range(1, (server_page.pages + 1)) %}
            {% if i == server_page.page %}
                <li class="disabled"><a href="#">{{ i }}</a></li>
            {% else %}
                <li><a href="/db/server/query/page?page_now={{ i }}">{{ i }}</a></li>
            {% endif %}
        {% endfor %}

        {% if server_page.has_next %}
            <li><a href="/db/server/query/page?page_now={{ server_page.next_num }}">&raquo;</a></li>
        {% else %}
            <li class="disabled"><a href="#">&raquo;</a></li>
        {% endif %}
    </ul>
{% endblock %}

 

输入ip去提取服务器的硬件并入库的操做的代码以下(可输入多个ip，但ip不能重复，这个能够优化作个判断，若是ip已存在，就更新数据库的信息)：

#服务器信息入库
@app.route('/db/server/add', methods=['GET', 'POST'])
def db_server_add():
    ip = []
    p = re.compile(r'(?:(?:25[0-5]|2[0-4]\d|[01]?\d\d?)\.){3}(?:25[0-5]|2[0-4]\d|[01]?\d\d?)')
    if request.method == 'POST' and p.findall(request.form['searchip']):
        ip = p.findall(request.form['searchip'])
    for i in range(0, len(ip)):
        info = collect_info(ip[i], 'root', '/root/.ssh/id_rsa', 22)
        server1 = server.Server(ip=info['ip'], hostname = info['hostname'], cpu = info['cpuinfo'], memory = info['meminfo'],
                                 load=info['loadavg'], ulimit = info['ulimit'], version = info['release'], selinux = info['getenforce'])
        modules_db.server_add(server1)
    return render_template('200.html')

此代码就直接使用collect_info收集远程服务器的硬件信息，而后调用SQLAlchemy框架的db.session.add入库。

另外，SQLAlchemy框架把分页的功能弄得比较简单，SQLAlchemy框架自身把当前页，总页数等都进行了封装，分页的代码以下：

#服务器信息查询分页
@app.route('/db/server/query/page', methods=['GET', 'POST'])
def db_server_query_page():
    page_size = 6
    server_page = modules_db.server_query_page(1, page_size)
    page_now = request.values.get('page_now')
    if page_now:
        server_page = modules_db.server_query_page(int(page_now), page_size)
    object_list = server_page.items
    return render_template('db/db_server_query_page.html', object_list  = object_list, server_page = server_page)

SQLAlchemy框架的query.paginate有三个参数，第一个参数为当前页，第二个参数为每一个页的数据，第三个参数为超出是否报错。返回的server_page里面包含有分页的信息等，如server_page.items表明的是当前页面的全部内容，须要在前端展现。再回到前端页面能够看到我有用到object_list把当前页面的内容进行展示；server_page.pages提取分页的总页数；server_page.has_prev判断是否有上页，若是有，点击显示上一页server_page.prev_num

目前实现都是先从网上查询到的一些知识，另外，SQLAlchemy官方文档应该有更具体的使用说明能够参考。

 

最后附上本人的网络课堂地址，若有兴趣请点击： 实践哥