Python的Web框架之Flask（6）

另一种方式连接数据库flasktest1，并建立roles和users两个表，并添加测试数据

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import and_, or_, not_

app = Flask(__name__)


host = 'localhost'
port = 3306
username = 'root'
password = 'root'
db = 'flasktest1'
connect_str = 'mysql+pymysql://{}:{}@{}:{}/{}'.format(username,password,host,port,db)
#  设置数据库连接
app.config['SQLALCHEMY_DATABASE_URI'] = connect_str
# 设置每次请求结束后自动提交数据库的改动
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
# 动态追踪设置
app.config['SQLALCHEMY_TRACK_MODUFICATIONS'] = True
# 显示原始sql
app.config['SQLALCHEMY_ECHO'] = True

# 数据库要和 app关联
db = SQLAlchemy(app)

class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    db.relationship('User', backref='role')
    def __repr__(self):
        return 'Role:{}'.format(self.name)

class User(db.Model):
    # 定义表名
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    email = db.Column(db.String(64), unique=True)
    password = db.Column(db.String(64))
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

    # relationship描述 role和user之间的关系, 第一个参数是类名User
    # 为类User声明新属性是方法

    def __repr__(self):
        return 'User:{}'.format(self.name)


@app.route('/')
def index():
    # 删除数据
    # 删除user中的第一条
    # user = User.query.first()
    # db.session.delete(user)
    # print("删除user中的第一条")
    # print(User.query.all())
    # 更新数据
    user = User.query.first()
    user.name = 'eric'
    db.session.commit()
    print("更新数据user中的第一条")
    print(User.query.first())
    return 'index'
if __name__ == '__main__':
    # 删除表
    db.drop_all()
    # 创建表
    db.create_all()
    ro1 = Role(name='admin')
    ro2 = Role(name='user')
    db.session.add_all([ro1, ro2])
    db.session.commit()
    us1 = User(name='wang', email='wang@163.com', password='123456', role_id=ro1.id)
    us2 = User(name='zhang', email='zhang@189.com', password='201512', role_id=ro2.id)
    us3 = User(name='chen', email='chen@126.com', password='987654', role_id=ro2.id)
    us4 = User(name='zhou', email='zhou@163.com', password='456789', role_id=ro1.id)
    us5 = User(name='tang', email='tang@neuedu.com', password='158104', role_id=ro2.id)
    us6 = User(name='wu', email='wu@gmail.com', password='5623514', role_id=ro2.id)
    us7 = User(name='qian', email='qian@gmail.com', password='1543567', role_id=ro1.id)
    us8 = User(name='liu', email='liu@neuedu.com', password='867322', role_id=ro1.id)
    us9 = User(name='li', email='li@163.com', password='4526342', role_id=ro2.id)
    us10 = User(name='sun', email='sun@163.com', password='235523', role_id=ro2.id)
    db.session.add_all([us1, us2, us3, us4, us5, us6, us7, us8, us9, us10])
    db.session.commit()

    app.run()

两个表的关联：

relationship描述 role和user之间的关系, 第一个参数是类名User

查询数据

1、查询所有用户数据
User.query.all()
2、查询有多少个用户
User.query.count()
3、查询第1个用户
User.query.first()
4、查询id为4的用户[3种方式]
User.query.get(4)
User.query.filter(User.id == 4).all()
User.query.filter_by(id=4).all()
5、查询名字结尾字符为g的所有数据[开始/包含]
User.query.filter(User.name.endswith("g")).all()
6、查询名字不等于wang的所有数据[2种方式]
User.query.filter(User.name != 'wang').all()
User.query.filter(not_(User.name == 'wang')).all()
7、查询名字和邮箱都以 li 开头的所有数据[2种方式]
User.query.filter(and_(User.name.startswith('li'), User.email.startswith('li'))).all()
User.query.filter(and_(User.name.like('li%'), User.email.like('li%'))).all()
8、查询password是 `123456` 或者 `email` 以 `neuedu.com` 结尾的所有数据
User.query.filter(or_(User.password == '123456', User.email.endswith('neuedu.com'))).all()
9、查询id为 [1, 3, 5, 7, 9] 的用户列表
User.query.filter(User.id.in_([1, 3, 5, 7, 9]))
10、查询name为liu的角色数据
User.query.filter_by(name="liu").all()
11、查询所有用户数据，并以邮箱排序
User.query.order_by(User.email).all()