linkkf 로직수정중

lib/framework/init_route.py

@@ -4,7 +4,6 @@ import traceback
 from flask import (jsonify, redirect, render_template, request,
                    send_from_directory)
 from flask_login import login_required
-
 from framework import F
 
 
@@ -86,27 +85,31 @@ def open_file(path):
 @F.app.route("/file/<path:path>")
 @F.check_api
 def file2(path):
-    # 윈도우 drive 필요 없음
+    import platform
+    if platform.system() == 'Windows':
+        path = os.path.splitdrive(path)[1][1:]
     return send_from_directory('/', path, as_attachment=True)
 
 
 @F.app.route("/upload", methods=['GET', 'POST'])
+@login_required
 def upload():
     try:
         if request.method == 'POST':
             f = request.files['file']
-            from werkzeug import secure_filename
+            from werkzeug.utils import secure_filename
             upload_path = F.SystemModelSetting.get('path_upload')
             os.makedirs(upload_path, exist_ok=True)
             f.save(os.path.join(upload_path, secure_filename(f.filename)))
             return jsonify('success')
-    except Exception as exception:
-        F.logger.error('Exception:%s', exception)
+    except Exception as e:
+        F.logger.error(f"Exception:{str(e)}")
         F.logger.error(traceback.format_exc())
         return jsonify('fail')
 
 
 @F.app.route("/videojs", methods=['GET', 'POST'])
+@login_required
 def videojs():
     data = {}
     data['play_title'] = request.form['play_title']
@@ -116,9 +119,33 @@ def videojs():
         data['play_subtitle_src'] = request.form['play_subtitle_src']
     return render_template('videojs.html', data=data)
 
+@F.app.route("/videojs_drm", methods=['GET', 'POST'])
+@login_required
+def videojs_drm():
+    data = {}
+    data['play_title'] = request.form['play_title']
+    data['play_source_src'] = request.form['play_source_src']
+    data['play_source_type'] = request.form['play_source_type']
+    if 'play_subtitle_src' in request.form:
+        data['play_subtitle_src'] = request.form['play_subtitle_src']
+    return render_template('videojs_drm.html', data=data)
+
+@F.app.route("/videojs_discord", methods=['GET', 'POST'])
+@login_required
+def videojs_og():
+    data = {}
+    """
+    data['play_title'] = request.form['play_title']
+    data['play_source_src'] = request.form['play_source_src']
+    data['play_source_type'] = request.form['play_source_type']
+    if 'play_subtitle_src' in request.form:
+        data['play_subtitle_src'] = request.form['play_subtitle_src']
+    """
+    return render_template('videojs_discord.html', data=data)
 
 
 @F.app.route("/headers", methods=['GET', 'POST'])
+@login_required
 def headers():
     from support import d
     F.logger.info(d(request.headers))
@@ -127,6 +154,7 @@ def headers():
 
 # 3.10에서 이거 필수
 @F.socketio.on('connect', namespace=f'/framework')
+@login_required
 def connect():
     pass