flask使用SQLAlchemy操作mysql
博客专区 > 鸿弟 的博客 > 博客详情
flask使用SQLAlchemy操作mysql
鸿弟 发表于1年前
flask使用SQLAlchemy操作mysql
  • 发表于 1年前
  • 阅读 73
  • 收藏 1
  • 点赞 0
  • 评论 0

腾讯云 新注册用户 域名抢购1元起>>>   

注意:看这篇博客需要先看我之前写的一篇博客,在 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官方文档应该有更具体的使用说明可以参考。

共有 人打赏支持
粉丝 64
博文 51
码字总数 33082
×
鸿弟
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: