flask笔记：5：数据库

忘是亡心i 2022-08-22 01:28 127阅读 0赞

linux和mac下需要将项目文件夹赋予权限，不然不能执行

flask中使用Flask-SQLAlchemy扩展来管理程序数据。

修改配置文件 myblog/config.py

CSRF_ENABLED=True
    SECRET_KEY='you-will-never-guess'
    import os
    basedir = os.path.abspath(os.path.dirname(__file__))
    SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(basedir, 'app.db')
    SQLALCHEMY_MIGRATE_REPO = os.path.join(basedir, 'db_repository')
    SQLALCHEMY_TRACK_MODIFICATIONS=True
    #db文件创建在与创建脚本同一目录下
    # SQLALCHEMY_DATABASE_URI是the Flask-SQLAlchemy必需的扩展。这是我们的数据库文件的路径。
    # SQLALCHEMY_MIGRATE_REPO 是用来存储SQLAlchemy-migrate数据库文件的文件夹。
    #SQLALCHEMY_TRACK_MODIFICATIONS 不设为True会报错(貌似段代码没有也能正常运行)

修改初始化脚本 app/\_\_init\_\_.py

from flask import Flask
    from flask.ext.sqlalchemy import SQLAlchemy
    app=Flask(__name__)
    app.config.from_object('config')
    db=SQLAlchemy(app)
    from app import views,models
    #导入Flask-SQLAlchemy
    #创建了一个 db 对象，这是我们的数据库
    #导入一个新的模块，叫做 models

编写models模块 app/models.py

from app import db
    
    class User(db.Model):
        id = db.Column(db.Integer, primary_key=True)
        nickname = db.Column(db.String(64), index=True, unique=True)
        email = db.Column(db.String(120), index=True, unique=True)
        posts = db.relationship('Post', backref='author', lazy='dynamic')
    
        def __repr__(self):
            return '<User %r>' % (self.nickname)
    
    class Post(db.Model):
        id = db.Column(db.Integer, primary_key = True)
        body = db.Column(db.String(140))
        timestamp = db.Column(db.DateTime)
        user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    
        def __repr__(self):
            return '<Post %r>' % (self.body)

表关系：

![Center][]

\#数据模型类要继承db.Model

db.Column( )是创建一列

db.Integer 是整数类型

db.String 是字符串类型

primary\_key = True 是主键

index=True 增加索引

unique=True 整表唯一

db.relationship( )，第一个参数是类名，第二个backref名字好像是随便(这个参数是在新增数据时user\_id 用 authou 代替 )，第三个固定lazy='dynamic'

db.relationship( )定义主键和外键的联系

db.ForeignKey( )是外键

\_\_repr\_\_方法告诉Python如何打印class对象，方便我们调试使用。

创建数据库 mybolg/db\_cj.py

from app import db
    db.create_all()
    #create_all( )是创建数据库

执行这个脚本就会创建数据库

测试新增用户 xz\_user.py

from app import db, models
    
    def add_com (u):
        db.session.add(u)
        db.session.commit()
    
    u = models.User(nickname='john', email='john@email.com')
    add_com(u)
    w=models.User(nickname='susan', email='susan@email.com')
    add_com(w)
    #session.add( )添加数据
    #session.commit( )提交事务

![Center 1][]

测试新增文章 xz\_body.py

import datetime
    from app import db, models
    
    u = models.User.query.get(1)
    p = models.Post(body='my first post!', timestamp=datetime.datetime.utcnow(), author=u)
    db.session.add(p)
    db.session.commit()
    #authou就是Post数据类里的user_id(应该是的)

![Center 2][]

测试查询用户 cx\_user.py

from app import db, models
    users = models.User.query.all()
    print users
    for u in users :
        print u.id,u.nickname
    e=models.User.query.get(1)
    print e
    p=models.User.query.filter_by(nickname="john").first()
    print p.id
    print p.email
    d=models.User.query.filter(models.User.email.endswith('@email.com')).all()
    print d

结果：

\[<User u'john'>, <User u'susan'>\]

1 john

2 susan

john@email.com

\[<User u'john'>, <User u'susan'>\]

\#query这个属性是查询，query.all( )是查询所有

\#query.get( )是查询主键

\#filter\_by( )是单条过滤条件，first( )是选择第一条

\#query.filter\_by(过滤条件).first( )根据过滤条件查询第一条

\#filter( )多条过滤条件

测试查询文章 cx\_body.py

from app import db, models
    u = models.User.query.get(1)
    print u
    posts = u.posts.all()
    print posts
    e=models.User.query.get(2)
    print e.posts.all()

结果：

\[<Post u'my first post!'>\]

\[  \]

测试清空数据库 db\_qk.py

from app import db, models
    users = models.User.query.all()
    for u in users:
        db.session.delete(u)
    posts = models.Post.query.all()
    for p in posts:
        db.session.delete(p)
    db.session.commit()
    #session.delete( )删除

![Center 3][]

![Center 4][]

[Center]: /images/20220724/abfdd0c9fce34bc3b6d927ed18145b68.png
[Center 1]: /images/20220724/eef42312db6e4c11bf628c4049cf87e0.png
[Center 2]: /images/20220724/1ae701a0ed83492cb36156e82bd81a12.png
[Center 3]: /images/20220724/ef380081db974f7d9469babda7da16c9.png
[Center 4]: /images/20220724/142f578d655d45ef8e54806f12db33a9.png