-
Notifications
You must be signed in to change notification settings - Fork 9
/
app.py
99 lines (86 loc) · 2.5 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from flask import Flask, render_template, request, jsonify
from pymongo import MongoClient
from bson.objectid import ObjectId
from flask_cors import CORS
import yaml
app = Flask(__name__)
config = yaml.load(open('database.yaml'))
client = MongoClient(config['uri'])
# db = client.lin_flask
db = client['lin_flask']
CORS(app)
@app.route('/')
def index():
return render_template('home.html')
@app.route('/data', methods=['POST', 'GET'])
def data():
# POST a data to database
if request.method == 'POST':
body = request.json
name = body['name']
age = body['age']
# db.users.insert_one({
db['users'].insert_one({
"name": name,
"age": age
})
return jsonify({
'status': 'Data is posted to MongoDB!',
'name': name,
'age': age
})
# GET all data from database
if request.method == 'GET':
allData = db['users'].find()
dataJson = []
for data in allData:
id = data['_id']
name = data['name']
age = data['age']
dataDict = {
'id': str(id),
'name': name,
'age': age
}
dataJson.append(dataDict)
print(dataJson)
return jsonify(dataJson)
@app.route('/data/<string:id>', methods=['GET', 'DELETE', 'PUT'])
def onedata(id):
# GET a specific data by id
if request.method == 'GET':
data = db['users'].find_one({'_id': ObjectId(id)})
id = data['_id']
name = data['name']
age = data['age']
dataDict = {
'id': str(id),
'name': name,
'age': age
}
print(dataDict)
return jsonify(dataDict)
# DELETE a data
if request.method == 'DELETE':
db['users'].delete_many({'_id': ObjectId(id)})
print('\n # Deletion successful # \n')
return jsonify({'status': 'Data id: ' + id + ' is deleted!'})
# UPDATE a data by id
if request.method == 'PUT':
body = request.json
name = body['name']
age = body['age']
db['users'].update_one(
{'_id': ObjectId(id)},
{
"$set": {
"name":name,
"age":age
}
}
)
print('\n # Update successful # \n')
return jsonify({'status': 'Data id: ' + id + ' is updated!'})
if __name__ == '__main__':
app.debug = True
app.run()