server.py 5.45 KB
Newer Older
1
2
from flask import Flask, render_template, request, redirect, url_for, session
import psycopg2
Vladislav Rykov's avatar
Vladislav Rykov committed
3
import bcrypt
4
import misc
Vladislav Rykov's avatar
Vladislav Rykov committed
5
import dao.user.user as ud
Vladislav Rykov's avatar
Vladislav Rykov committed
6
import dao.application.application as ad
7
import dao.device.device as dd
8

9
APP_KEY_LEN = 8
10
11

server = Flask(__name__, template_folder='templates/')
Vladislav Rykov's avatar
Vladislav Rykov committed
12

13
@server.route('/')
14
def index():
Vladislav Rykov's avatar
Vladislav Rykov committed
15
    if 'name' in session and len(session['name']) > 0:
Vladislav Rykov's avatar
Vladislav Rykov committed
16
17
        ah = ad.ApplicationDao()
        apps = ah.get_list(session['name'].encode('utf-8'))
Vladislav Rykov's avatar
Vladislav Rykov committed
18
19

        session.pop('appkey', None)
Vladislav Rykov's avatar
Vladislav Rykov committed
20
        print('apps: ', apps)
21
22
23
24
        if apps[0]:
            return render_template('index.html', apps=apps[1])
        else:
            return render_template('index.html', feedback=apps[1])
25
26
27
28



@server.route('/signup', methods=['GET', 'POST'])
29
30
31
32
33
def signup():
    if request.method == 'GET':
        return render_template('signup.html')
    else: 
        username = request.form['username']
Vladislav Rykov's avatar
Vladislav Rykov committed
34
        password = request.form['password'].encode('utf-8')
35
36
37
38
39

        if (username == '' or password == ''):
            feedback = 'Username or password fields cannot be empty'
            return render_template('signup.html', feedback=feedback)
        else:
Vladislav Rykov's avatar
Vladislav Rykov committed
40
41
42
43
            uh = ud.UserDao()
            res = uh.create(username, password)
            if (not res[0]):
                return render_template('signup.html', feedback=res[1])
Vladislav Rykov's avatar
Vladislav Rykov committed
44
45
            else:
                session['name'] = username
46
        
Vladislav Rykov's avatar
Vladislav Rykov committed
47
                return redirect(url_for('index'))
48

49
50
51


@server.route('/login', methods=['GET', 'POST'])
Vladislav Rykov's avatar
Vladislav Rykov committed
52
def login():
Vladislav Rykov's avatar
Vladislav Rykov committed
53
54
55
56
    if request.method == 'GET':
        return render_template('login.html')
    else: 
        username = request.form['username']
Vladislav Rykov's avatar
Vladislav Rykov committed
57
        password = request.form['password'].encode('utf-8')
Vladislav Rykov's avatar
Vladislav Rykov committed
58
59
60
61
62

        if (username == '' or password == ''):
            feedback = 'Username or password fields cannot be empty'
            return render_template('login.html', feedback=feedback)
        else:
Vladislav Rykov's avatar
Vladislav Rykov committed
63
64
65
66
            uh = ud.UserDao()
            res = uh.get(username, password)
            if (not res[0]):
                return render_template('login.html', feedback=msg[1])
Vladislav Rykov's avatar
Vladislav Rykov committed
67
68
            else:
                session['name'] = username
Vladislav Rykov's avatar
Vladislav Rykov committed
69
        
Vladislav Rykov's avatar
Vladislav Rykov committed
70
71
72
                return redirect(url_for('index'))


73
74

@server.route('/logout')
Vladislav Rykov's avatar
Vladislav Rykov committed
75
76
77
def logout():
    session.clear()
    return redirect(url_for('index'))
Vladislav Rykov's avatar
Vladislav Rykov committed
78
79


80
81
82
83
84
85
86
87
88

@server.route('/new-app')
def new_application():
    return render_template('new-app.html')



@server.route('/app', methods=['GET', 'POST'])
def app():
Vladislav Rykov's avatar
Vladislav Rykov committed
89
    ah = ad.ApplicationDao()
90
    if request.method == 'GET':
91
        dh = dd.DeviceDao()
Vladislav Rykov's avatar
Vladislav Rykov committed
92
93
            
        session['appkey'] = request.args.get('appkey')
