mirror of
https://github.com/apache/superset.git
synced 2024-09-12 08:39:45 -04:00
131 lines
3.8 KiB
Python
Executable File
131 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
from subprocess import Popen
|
|
|
|
from flask_migrate import MigrateCommand
|
|
from flask_script import Manager
|
|
|
|
import caravel
|
|
from caravel import app, ascii_art, db, data, utils
|
|
|
|
config = app.config
|
|
|
|
manager = Manager(app)
|
|
manager.add_command('db', MigrateCommand)
|
|
|
|
|
|
@manager.option(
|
|
'-d', '--debug', action='store_true',
|
|
help="Start the web server in debug mode")
|
|
@manager.option(
|
|
'-a', '--address', default=config.get("CARAVEL_WEBSERVER_ADDRESS"),
|
|
help="Specify the address to which to bind the web server")
|
|
@manager.option(
|
|
'-p', '--port', default=config.get("CARAVEL_WEBSERVER_PORT"),
|
|
help="Specify the port on which to run the web server")
|
|
@manager.option(
|
|
'-w', '--workers', default=config.get("CARAVEL_WORKERS", 16),
|
|
help="Number of gunicorn web server workers to fire up")
|
|
@manager.option(
|
|
'-t', '--timeout', default=config.get("CARAVEL_WEBSERVER_TIMEOUT"),
|
|
help="Specify the timeout (seconds) for the gunicorn web server")
|
|
def runserver(debug, address, port, timeout, workers):
|
|
"""Starts a Caravel web server"""
|
|
debug = debug or config.get("DEBUG")
|
|
if debug:
|
|
app.run(
|
|
host='0.0.0.0',
|
|
port=int(port),
|
|
debug=True)
|
|
else:
|
|
cmd = (
|
|
"gunicorn "
|
|
"-w {workers} "
|
|
"--timeout {timeout} "
|
|
"-b {address}:{port} "
|
|
"--limit-request-line 0 "
|
|
"--limit-request-field_size 0 "
|
|
"caravel:app").format(**locals())
|
|
print("Starting server with command: " + cmd)
|
|
Popen(cmd, shell=True).wait()
|
|
|
|
@manager.command
|
|
def init():
|
|
"""Inits the Caravel application"""
|
|
utils.init(caravel)
|
|
|
|
@manager.option(
|
|
'-v', '--verbose', action='store_true',
|
|
help="Show extra information")
|
|
def version(verbose):
|
|
"""Prints the current version number"""
|
|
s = (
|
|
"\n{boat}\n\n"
|
|
"-----------------------\n"
|
|
"Caravel {version}\n"
|
|
"-----------------------").format(
|
|
boat=ascii_art.boat, version=caravel.VERSION)
|
|
print(s)
|
|
if verbose:
|
|
print("[DB] : " + "{}".format(db.engine))
|
|
|
|
@manager.option(
|
|
'-t', '--load-test-data', action='store_true',
|
|
help="Load additional test data")
|
|
def load_examples(load_test_data):
|
|
"""Loads a set of Slices and Dashboards and a supporting dataset """
|
|
print("Loading examples into {}".format(db))
|
|
|
|
data.load_css_templates()
|
|
|
|
print("Loading energy related dataset")
|
|
data.load_energy()
|
|
|
|
print("Loading [World Bank's Health Nutrition and Population Stats]")
|
|
data.load_world_bank_health_n_pop()
|
|
|
|
print("Loading [Birth names]")
|
|
data.load_birth_names()
|
|
|
|
print("Loading [Random time series data]")
|
|
data.load_random_time_series_data()
|
|
|
|
print("Loading [Random long/lat data]")
|
|
data.load_long_lat_data()
|
|
|
|
print("Loading [Multiformat time series]")
|
|
data.load_multiformat_time_series_data()
|
|
|
|
if load_test_data:
|
|
print("Loading [Unicode test data]")
|
|
data.load_unicode_test_data()
|
|
|
|
@manager.command
|
|
def refresh_druid():
|
|
"""Refresh all druid datasources"""
|
|
session = db.session()
|
|
from caravel import models
|
|
for cluster in session.query(models.DruidCluster).all():
|
|
try:
|
|
cluster.refresh_datasources()
|
|
except Exception as e:
|
|
print(
|
|
"Error while processing cluster '{}'\n{}".format(
|
|
cluster, str(e)))
|
|
logging.exception(e)
|
|
cluster.metadata_last_refreshed = datetime.now()
|
|
print(
|
|
"Refreshed metadata from cluster "
|
|
"[" + cluster.cluster_name + "]")
|
|
session.commit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
manager.run()
|