board_parser/app.py at master · nickpasko/board_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
import uuid

from flask import Flask, request
from flask import Response
from flask import session
from flask_restful import Api

from werkzeug.utils import secure_filename
from werkzeug.exceptions import BadRequest

from parser import read_geometries, read_board

app = Flask(__name__)
api = Api(app)
app.secret_key = 'LJSHDF*O@Y$Ksw9f;32lirF*(Y&.#LI21r40389gy:>IH34'

storage_path='files/'

@app.route('/board/init', methods=['GET'])
def init_board():
    if 'board_ids' not in session or not session['board_ids']:
        print('not in session')
        session['board_ids'] = []
    board_id = str(uuid.uuid4())
    tmp_list = session['board_ids']
    tmp_list.append(board_id)
    session['board_ids'] = tmp_list
    # print(len(session))
    print(len(session['board_ids']))
    return "{'board_id':" + board_id + "}"


@app.route('/board/<string:board_id>/check', methods=['GET'])
def board_check(board_id):
    print('board_id: '+board_id)
    if 'board_ids' not in session or not session['board_ids']:
        return BadRequest('board id list failure')
    print(session['board_ids'])
    print(len(session['board_ids']))
    print('board_ids:')
    for x in session['board_ids']:
        print(x)
    if board_id not in session['board_ids']:
        return BadRequest('board_id not found')
    return "{'board_id':" + board_id + "}"

@app.route('/board/<string:board_id>/upload/board', methods=['POST'])
def upload_board_file(board_id):
    if 'board_ids' not in session:
        return BadRequest('board id list failure')
    if board_id not in session['board_ids']:
        return BadRequest('board_id not found')
    f = request.files['file']
    filename = secure_filename(f.filename)
    f.save(storage_path+filename)
    session[board_id] = (filename, '')

    return Response('board file uploaded successfully')


@app.route('/board/<string:board_id>/upload/components', methods=['POST'])
def upload_components_file(board_id):
    if 'board_ids' not in session:
        return BadRequest('board id list failure')
    if board_id not in session['board_ids']:
        return BadRequest('board_id not found')
    f = request.files['file']
    filename = secure_filename(f.filename)
    f.save(storage_path+filename)
    session[board_id] = (session[board_id][0], filename)

    return Response('components file uploaded successfully')


@app.route('/board/<string:board_id>/parse', methods=['GET'])
def parse_board(board_id):
    if 'board_ids' not in session \
            or board_id not in session['board_ids'] \
            or board_id not in session \
            or not session[board_id]:
        return BadRequest('board not initialized properly')
    if session[board_id][0] == '':
        return BadRequest('board file not uploaded')
    if session[board_id][1] == '':
        return BadRequest('components file not uploaded')

    components_file = storage_path+session[board_id][1]
    components_lines = [line.rstrip('\n') for line in open(components_file)]
    board_file = storage_path+session[board_id][0]
    board_lines = [line.rstrip('\n') for line in open(board_file)]

    component_geometries = read_geometries(components_lines)
    component_board = read_board(board_lines, component_geometries)
    return component_board.to_json()


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