-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
118 lines (85 loc) · 2.94 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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import flask
from flask import request, jsonify
import sqlite3
app = flask.Flask(__name__)
app.config["DEBUG"] = True
DATABASE = 'test.db'
def dict_factory(cursor, row):
d = {}
for idx, col in enumerate(cursor.description):
d[col[0]] = row[idx]
return d
@app.route('/', methods=['GET'])
def home():
return '''<h1> Bee Hack </h1>'''
@app.route('/api/v1/resources/devices/all', methods=['GET'])
def api_all():
conn = sqlite3.connect(DATABASE)
conn.row_factory = dict_factory
cur = conn.cursor()
all_devices = cur.execute('SELECT * FROM devices;').fetchall()
return jsonify(all_devices)
@app.route('/api/v1/resources/devices', methods=['GET'])
def api_filter():
query_parameters = request.args
id_dev = query_parameters.get('id')
ip = query_parameters.get('device_ip')
campus = query_parameters.get('campus')
division = query_parameters.get('division')
location = query_parameters.get('device_location')
query = "SELECT * FROM devices WHERE"
to_filter = []
if id_dev:
query += ' id=? AND'
to_filter.append(id_dev)
if ip:
query += ' device_ip=? AND'
to_filter.append(ip)
if campus:
query += ' campus=? AND'
to_filter.append(campus)
if division:
query += ' division=? AND'
to_filter.append(division)
if location:
query += ' device_location=? AND'
to_filter.append(location)
if not (id_dev or ip or campus or division or location):
pass
query = query[:-4] + ';'
conn = sqlite3.connect('test.db')
conn.row_factory = dict_factory
cur = conn.cursor()
results = cur.execute(query, to_filter).fetchall()
return jsonify(results)
@app.route('/api/v1/resources/devices/modify', methods=['POST', 'DELETE'])
def update_device():
if request.method == 'POST':
data = request.form
id_dev = data.get('id')
ip = data.get('device_ip')
campus = data.get('campus')
division = data.get('division')
location = data.get('device_location')
conn = sqlite3.connect(DATABASE)
print(id_dev, ip, campus, division, location)
device = [(id_dev, ip, campus, division, location)]
conn.executemany('insert into devices values (?, ?, ?, ?, ?)', device)
conn.commit()
return "ok"
elif request.method == 'DELETE':
data = request.form
ip = data.get('device_ip')
campus = data.get('campus')
division = data.get('division')
location = data.get('device_location')
conn = sqlite3.connect(DATABASE)
print(ip, campus, division, location)
device = [(ip, campus, division, location)]
conn.executemany('delete from devices where device_ip = ? and campus = ? and division = ? and device_location = ?', device)
conn.commit()
return "ok"
else:
return 'algo pasó'
if __name__ == '__main__':
app.run()