1 #coding:utf8
2
3 from flask import Flask 4 from flask_sqlalchemy import SQLAlchemy 5
6 app = Flask(__name__) 7
8 # 设置链接数据库的url
9 app.config['SQLALCHEMY_DATABASE_URI']='mysql://root:mysql@127.0.0.1:3306/flask'
10
11 # 每次请求结束后自动提交数据
12 app.config['SQLALCHEMY_COMMIT_ON_TEAMDOWN'] = True 13
14 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True 15
16 # 查询时会显示原始的sql语句
17 app.config['SQLALCHEMY_ECHO'] = True 18
19 # 注册数据库
20 db = SQLAlchemy(app) 21
22
23
24
25 class Role(db.Model): 26 # 定义表名
27 __tablename__ = 'roles'
28 # 定义列对象
29 id = db.Column(db.Integer,primary_key=True) 30 name = db.Column(db.String(64),unique=True) 31
32 # 关系字段 用于两个方向查
33 user = db.relationship('User',backref='rel') 34
35 # repr() 方法现实一个可读字符串
36 def __repr__(self): 37 return str(self.name) 38
39
40 class User(db.Model): 41 __tablename__ = 'users'
42 id = db.Column(db.Integer,primary_key=True) 43 name = db.Column(db.String(64),unique=True,index=True) 44 email = db.Column(db.String(64),unique=True) 45 pswd = db.Column(db.String(64)) 46 role_id = db.Column(db.Integer,db.ForeignKey("roles.id")) 47
48 def __repr__(self): 49 return str(self.name) 50
51
52 if __name__ == '__main__': 53 db.drop_all() # 删除表
54 db.create_all() # 创建表
55 ro1 = Role(name = 'admin') # 创建对象
56 ro2 = Role(name = 'user' ) 57 # 向数据库添加数据
58 db.session.add_all([ro1,ro2]) 59 # 提交
60 # db.session.commit()
61
62
63 us1 = User(name='wang', email='wang@163.com', pswd='123456', role_id=ro1.id) 64 us2 = User(name='zhang', email='zhang@189.com', pswd='201512', role_id=ro2.id) 65 us3 = User(name='chen', email='chen@126.com', pswd='987654', role_id=ro2.id) 66 us4 = User(name='zhou', email='zhou@163.com', pswd='456789', role_id=ro1.id) 67 db.session.add_all([us1, us2, us3, us4]) 68 db.session.commit() 69
70
71 app.run(debug=True) 72