首先安装flask_mongoengine库:
pip install flask_mongoengine
使用Flask框架运行程序
注意:生成的集合名称是类名的小写字母
示例代码:
import datetime
from bson.json_util import dumps
from flask import Flask, Response, request
from flask_mongoengine import MongoEngine
app = Flask(__name__)
db = MongoEngine()
class Movie(db.Document):
name = db.StringField(required=True, unique=True)
casts = db.ListField(db.StringField(), required=True)
genres = db.ListField(db.StringField(), required=True)
created_at = db.DateTimeField(default=datetime.datetime.utcnow)
@app.route('/movies')
def get_movies():
movies = Movie.objects()
movies_list = []
for movie in movies:
movie_dict = movie.to_mongo().to_dict()
movie_dict['created_at'] = movie.created_at.isoformat()
movies_list.append(movie_dict)
movies_json = dumps(movies_list)
return Response(movies_json, mimetype="application/json", status=200)
@app.route('/movies', methods=['POST'])
def add_movie():
body = request.get_json()
movie = Movie(**body).save()
id = movie.id
return {'id': str(id)}, 200
if __name__ == '__main__':
app.config['MONGODB_SETTINGS'] = {
'host': 'mongodb://root:88888888@192.168.124.49:27017/test-movie?authSource=admin&retryWrites=true&w=majority'
}
db.init_app(app)
app.run()
程序运行后,使用postman工具多次发起post请求往数据库写数据:
然后在网页上执行get请求:
查看数据库数据生成情况:
上述程序启动后,使用脚本生成大量的数据:
import requests
import json
url = "http://127.0.0.1:5000/movies"
headers = {
'Content-Type': 'application/json'
}
for i in range(5, 100):
payload = json.dumps({
"name": f"dgw{i}",
"casts": [
f"aaa{i}",
f"bbb{i}"
],
"genres": [
f"ccc{i}",
f"ddd{i}"
]
})
response = requests.request("POST", url, headers=headers, data=payload)
print(response.text)
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由半码博客整理,本文链接:https://www.bmabk.com/index.php/post/142854.html