Forráskód Böngészése

자잘한 애들 분리

Surplus_Up (2DU) 7 éve
szülő
commit
71dfe6baee
6 módosított fájl, 34 hozzáadás és 11 törlés
  1. 4 10
      app.py
  2. 6 0
      route/error_404.py
  3. 0 1
      route/login.py
  4. 9 0
      route/logout.py
  5. 9 0
      route/main_file.py
  6. 6 0
      route/search.py

+ 4 - 10
app.py

@@ -364,7 +364,7 @@ def block_log(name = None, tool = None):
             
 @app.route('/search', methods=['POST'])
 def search():
-    return redirect('/search/' + url_pas(flask.request.form.get('search', 'test')))
+    return search_2(conn)
 
 @app.route('/goto', methods=['POST'])
 def goto():
@@ -471,10 +471,7 @@ def check_key(tool = 'check_pass_key'):
            
 @app.route('/logout')
 def logout():
-    flask.session['state'] = 0
-    flask.session.pop('id', None)
-
-    return redirect('/user')
+    return logout_2(conn)
     
 @app.route('/ban/<name>', methods=['POST', 'GET'])
 def user_ban(name = None):
@@ -565,14 +562,11 @@ def views(name = None):
 
 @app.route('/<data>')
 def main_file(data = None):
-    if re.search('\.txt$', data):
-        return flask.send_from_directory('./', data)
-    else:
-        return redirect('/w/' + url_pas(wiki_set(2)))
+    return main_file_2(conn, data)
 
 @app.errorhandler(404)
 def error_404(e):
-    return redirect('/w/' + url_pas(wiki_set(2)))
+    return error_404_2(conn)
 
 if __name__=="__main__":
     app.secret_key = rep_key

+ 6 - 0
route/error_404.py

@@ -0,0 +1,6 @@
+from .tool.func import *
+
+def error_404_2(conn):
+    curs = conn.cursor()
+
+    return redirect('/w/' + url_pas(wiki_set(2)))

+ 0 - 1
route/login.py

@@ -1,5 +1,4 @@
 from .tool.func import *
-from flask import request
 
 def login_2(conn):
     curs = conn.cursor()

+ 9 - 0
route/logout.py

@@ -0,0 +1,9 @@
+from .tool.func import *
+
+def logout_2(conn):
+    curs = conn.cursor()
+
+    flask.session['state'] = 0
+    flask.session.pop('id', None)
+
+    return redirect('/user')

+ 9 - 0
route/main_file.py

@@ -0,0 +1,9 @@
+from .tool.func import *
+
+def main_file_2(conn, data):
+    curs = conn.cursor()
+
+    if re.search('\.txt$', data):
+        return flask.send_from_directory('./', data)
+    else:
+        return redirect('/w/' + url_pas(wiki_set(2)))

+ 6 - 0
route/search.py

@@ -0,0 +1,6 @@
+from .tool.func import *
+
+def search_2(conn):
+    curs = conn.cursor()
+
+    return redirect('/search/' + url_pas(flask.request.form.get('search', 'test')))