You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.5 KiB
84 lines
2.5 KiB
from couchpotato.api import api_docs, api_docs_missing
|
|
from couchpotato.core.auth import requires_auth
|
|
from couchpotato.core.event import fireEvent
|
|
from couchpotato.core.helpers.request import getParams, jsonified
|
|
from couchpotato.core.helpers.variable import md5
|
|
from couchpotato.core.logger import CPLog
|
|
from couchpotato.environment import Env
|
|
from flask.app import Flask
|
|
from flask.blueprints import Blueprint
|
|
from flask.globals import request
|
|
from flask.helpers import url_for
|
|
from flask.templating import render_template
|
|
from sqlalchemy.engine import create_engine
|
|
from sqlalchemy.orm import scoped_session
|
|
from sqlalchemy.orm.session import sessionmaker
|
|
from werkzeug.utils import redirect
|
|
import os
|
|
import time
|
|
|
|
log = CPLog(__name__)
|
|
|
|
app = Flask(__name__, static_folder = 'nope')
|
|
web = Blueprint('web', __name__)
|
|
|
|
|
|
def get_session(engine = None):
|
|
return Env.getSession(engine)
|
|
|
|
def addView(route, func, static = False):
|
|
web.add_url_rule(route + ('' if static else '/'), endpoint = route if route else 'index', view_func = func)
|
|
|
|
""" Web view """
|
|
@web.route('/')
|
|
@requires_auth
|
|
def index():
|
|
return render_template('index.html', sep = os.sep, fireEvent = fireEvent, env = Env)
|
|
|
|
""" Api view """
|
|
@web.route('docs/')
|
|
@requires_auth
|
|
def apiDocs():
|
|
from couchpotato import app
|
|
routes = []
|
|
for route, x in sorted(app.view_functions.iteritems()):
|
|
if route[0:4] == 'api.':
|
|
routes += [route[4:].replace('::', '.')]
|
|
|
|
if api_docs.get(''):
|
|
del api_docs['']
|
|
del api_docs_missing['']
|
|
return render_template('api.html', fireEvent = fireEvent, routes = sorted(routes), api_docs = api_docs, api_docs_missing = sorted(api_docs_missing))
|
|
|
|
@web.route('getkey/')
|
|
def getApiKey():
|
|
|
|
api = None
|
|
params = getParams()
|
|
username = Env.setting('username')
|
|
password = Env.setting('password')
|
|
|
|
if (params.get('u') == md5(username) or not username) and (params.get('p') == password or not password):
|
|
api = Env.setting('api_key')
|
|
|
|
return jsonified({
|
|
'success': api is not None,
|
|
'api_key': api
|
|
})
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(error):
|
|
index_url = url_for('web.index')
|
|
url = request.path[len(index_url):]
|
|
|
|
if url[:3] != 'api':
|
|
if request.path != '/':
|
|
r = request.url.replace(request.path, index_url + '#' + url)
|
|
else:
|
|
r = '%s%s' % (request.url.rstrip('/'), index_url + '#' + url)
|
|
return redirect(r)
|
|
else:
|
|
if not Env.get('dev'):
|
|
time.sleep(0.1)
|
|
return 'Wrong API key used', 404
|
|
|
|
|