from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

#设置连接数据库的URL
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/test'

app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
#查询时会显示原始SQL语句
app.config['SQLALCHEMY_ECHO'] = True
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)
    us = db.relationship('User', backref='role')

    #repr()方法显示一个可读字符串
    def __repr__(self):
        return 'Role:%s'% self.name

class User(db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True, index=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'))

    def __repr__(self):
        return 'User:%s'%self.name
if __name__ == '__main__':
    app.run(debug=True)

# 模型之前的关联

class Role(db.Model):
    ...
    #关键代码
    us = db.relationship('User', backref='role', lazy='dynamic')
    ...

class User(db.Model):
    ...
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))

#  常用的SQLAlchemy查询过滤器
        filter()         把过滤器添加到原查询上，返回一个新查询
        filter_by() 把等值过滤器添加到原查询上，返回一个新查询
        limit            使用指定的值限定原查询返回的结果
        offset()       偏移原查询返回的结果，返回一个新查询
        order_by()  根据指定条件对原查询结果进行排序，返回一个新查询
        group_by()  根据指定条件对原查询结果进行分组，返回一个新查询

# 常用的SQLAlchemy查询执行器

        all()       以列表形式返回查询的所有结果
        first()   返回查询的第一个结果，如果未查到，返回None
        first_or_404()   返回查询的第一个结果，如果未查到，返回404
        get()   返回指定主键对应的行，如不存在，返回None
        get_or_404()    返回指定主键对应的行，如不存在，返回404
        count() 返回查询结果的数量
        paginate()  返回一个Paginate对象，它包含指定范围内的结果

# 表的操作
# 创建表：
db.create_all()
#  删除表
db.drop_all()

# 数据的操作
# db.session.rollback()表示回滚，db.session.commit()表示执行事务，同MySQL事务
# 插入一条数据
ro1 = Role(name='admin')
db.session.add(ro1)
db.session.commit()
#再次插入一条数据
ro2 = Role(name='user')
db.session.add(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@itheima.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@itheima.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()

# 数据库的查询操作
User.query.first()
User.query.all()

#filter模糊查询，返回名字结尾字符为g的所有数据。
User.query.filter_by(name='wang').all()
# filter模糊查询，返回名字结尾字符为g的所有数据。
User.query.filter(User.name.endswith('g')).all()
#get():参数为主键，如果主键不存在没有返回内容
User.query.get()

#   逻辑或与非操作
#逻辑非，返回名字不等于wang的所有数据
User.query.filter(User.name!='wang').all()
# not_ 相当于取反
from sqlalchemy import not_
User.query.filter(not_(User.name=='chen')).all()
# 逻辑与，需要导入and，返回and()条件满足的所有数据
from sqlalchemy import and_
User.query.filter(and_(User.name!='wang',User.email.endswith('163.com'))).all()
# 逻辑或，需要导入or_
from sqlalchemy import or_
User.query.filter(or_(User.name!='wang',User.email.endswith('163.com'))).all()

# flask是在查询直接才进行的更新和删除操作
# 查询数据后删除
user = User.query.first()
db.session.delete(user)
db.session.commit()
User.query.all()

#更新数据
user = User.query.first()
user.name = 'dong'
db.session.commit()
User.query.first()

# 关联表的操作
#  角色和用户的关系是一对多的关系，一个角色可以有多个用户，一个用户只能属于一个角色。

#
#  查询roles表id为1的角色
ro1 = Role.query.get(1)
#  查询该角色的所有用户
ro1.us.all()

# 
#查询users表id为3的用户
us1 = User.query.get(3)
#查询用户属于什么角色
us1.role