def __init__(self, options, config=None):
if not config:
config = StoqConfig()
self.settings = config.get_settings()
self.enable_production = False
self.config = config
self.remove_demo = False
self.has_installed_db = False
self.options = options
self.plugins = []
self.db_is_local = False
self.enable_online_services = True
self.auth_type = TRUST_AUTHENTICATION
if config.get('Database', 'enable_production') == 'True':
self.remove_demo = True
if self.remove_demo:
first_step = PluginStep(self)
else:
first_step = WelcomeStep(self)
BaseWizard.__init__(self, None, first_step, title=self.title)
self.get_toplevel().set_deletable(False)
self.next_button.grab_focus()
def _read_config(self,
options,
create=False,
register_station=True,
check_schema=True,
load_plugins=True):
from stoqlib.lib.configparser import StoqConfig
from stoq.lib.startup import setup
config = StoqConfig()
if options.load_config and options.filename:
config.load(options.filename)
else:
config.load_default()
if create:
config.create()
# FIXME: This should be removed and we should just import
# the global settings after fixing StoqConfig to
# only update the global settings.
self._db_settings = config.get_settings()
setup(config,
options,
register_station=register_station,
check_schema=check_schema,
load_plugins=load_plugins)
return config
def main(args):
if platform.system() == 'Windows':
_windows_fixes()
handler = StoqServerCmdHandler()
if not args:
handler.cmd_help()
return 1
cmd = args[0]
args = args[1:]
parser = get_option_parser()
handler.add_options(cmd, parser)
options, args = parser.parse_args(args)
config = StoqConfig()
filename = (options.filename
if options.load_config and options.filename else APP_CONF_FILE)
config.load(filename)
# FIXME: This is called only when register_station=True. Without
# this, db_settings would not be updated. We should fix it on Stoq
config.get_settings()
register_config(config)
global SENTRY_URL, raven_client
SENTRY_URL = config.get('Sentry', 'url') or SENTRY_URL
raven_client = raven.Client(SENTRY_URL, release=stoqserver.version_str)
setup_excepthook(SENTRY_URL)
return handler.run_cmd(cmd, options, *args)
def setUp(self):
super().setUp()
register_config(StoqConfig())
app = bootstrap_app()
app.testing = True
self.client = app.test_client()
def main(args):
# Do this as soon as possible so we can log any early traceback
setup_excepthook()
if platform.system() == 'Windows':
_windows_fixes()
handler = StoqServerCmdHandler()
if not args:
handler.cmd_help()
return 1
cmd = args[0]
args = args[1:]
parser = get_option_parser()
handler.add_options(cmd, parser)
options, args = parser.parse_args(args)
config = StoqConfig()
filename = (options.filename
if options.load_config and options.filename else
APP_CONF_FILE)
config.load(filename)
# FIXME: This is called only when register_station=True. Without
# this, db_settings would not be updated. We should fix it on Stoq
config.get_settings()
register_config(config)
return handler.run_cmd(cmd, options, *args)
def main(args):
handler = StoqServerCmdHandler()
if not args:
handler.cmd_help()
return 1
cmd = args[0]
args = args[1:]
parser = get_option_parser()
handler.add_options(cmd, parser)
options, args = parser.parse_args(args)
config = StoqConfig()
filename = (options.filename
if options.load_config and options.filename else APP_CONF_FILE)
config.load(filename)
# FIXME: This is called only when register_station=True. Without
# this, db_settings would not be updated. We should fix it on Stoq
config.get_settings()
register_config(config)
# FIXME: Maybe we should check_schema and load plugins here?
setup(config=config,
options=options,
register_station=False,
check_schema=False,
load_plugins=True)
handler.run_cmd(cmd, options, *args)
def _setup_test_environment(request):
plugin_config = _get_plugin_configs(request.config)
if plugin_config['skip_env_setup']:
return
stoq_config = StoqConfig()
stoq_config.load_default()
register_config(stoq_config)
quick = plugin_config['quick_mode'] or _to_falsy(
os.environ.get("STOQLIB_TEST_QUICK", None))
bootstrap_suite(
address=os.environ.get("STOQLIB_TEST_HOSTNAME"),
dbname=os.environ.get("STOQLIB_TEST_DBNAME"),
port=int(os.environ.get("STOQLIB_TEST_PORT") or 0),
username=os.environ.get("STOQLIB_TEST_USERNAME"),
password=os.environ.get("STOQLIB_TEST_PASSWORD"),
quick=quick,
)
manager = get_plugin_manager()
for plugin_name in plugin_config['extra_plugins']:
_register_plugin(plugin_name)
with stoqlib.api.new_store() as store:
manager.install_plugin(store, plugin_name)
manager.activate_plugin(plugin_name)
plugin_cls = plugin_config['plugin_cls']
if plugin_cls:
_install_plugin(plugin_cls)
def _load_configuration(self):
from stoqlib.lib.configparser import StoqConfig
log.debug('reading configuration')
self._config = StoqConfig()
if self._options.filename:
self._config.load(self._options.filename)
else:
self._config.load_default()
def setUp(self):
super().setUp()
from stoqntk.ntkui import NtkUI
register_config(StoqConfig())
self.plugin = NtkUI()
app = bootstrap_app()
app.testing = True
self.client = app.test_client()
def main(args):
parser = get_option_parser()
group = optparse.OptionGroup(parser, 'Daemon')
group.add_option('',
'--daemon-id',
action="store",
dest="daemon_id",
help='Daemon Identifier')
parser.add_option_group(group)
options, args = parser.parse_args(args)
if not options.daemon_id:
raise SystemExit("Need a daemon id")
from stoqlib.lib.message import error
from stoqlib.lib.configparser import StoqConfig
log.debug('reading configuration')
config = StoqConfig()
if options.filename:
config.load(options.filename)
else:
config.load_default()
settings = config.get_settings()
try:
store_dsn = settings.get_store_dsn()
except:
type, value, trace = sys.exc_info()
error(
_("Could not open the database config file"),
_("Invalid config file settings, got error '%s', "
"of type '%s'") % (value, type))
from stoqlib.exceptions import StoqlibError
from stoqlib.database.exceptions import PostgreSQLError
from stoq.lib.startup import setup
log.debug('calling setup()')
# XXX: progress dialog for connecting (if it takes more than
# 2 seconds) or creating the database
try:
setup(config, options, register_station=True, check_schema=False)
except (StoqlibError, PostgreSQLError) as e:
error(_('Could not connect to the database'),
'error=%s dsn=%s' % (str(e), store_dsn))
raise SystemExit("Error: bad connection settings provided")
daemon = Daemon(options.daemon_id)
daemon.run()
def setup_sentry(options):
config = StoqConfig()
filename = (options.filename
if options.load_config and options.filename else APP_CONF_FILE)
config.load(filename)
# FIXME: This is called only when register_station=True. Without
# this, db_settings would not be updated. We should fix it on Stoq
config.get_settings()
register_config(config)
global SENTRY_URL, raven_client
SENTRY_URL = config.get('Sentry', 'url') or SENTRY_URL
raven_client = raven.Client(SENTRY_URL,
release=stoqserver.version_str,
transport=SilentTransport)
setup_excepthook()
def __init__(self, options, config=None):
if not config:
config = StoqConfig()
self.settings = config.get_settings()
if is_windows:
self.settings.username = WINDOWS_DEFAULT_USER
self.settings.password = WINDOWS_DEFAULT_PASSWORD
self.settings.dbname = WINDOWS_DEFAULT_DBNAME
self.settings.address = "localhost"
self.settings.port = 5432
self.link_request_done = False
self.create_examples = False
self.config = config
self.enable_production = False
self.has_installed_db = False
self.options = options
self.db_is_local = False
self.enable_online_services = True
self.auth_type = TRUST_AUTHENTICATION
if config.get('Database', 'enable_production') == 'True':
self.enable_production = True
if self.enable_production:
first_step = CreateDatabaseStep(self)
elif is_windows and self.try_connect(self.settings, warn=False):
self.auth_type = PASSWORD_AUTHENTICATION
self.write_pgpass()
first_step = self.connect_for_settings()
else:
first_step = DatabaseLocationStep(self)
BaseWizard.__init__(self, None, first_step, title=self.title)
self.get_toplevel().set_deletable(False)
self.next_button.grab_focus()
def setup(config=None, options=None, register_station=True, check_schema=True,
load_plugins=True):
"""
Loads the configuration from arguments and configuration file.
@param config: a StoqConfig instance
@param options: a Optionparser instance
@param register_station: if we should register the branch station.
@param check_schema: if we should check the schema
@param load_plugins: if we should load plugins for the system
"""
# NOTE: No GUI calls are allowed in here
# If you change anything here, you need to verify that all
# callsites are still working properly.
# bin/stoq
# bin/stoqdbadmin
# python stoq/tests/runtest.py
if options is None:
parser = get_option_parser()
options, args = parser.parse_args(sys.argv)
if options.verbose:
from kiwi.log import set_log_level
set_log_level('stoq*', 0)
setup_path()
if config is None:
config = StoqConfig()
if options.filename:
config.load(options.filename)
else:
config.load_default()
config.set_from_options(options)
register_config(config)
if options and options.sqldebug:
enable_debugging()
from stoq.lib.applist import ApplicationDescriptions
provide_utility(IApplicationDescriptions, ApplicationDescriptions(),
replace=True)
db_settings = config.get_settings()
try:
default_store = get_default_store()
except DatabaseError as e:
# Only raise an error if a database is actually required
if register_station or load_plugins or check_schema:
error(e.short, str(e.msg))
else:
default_store = None
if register_station:
db_settings.check_version(default_store)
if check_schema:
migration = StoqlibSchemaMigration()
migration.check()
if options and options.sqldebug:
enable_debugging()
set_current_branch_station(default_store, station_name=None)
if load_plugins:
from stoqlib.lib.pluginmanager import get_plugin_manager
manager = get_plugin_manager()
manager.activate_installed_plugins()
if check_schema:
if not default_store.table_exists('system_table'):
error(
_("Database schema error"),
_("Table 'system_table' does not exist.\n"
"Consult your database administrator to solve this problem."))
if check_schema:
migration = StoqlibSchemaMigration()
migration.check()