diff --git a/bdd/Dockerfiles/app/app/app.py b/bdd/Dockerfiles/app/app/app.py index 362185a..e2b0606 100755 --- a/bdd/Dockerfiles/app/app/app.py +++ b/bdd/Dockerfiles/app/app/app.py @@ -2,7 +2,8 @@ from flask import Flask import pandas as pd -import pysqlite3 as sql +import sqlite3 as sql +import json def create_app(): @@ -11,11 +12,32 @@ def create_app(): data=pd.read_csv("/Datasets/database/questions.csv",delimiter=",") data.to_sql('questions',con,if_exists='replace',index=False) - @app.route('/question/',methods=['GET']) + @app.route('/get_question//',methods=['GET']) def get_question(id=1): with sql.connect("/Datasets/database/database.db") as con: cur = con.cursor() row = cur.execute("select * from questions where id=\'"+str(id)+"\';") return(row.fetchone()[2]) + @app.route('/create//') + def create(token='TEST'): + with sql.connect("/Datasets/database/database.db") as con: + data=pd.read_csv("/Datasets/database/"+token+".csv",delimiter=";") + data.to_sql(token,con,if_exists='replace',index=False) + return '1' + + @app.route('/update////') + def update(token,attr,value): + with sql.connect("/Datasets/database/database.db") as con: + cur = con.cursor() + sql_update= "update " + token + " set excluded = 1 where "+attr+"="+str(value)+";" + cur.execute(sql_update) + cur.execute("select x,y from "+token+" where "+attr+"=\'"+str(value)+"\';") + rows = cur.fetchall() + dico = dict() + dico["img"]=rows + cur = con.cursor() + cur.execute("SELECT COUNT(*) FROM "+token+" where excluded = 0") + dico["Nb"]=cur.fetchone()[0] + return(json.dumps(dico)) return app