在上一篇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,由于缺乏资料,这个学习起来比较困难,有兴趣的可以研究一下