diff --git a/src/webserver.py b/src/webserver.py index d1d60bd0fd45efbe49b0a594fe98e3ab3f403c5d..f449d6a2438c87328c2258fc0a17b61a1adf4cbe 100644 --- a/src/webserver.py +++ b/src/webserver.py @@ -18,6 +18,14 @@ csv_path = '../Studenten.csv' char_list = htr.char_list_from_file() chars = ''.join(char_list) word_chars = open('../model/wordCharList.txt').read().splitlines()[0] +matrikel_numbers = [] + +@app.route('/getNames', methods=['GET']) +def getNames(): + return jsonify(matrikel_numbers) + + + @app.route('/predictNachname', methods=['POST']) def predictNach(): @@ -67,6 +75,7 @@ def predictVor(): } return jsonify(result) + def split_Student_Names(): #csv looks like: Vorname;Nachname;Matrikelnummer #need to put Vorname in one list, Nachname in another @@ -74,7 +83,8 @@ def split_Student_Names(): #create /data/Vorname.txt and /data/Nachname.txt vorname_file = open('../data/Vorname.txt', 'w') nachname_file = open('../data/Nachname.txt', 'w') - + matrikelnummer_file = open('../data/Matrikelnummer.txt', 'w') + numbers=[] with open(csv_path, 'r') as csv_file: lines = csv_file.readlines() for line in lines[1:]: @@ -85,11 +95,14 @@ def split_Student_Names(): print(vorname, nachname, matrikelnummer) vorname_file.write(vorname + " ") nachname_file.write(nachname + " ") + matrikelnummer_file.write(matrikelnummer + " ") + numbers.append(matrikelnummer) + return numbers if __name__ == '__main__': #split csv file into Vorname, Nachname and Matrikelnummer - split_Student_Names() + matrikel_numbers = split_Student_Names() app.run(debug=True,port=8000)