Vladislav Rykov's avatar
Vladislav Rykov committed
94

Vladislav Rykov's avatar
Vladislav Rykov committed
95
        app = ah.get(session['appkey'])
96
        devs = dh.get_list(app[1][1])
Vladislav Rykov's avatar
Vladislav Rykov committed
97
        
98
99
        print('devs : ', devs)
        return render_template('app.html', app=app[1], devs=devs[1])
100
101
102
103
104
    else:
        if request.form['appname'] == '':
            error = 'Application name cannot be empty.'
            return render_template('new-app.html', feedback=error)
        else:
105
106
            appkey = misc.rand_str(APP_KEY_LEN)
            res = ah.create(request.form['appname'], appkey, session['name'], request.form['appdesc'])
Vladislav Rykov's avatar
Vladislav Rykov committed
107
            
Vladislav Rykov's avatar
Vladislav Rykov committed
108
109
110
            if not res[0]:
                return render_template('new-app.html', feedback=res[1])
            
111
112
113
114
115
116
117
            dh = dd.DeviceDao()
            res = dh.create_table(appkey)
            
            if not res[0]:
                ah.delete(appkey)
                return render_template('new-app.html', feedback=res[1])
            
Vladislav Rykov's avatar
Vladislav Rykov committed
118
            return redirect(url_for('index'))
119

120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
@server.route('/delete-app')
def delete_app():
    dh = dd.DeviceDao()
    devs = dh.get_list(session['appkey'])
    
    for dev in devs[1]:
        dh.delete_datatable(session['appkey'], dev[1])
    
    dh.delete_table(session['appkey'])
    
    ah = ad.ApplicationDao()
    res = ah.delete(session['appkey'])
    
    if not res[0]:
        return redirect(url_for('app'))
    else:
        return redirect(url_for('index'))
137

Vladislav Rykov's avatar
Vladislav Rykov committed
138
139
140
141
142
143
144
145
146
147
@server.route('/add-dev')
def new_dev():
    dh = dd.DeviceDao()
    dev_list = dh.get_list(session['appkey'])
    
    print('dev list : ', dev_list)

    if not dev_list[0]:
        return render_template('add-dev.html', feedback=dev_list[1])
    else:
Vladislav Rykov's avatar
Vladislav Rykov committed
148
        return render_template('add-dev.html', free_ids=misc.prep_id_range(dev_list[1]))
Vladislav Rykov's avatar
Vladislav Rykov committed
149
150
151
152
153
154
155
 


@server.route('/dev', methods=['GET', 'POST'])
def dev():
    dh = dd.DeviceDao()
    if request.method == 'GET':
156
        dev = dh.get(session['appkey'], request.args.get('id'))
Vladislav Rykov's avatar
Vladislav Rykov committed
157
158
        ltup = 'recently'

159
        return render_template('dev.html', dev=dev[1], appkey=session['appkey'], ltup=ltup)
Vladislav Rykov's avatar
Vladislav Rykov committed
160
161
    else:
        res = dh.create(request.form['devname'], request.form['devid'], session['appkey'], request.form['devdesc'])
162

Vladislav Rykov's avatar
Vladislav Rykov committed
163
164
165
        if not res[0]:
            return render_template('add-dev.html', feedback=res[1])
        else:
166
167
168
169
170
171
172
            res = dh.create_datatable(session['appkey'], request.form['devid'])
            
            if not res[0]:
                dh.delete(session['appkey'], request.form['devid'])
                return render_template('add-dev.html', feedback=res[1])
            else:
                return redirect(url_for('app'))
Vladislav Rykov's avatar
Vladislav Rykov committed
173
174


Vladislav Rykov's avatar
Vladislav Rykov committed
175
176
177
178
179
180
181
182
183
@server.route('/delete-dev')
def delete_dev():
    dh = dd.DeviceDao()
    dh.delete_datatable(session['appkey'], request.args.get('id'))
    res = dh.delete(session['appkey'], request.args.get('id'))

    return redirect(url_for('app', appkey=session['appkey']))


184
if __name__ == '__main__':
185
186
187
188
    server.secret_key = 'sdjfklsjf^$654sd^#sPH'
    server.run(debug = True, host='0.0.0.0')