ソースを参照

아주 사소한 것 추가

Surplus_Up (2DU) 7 年 前
コミット
33ef6e9468
2 ファイル変更15 行追加9 行削除
  1. 4 3
      app.py
  2. 11 6
      route/search_goto.py

+ 4 - 3
app.py

@@ -389,11 +389,12 @@ def search():
     return search_2(conn)
 
 @app.route('/goto', methods=['POST'])
-def search_goto():
-    return search_goto_2(conn)
+@app.route('/goto/<everything:name>', methods=['POST'])
+def search_goto(name = 'test'):
+    return search_goto_2(conn, name)
 
 @app.route('/search/<everything:name>')
-def search_deep(name = ''):
+def search_deep(name = 'test'):
     return search_deep_2(conn, name)
          
 @app.route('/raw/<everything:name>')

+ 11 - 6
route/search_goto.py

@@ -1,11 +1,16 @@
 from .tool.func import *
 
-def search_goto_2(conn):
+def search_goto_2(conn, name):
     curs = conn.cursor()
 
-    curs.execute("select title from data where title = ?", [flask.request.form.get('search', 'test')])
-    data = curs.fetchall()
-    if data:
-        return redirect('/w/' + url_pas(flask.request.form.get('search', 'test')))
+    if flask.request.form.get('search', None):
+        data = flask.request.form.get('search', 'test')
     else:
-        return redirect('/search/' + url_pas(flask.request.form.get('search', 'test')))
+        data = name
+
+    curs.execute("select title from data where title = ?", [data])
+    t_data = curs.fetchall()
+    if t_data:
+        return redirect('/w/' + url_pas(data))
+    else:
+        return redirect('/search/' + url_pas(data))