Pyhton日记——Flask写一个简易的知乎日报API

在上一篇Python日记——用Django做一个简易的知乎日报API中，很多人都了给了自己一些宝贵的意见，其中有一位朋友说到为啥不用轻巧的Flask呢？于是我就用Flask写了一个知乎日报的API供大家学习。

首先在PyCharm中新建Flask项目

之后下载所需的包

pip install flask
pip install flask-restful
pip install flask-sqlalchemy

flask-restful中的Api和Resource管理起路由很方便
flask-sqlalchemy中的SQLAlchemy可以用orm处理数据库，告别繁琐的sql语句

头部改为

# -*- coding: utf-8 -*-
from flask import Flask
from flask.ext.sqlalchemy import SQLAlchemy
from flask.ext.restful import Api,Resource

顶部配置

app = Flask(__name__)
# 格式为mysql://{用户名}:{密码}@{host}:{端口}/{数据库名}
app.config['SQLALCHEMY_DATABASE_URI']='mysql://root:@localhost:3306/flaskdb'
#这个提示你设为true
app.config['SQLALCHEMY_TRACK_MODIFICATIONS']='true'
db = SQLAlchemy(app)
api=Api(app)

Bean的编写
也是很普通的Bean的写法，只是多了一项配置而已，还要自己写一个json的转换方法

from datetime import datetime
class News(db.Model):
    # 一定要设置mysql_charset为utf8,不然会乱码
    __table_args__ = {'mysql_engine': 'InnoDB','mysql_charset': 'utf8'}
    # id为自增主键
    id=db.Column(db.Integer,primary_key=True,autoincrement=True)
    # created为创建的时间
    created = db.Column(db.DateTime,default = datetime.now)
    title = db.Column(db.String(100))
    image = db.Column(db.String(100))
    theme_id =db.Column(db.Integer)
    # 要将这个Bean手动转换为json
    def to_json(self):
        return {'id':self.id,'created':self.created.strftime("%Y-%m-%d %H:%M:%S"),'title':self.title,'image':self.image,'theme_id':self.theme_id}



class NewsDetail(db.Model):
    __table_args__ = {'mysql_engine': 'InnoDB','mysql_charset': 'utf8'}
    id=db.Column(db.Integer,primary_key=True,autoincrement=True)
    # 关联new的id为外键记得是小写
    news_id = db.Column(db.Integer, db.ForeignKey('news.id'))
    created = db.Column(db.DateTime,default = datetime.now)
    content  = db.Column(db.String(1000))
    image = db.Column(db.String(100))
    def to_json(self):
        return {'id':self.id,'news_id':self.news_id,'created':self.created.strftime("%Y-%m-%d %H:%M:%S"),'content':self.content,'image':self.image}

创建api

class NewsList(Resource):
    # get方法实现的内容
    def get(self):
        news_list=[]
        # 限制查询10个
        news=News.query.limit(10).all()
        for n in news:
            # 逐个转化为json后添加进数组
            news_list.append(n.to_json())
        # 完善一下json
        return {"stories":news_list}

class NewsDetailContent(Resource):
    def get(self,news_id):
        # 查询相应news_id的NewsDetail
        news_detail=NewsDetail.query.filter_by(news_id=news_id).first()
        return news_detail.to_json()

class ThemeNewsList(Resource):
    def get(self,theme_id):
        # 查询相应theme_id的News组
        news=NewsDetail.query.filter_by(theme_id=theme_id).all()
        news_list=[]
        for n in news:
            news_list.append(n.to_json)
        return {"stories":news_list}

# 查询最新消息的api
api.add_resource(NewsList,'/api/news/latest')
# 查询具体内容的api
api.add_resource(NewsDetailContent,'/api/news/<news_id>')
# 查询某个主题消息的api
api.add_resource(ThemeNewsList,'/api/news/theme/<theme_id>')

最后写入口方法

if __name__ == '__main__':
    # 自动根据上面的类建表，已建表则忽略
    db.create_all()
    app.run(debug=True)

我事先已经在数据库添加好一些数据了

效果如下

最新消息
消息详情
主题消息

最后推荐一个自动生成api的神器来自flask.ext.restless的APIManager，由于缺乏资料，这个学习起来比较困难，有兴趣的可以研究一下