Skip to content

Commit c8413ac

Browse files
authored
Merge pull request #45 from simonseo/flaskBranch
Created /getUsers api which provides list of all userids
2 parents fe0291c + 4768e39 commit c8413ac

File tree

2 files changed

+25
-0
lines changed

2 files changed

+25
-0
lines changed

flask/flaskapp.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from readCSV import ReadCSV
55
from dbsetup import DBsetup
66
from getUserSubs import getUserSubs
7+
from getUsers import getUsers
78
import databaseOperations as db
89
import sqlite3
910
import os
@@ -96,6 +97,10 @@ def unSubAll():
9697
data = json.loads(request.data.decode("utf-8"))
9798
return db.unSubAll(data["userid"])
9899

100+
@app.route("/getUsers", methods=['POST','GET'])
101+
def getListOfUsers():
102+
return getUsers()
103+
99104
if __name__ == "__main__":
100105
DBsetup()
101106
app.run()#host=HOSTADDRESS, port=80)

flask/getUsers.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
import csv
2+
import requests, json
3+
import sqlite3
4+
import databaseOperations as db
5+
from flask import jsonify
6+
7+
8+
def getUsers():
9+
conn = sqlite3.connect('posts.db')
10+
c = conn.cursor()
11+
to_send={"users":[]}
12+
with conn:
13+
c.execute("SELECT userID from users")
14+
users_selected=c.fetchall()
15+
for user in users_selected:
16+
to_send["users"].append(user[0])
17+
18+
conn.commit()
19+
conn.close()
20+
return jsonify(to_send)

0 commit comments

Comments
 (0)