def hazard_restore_local(*argv): """ Use the current django settings to restore hazard """ from openquake.engine.db.models import set_django_settings_module set_django_settings_module() from django.conf import settings default_cfg = settings.DATABASES['default'] host = default_cfg['HOST'] or 'localhost' name = default_cfg['NAME'] user = default_cfg['USER'] pwd = default_cfg['PASSWORD'] port = str(default_cfg['PORT'] or 5432) def h(dflt): return 'default: %s' % dflt p = argparse.ArgumentParser() p.add_argument('tarfile', help='mandatory argument') p.add_argument('host', nargs='?', default=host, help=h(host)) p.add_argument('dbname', nargs='?', default=name, help=h(name)) p.add_argument('user', nargs='?', default=user, help=h(user)) p.add_argument('password', nargs='?', default=pwd, help=h(pwd)) p.add_argument('port', nargs='?', default=port, help=h(port)) arg = p.parse_args(argv) hazard_restore_remote(arg.tarfile, arg.host, arg.dbname, arg.user, arg.password, arg.port)
def hazard_restore_local(*argv): """ Use the current django settings to restore hazard """ from openquake.engine.db.models import set_django_settings_module set_django_settings_module() from django.conf import settings default_cfg = settings.DATABASES['default'] host = default_cfg['HOST'] or 'localhost' name = default_cfg['NAME'] user = default_cfg['USER'] pwd = default_cfg['PASSWORD'] port = str(default_cfg['PORT'] or 5432) def h(dflt): return 'default: %s' % dflt p = argparse.ArgumentParser() p.add_argument('directory', help='mandatory argument') p.add_argument('host', nargs='?', default=host, help=h(host)) p.add_argument('dbname', nargs='?', default=name, help=h(name)) p.add_argument('user', nargs='?', default=user, help=h(user)) p.add_argument('password', nargs='?', default=pwd, help=h(pwd)) p.add_argument('port', nargs='?', default=port, help=h(port)) arg = p.parse_args(argv) hazard_restore_remote(arg.directory, arg.host, arg.dbname, arg.user, arg.password, arg.port)
def hazard_restore_local(tar): """ Use the current django settings to restore hazard """ from openquake.engine.db.models import set_django_settings_module set_django_settings_module() from django.conf import settings default_cfg = settings.DATABASES['default'] hazard_restore_remote( tar, default_cfg['HOST'], default_cfg['NAME'], default_cfg['USER'], default_cfg['PASSWORD'], # avoid passing an empty string to psycopg default_cfg['PORT'] or None)
config.abort_if_no_config_available() try: import celeryconfig except ImportError: sys.path.append('/usr/openquake/engine') try: import oqpath oqpath.set_oq_path() except ImportError: pass from openquake.engine.db.models import set_django_settings_module set_django_settings_module() import openquake.engine from openquake.engine import __version__ from openquake.engine import engine from openquake.engine.db import models from openquake.engine.export import hazard as hazard_export from openquake.engine.export import risk as risk_export from openquake.engine.input import source from openquake.engine.tools.import_gmf_scenario import import_gmf_scenario from openquake.engine.tools.import_hazard_curves import import_hazard_curves HAZARD_OUTPUT_ARG = "--hazard-output-id" HAZARD_CALCULATION_ARG = "--hazard-calculation-id" MISSING_HAZARD_MSG = ("Please specify the ID of the hazard output (or "