-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
70 lines (53 loc) · 2.35 KB
/
main.py
File metadata and controls
70 lines (53 loc) · 2.35 KB
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
from flask import Flask
from flask_restful import Resource, Api, reqparse
import pandas as pd
import ast
import json
app = Flask(__name__)
api = Api(app)
class Users(Resource):
def get(self):
data = pd.read_csv('policies.csv')
data = data.to_dict() # convert dataframe to dictionary
return {'data': data}, 200 # return data and 200 OK code
def post(selfs):
parser = reqparse.RequestParser()
parser.add_argument('user_id', required=True, type=str)
parser.add_argument('policy_num', required=True, type=str)
parser.add_argument('name', required=True, type=str)
parser.add_argument('surname', required=True, type=str)
parser.add_argument('dob', required=True, type=str)
parser.add_argument('sex', required=True, type=str)
parser.add_argument('country', required=True, type=str)
parser.add_argument('start_date', required=True, type=str)
parser.add_argument('finish_date', required=True, type=str)
parser.add_argument('travel_area', required=True, type=str)
args = parser.parse_args()
data = pd.read_csv('policies.csv')
if args['user_id'] in data['user_id']:
return {'message': 'user already exists', 409}
else:
data = data.append({
'user_id': args['user_id'],'policy_num': args['policy_num']
'name': args['name'], 'surname': args['surname'], 'dob': args['dob'],
'sex': args['sex'], 'country': args['country'], 'start_date': args['start_date'],
'finish_date': args['finish_date'], 'travel_area': args['travel_area']},
ignore_index=True
)
return {'data' : data.to_dict()}, 200
def delete(self):
parser = reqparse.RequestParser()
parser.add_argument('user_id', required = True, type=int)
args = parser.parse_args()
data = pd.read_csv('policies.csv')
if args['user_id'] in data['user_id']:
data = data[data['user_id']] != args['user_id']
data.to_csv('user_path', index=False)
return {'data': data.to_ct()}, 200
else:
return {
'message': 'entered Id does not exist'
}, 404
api.add_resource(Users, '/users') # '/users' is our entry point for Users
if __name__ == '__main__':
app.run()