80 lines
1.8 KiB
Python
Executable File
80 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
"""
|
|
HTTP server wrapper around weboob
|
|
"""
|
|
import logging
|
|
import os
|
|
|
|
from bottle import post, request, response, route, run
|
|
|
|
from cozyweboob import main as cozyweboob
|
|
from cozyweboob import WeboobProxy
|
|
from cozyweboob.tools.env import is_in_debug_mode
|
|
from cozyweboob.tools.jsonwriter import pretty_json
|
|
|
|
# Module specific logger
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@post("/fetch")
|
|
def fetch_view():
|
|
"""
|
|
Fetch from weboob modules.
|
|
"""
|
|
params = request.forms.get("params")
|
|
response.content_type = "application/json"
|
|
return pretty_json(cozyweboob(params))
|
|
|
|
|
|
@post("/download")
|
|
def download_view():
|
|
"""
|
|
Download from weboob modules.
|
|
"""
|
|
params = request.forms.get("params")
|
|
response.content_type = "application/json"
|
|
# TODO return pretty_json(proxy.download(params))
|
|
|
|
|
|
@route("/list")
|
|
def list_view():
|
|
"""
|
|
List all available weboob modules and their configuration options.
|
|
"""
|
|
proxy = WeboobProxy()
|
|
response.content_type = "application/json"
|
|
return pretty_json(proxy.list_modules())
|
|
|
|
|
|
def init():
|
|
"""
|
|
Init function
|
|
"""
|
|
# Debug only: Set logging level and format
|
|
if is_in_debug_mode():
|
|
logging.basicConfig(
|
|
format='%(levelname)s: %(message)s',
|
|
level=logging.INFO
|
|
)
|
|
# Ensure all modules are installed and up to date before starting the
|
|
# server
|
|
logger.info("Ensuring all modules are installed and up to date.")
|
|
proxy = WeboobProxy()
|
|
proxy.install_modules()
|
|
logger.info("Starting server.")
|
|
|
|
|
|
def main():
|
|
"""
|
|
Main function
|
|
"""
|
|
init()
|
|
# Get host to listen on
|
|
HOST = os.environ.get("COZYWEBOOB_HOST", "localhost")
|
|
PORT = os.environ.get("COZYWEBOOB_PORT", 8080)
|
|
run(host=HOST, port=PORT, debug=is_in_debug_mode())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|