JIIT-time-table-parser/server.py at main · tashifkhan/JIIT-time-table-parser · GitHub

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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
from flask import Flask, render_template
import os
import json
from flask import request
from werkzeug.utils import secure_filename

# conversion modules
from modules.xlsx_handeler import xlsx_to_csv_string
from modules.xls_handeler import xls_to_csv_string
from modules.pdf_handeler import pdf_to_csv_string

# processing modules
from modules.csv_handeler import csvstring_to_jsonstrings
from modules.BE62_creator import time_table_creator

# state management
global_state = {
    "time_table": None,
    "subjects_dict": None,
    "file_path": None
}

''' pipeline
from modules.xls_handeler import xls_to_csv_string
from modules.csv_handeler import csvstring_to_jsonstrings
from modules._creator import time_table_creator
from modules.xlsx_handeler import xlsx_to_csv_string
import json

def test_xls_to_csv_string():
    with open('data/xls/B Tech I Sem Odd 2024 Aug 8.xlsx', 'rb') as file:
        # csv_string = xls_to_csv_string(file)
        csv_string = xlsx_to_csv_string(file)
        print(csv_string)
        jsondict = csvstring_to_jsonstrings(csv_string)
        print(json.dumps(jsondict, indent=4))
        time_table = jsondict["timetable"]
        subjects_dict = jsondict["subjects"]
        # print(json.dumps(time_table_creator(time_table, subjects_dict, "A6", ["MA533", "NEC735", "HS315"]), indent=4))
        print(json.dumps(time_table_creator(time_table, subjects_dict, "A6", []), indent=4))

def test_xlsx_to_csv_string():
    with open('./data/xls/B. Tech V Sem ODD 2024 Aug 7.xls', 'rb') as file:
        csv_string = xls_to_csv_string(file)
        print(csv_string)
        jsondict = csvstring_to_jsonstrings(csv_string)
        print(json.dumps(jsondict, indent=4))
        time_table = jsondict["timetable"]
        subjects_dict = jsondict["subjects"]
        # print(json.dumps(time_table_creator(time_table, subjects_dict, "A6", ["MA533", "NEC735", "HS315"]), indent=4))
        print(json.dumps(time_table_creator(time_table, subjects_dict, "A6", []), indent=4))

if __name__ == "__main__":
    test_xls_to_csv_string()
    # test_xlsx_to_csv_string()
'''

app = Flask(__name__)

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/fileupload', methods=['GET','POST'])
def fileupload():
    # return render_template('fileupload.html')
    UPLOAD_FOLDER = 'uploads'
    if not os.path.exists(UPLOAD_FOLDER):
        os.makedirs(UPLOAD_FOLDER)

    if 'file' not in request.files:
        return 'No file uploaded', 400

    file = request.files['file']
    if file.filename == '':
        return 'No file selected', 400

    filename = secure_filename(file.filename)
    file.save(os.path.join(UPLOAD_FOLDER, filename))
    global_state["file_path"] = os.path.join(UPLOAD_FOLDER, filename)
    file_extension = os.path.splitext(filename)[1].lower()

    if file_extension == ".xls":
        csv_string = xls_to_csv_string(file)


    elif file_extension == ".xlsx":
        csv_string = xlsx_to_csv_string(file)

    elif file_extension == ".csv":
        csv_string = file.read()


    elif file_extension == ".pdf":
        csv_string = pdf_to_csv_string(file)

    else:
        return 'Invalid file format', 400

    jsondict = csvstring_to_jsonstrings(csv_string)
    time_table = jsondict["timetable"]
    subjects_dict = jsondict["subjects"]
    subjects_dict = sorted(subjects_dict, key=lambda x: x['Subject'].lower())
    for subject in subjects_dict:
        subject['Subject'] = subject['Subject'].title()

    global_state["time_table"] = time_table
    global_state["subjects_dict"] = subjects_dict

    return render_template('electives.html', time_table=time_table, subjects_dict=subjects_dict)

@app.route('/electives', methods=['POST'])
def electives_page():
    if not global_state['time_table'] or not global_state['subjects_dict']:
        return 'Please upload a file first', 400

    request_data = request.form
    request_data = dict(request_data)
    numumber_electives = int(request_data['elective_count'])
    electives = []
    for i in range(1, numumber_electives+1):
        elective = request_data[f'elective_{i}']
        electives.append(elective)
    result = time_table_creator(global_state['time_table'], global_state['subjects_dict'], request_data['batch'].capitalize(), electives)
    os.remove(global_state["file_path"])
    global_state["file_path"] = None
    global_state["time_table"] = None
    global_state["subjects_dict"] = None
    return render_template('schedule.html', your_schedule=result)

if __name__ == '__main__':
    app.run(debug=True)