Surplus_Up (2DU) 7 лет назад
Родитель
Сommit
0eeea1c736
1 измененных файлов с 10 добавлено и 4 удалено
  1. 10 4
      app.py

+ 10 - 4
app.py

@@ -3010,13 +3010,16 @@ def register():
 
 
             return redirect('/need_email')
             return redirect('/need_email')
         else:
         else:
+            curs.execute('select data from other where name = "encode"')
+            db_data = curs.fetchall()
+
             curs.execute("select id from user limit 1")
             curs.execute("select id from user limit 1")
             if not curs.fetchall():
             if not curs.fetchall():
-                curs.execute("insert into user (id, pw, acl, date) values (?, ?, 'owner', ?)", [flask.request.form.get('id', None), hashed, get_time()])
+                curs.execute("insert into user (id, pw, acl, date, encode) values (?, ?, 'owner', ?, ?)", [flask.request.form.get('id', None), hashed, get_time(), db_data[0][0]])
 
 
                 first = 1
                 first = 1
             else:
             else:
-                curs.execute("insert into user (id, pw, acl, date) values (?, ?, 'user', ?)", [flask.request.form.get('id', None), hashed, get_time()])
+                curs.execute("insert into user (id, pw, acl, date, encode) values (?, ?, 'user', ?, ?)", [flask.request.form.get('id', None), hashed, get_time(), db_data[0][0]])
 
 
                 first = 0
                 first = 0
 
 
@@ -3133,13 +3136,16 @@ def check_key(tool = 'check_pass_key'):
     if flask.request.method == 'POST':
     if flask.request.method == 'POST':
         if tool == 'check_key':
         if tool == 'check_key':
             if 'c_id' in flask.session and flask.session['c_key'] == flask.request.form.get('key', None):
             if 'c_id' in flask.session and flask.session['c_key'] == flask.request.form.get('key', None):
+                curs.execute('select data from other where name = "encode"')
+                db_data = curs.fetchall()
+
                 curs.execute("select id from user limit 1")
                 curs.execute("select id from user limit 1")
                 if not curs.fetchall():
                 if not curs.fetchall():
-                    curs.execute("insert into user (id, pw, acl, date) values (?, ?, 'owner', ?)", [flask.session['c_id'], flask.session['c_pw'], get_time()])
+                    curs.execute("insert into user (id, pw, acl, date, encode) values (?, ?, 'owner', ?)", [flask.session['c_id'], flask.session['c_pw'], get_time(), db_data[0][0]])
 
 
                     first = 1
                     first = 1
                 else:
                 else:
-                    curs.execute("insert into user (id, pw, acl, date) values (?, ?, 'user', ?)", [flask.session['c_id'], flask.session['c_pw'], get_time()])
+                    curs.execute("insert into user (id, pw, acl, date, encode) values (?, ?, 'user', ?)", [flask.session['c_id'], flask.session['c_pw'], get_time(), db_data[0][0]])
 
 
                     first = 0
                     first = 0