diff options
| -rw-r--r-- | settings.py | 1 | ||||
| -rw-r--r-- | youtube/search.py | 3 |
2 files changed, 3 insertions, 1 deletions
diff --git a/settings.py b/settings.py index bd6710c..5407922 100644 --- a/settings.py +++ b/settings.py @@ -147,6 +147,7 @@ For security reasons, enabling this is not recommended.''', }), ]) +program_directory = os.path.dirname(os.path.realpath(__file__)) acceptable_targets = settings_info.keys() | {'enable_comments', 'enable_related_videos'} diff --git a/youtube/search.py b/youtube/search.py index 0f6bbc4..34df76f 100644 --- a/youtube/search.py +++ b/youtube/search.py @@ -8,6 +8,7 @@ import base64 import mimetypes from flask import request import flask +import os # Sort: 1 # Upload date: 2 @@ -106,6 +107,6 @@ def get_search_page(): @yt_app.route('/opensearch.xml') def get_search_engine_xml(): - with open("youtube/opensearch.xml", 'rb') as f: + with open(os.path.join(settings.program_directory, 'youtube/opensearch.xml'), 'rb') as f: content = f.read().replace(b'$port_number', str(settings.port_number).encode()) return flask.Response(content, mimetype='application/xml') |