Python学习笔记（十一）----Flask中使用SQL

一、SQLAlchemy

SQLAlchemy是个不错的orm框架，但是有点尴尬，因为flask的目的就是轻量，大多数情况于我而言就是随便写几个接口，如果功能复杂质量要求高，我就直接用java了。所以用flask写接口，也许我们不需要关注数据库里各个表中的关系，而只要知道每个sql语句返回的数据结构就足够了。

二、Flask中随性的SQL

不把SQLAlchemy当orm用，但不代表SQLAlchemy没有用。我们可以使用它对session的管理，而不需要自己再造轮子。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy


app = Flask('web_name')
app.config.from_object(webconfig)
db = SQLAlchemy(app)


# 查询
def get_user(account, pwd):
    sql = ('select id,name,account,pwd,createtime ' 
           + 'from publisher ' 
           + 'where account = "{0}" and pwd = "{1}"').format(account, pwd)

    data = db.session.execute(sql).first()

    if data != None:
        user = Publisher()
        user.id = data['id']
        user.name = data['name']
        user.account = data['account']
        user.pwd = data['pwd']
        user.createtime = data['createtime']
        return user
    else:
        return None


# 插入、修改、删除
def create_token(user,token):
    sql = ('insert into token ' 
           + '(tokenvalue,createtime,publisher_id) values ' 
           + '("{0}",now(),{1})').format(token, user.id)

    affect = db.session.execute(sql).rowcount
    db.session.commit()
    if affect != 0 :
        return True
    else:
        return False

# 插入后，查询最后插入的数据
def add_node(publish_id,message,name):
    sql1 = ('insert into node ' 
           + '(message,createtime,name,state) values ' 
           + '("{0}",now(),"{1}",1)').format(message,name)
    affect = db.session.execute(sql1).rowcount

    if affect == 1:
        sql2 = 'SELECT LAST_INSERT_ID()'
        node_id = db.session.execute(sql2).first()[0]

        sql3 = ('insert into publish_node '
               +'(publisher_id , node_id) values '
               +'({0},{1})').format(publish_id,node_id)
        affect2 = db.session.execute(sql3).rowcount
        if affect2 == 1:
            db.session.commit()
            return True
        else:
            return False
    else:
        return False