def toriicli(ctx, project_path):
"""CLI utility for the Unity Torii library."""
config.setup_logging()
if ctx.invoked_subcommand not in SUBCOMMANDS_DONT_LOAD_CONFIG:
cfg = config.from_yaml(config.CONFIG_NAME)
if cfg is None:
raise SystemExit(1)
ctx.obj = ToriiCliContext(cfg, cfg.actual_project_dir or project_path)
def parse_command_line_args(for_server=False):
parser = get_cmd_line_args_parser(for_server)
cmdline_args = parser.parse_args()
config.setup_logging(cmdline_args)
if for_server:
# region COMPATIBILITY WITH CONFIG PARSER
cmdline_args.record_count = None
# endregion
return cmdline_args
def parse_command_line_args(for_server=False):
action_default = None
action_nargs = None
logging_setting_path = relative_app_path('logging.json')
actions = [
'upload', # action which pulls data from remote system and push to Oomnitza.
'generate-ini' # action which generates an example config.ini file.
]
parser = argparse.ArgumentParser()
if for_server:
parser.add_argument('--host', type=str, default='127.0.0.1')
parser.add_argument('--port', type=int, default=8000)
else:
parser.add_argument("action", default=action_default, nargs=action_nargs, choices=actions, help="Action to perform.")
parser.add_argument("connectors", nargs='*', default=[], help="Connectors to run.")
parser.add_argument('--record-count', type=int, default=None, help="Number of records to pull and process from connection.")
parser.add_argument('--workers', type=int, default=2, help="Number of async IO workers used to pull & push records.")
parser.add_argument('--version', action='store_true', help="Show the connector version.")
parser.add_argument('--show-mappings', action='store_true', help="Show the mappings which would be used by the connector.")
parser.add_argument('--testmode', action='store_true', help="Run connectors in test mode.")
parser.add_argument('--save-data', action='store_true', help="Saves the data loaded from other system.")
# parser.add_argument('--load-data', default="", help="Directory from which to load data.")
parser.add_argument('--ini', type=str, default=relative_app_path("config.ini"), help="Config file to use.")
parser.add_argument('--logging-config', type=str, default=logging_setting_path, help="Use to override logging config file to use.")
cmdline_args = parser.parse_args()
config.setup_logging(cmdline_args)
if for_server:
# region COMPATIBILITY WITH CONFIG PARSER
cmdline_args.record_count = None
# endregion
return cmdline_args
from flask.ext.sqlalchemy import get_debug_queries
from flask.ext.sqlalchemy import SQLAlchemy
from tabulate import tabulate
from ggrc import db
from ggrc import extensions
from ggrc import notifications
from ggrc import settings
from ggrc.gdrive import init_gdrive_routes
from ggrc.utils import benchmark
from ggrc.utils.issue_tracker_mock import init_issue_tracker_mock
if settings.ISSUE_TRACKER_MOCK and not settings.PRODUCTION:
init_issue_tracker_mock()
setup_logging(settings.LOGGING)
logger = getLogger(__name__)
app = Flask('ggrc', instance_relative_config=True) # noqa pylint: disable=invalid-name
app.config.from_object(settings)
if "public_config" not in app.config:
app.config.public_config = {}
for key in settings.exports:
app.config.public_config[key] = app.config[key]
# Configure Flask-SQLAlchemy for app
db.app = app
db.init_app(app)
import re
from logging import getLogger
from logging.config import dictConfig as setup_logging
from flask import Flask
from flask.ext.sqlalchemy import get_debug_queries
from flask.ext.sqlalchemy import SQLAlchemy
from tabulate import tabulate
from ggrc import db
from ggrc import extensions
from ggrc import notifications
from ggrc import settings
setup_logging(settings.LOGGING)
# pylint: disable=invalid-name
logger = getLogger(__name__)
app = Flask('ggrc', instance_relative_config=True) # noqa: valid constant name
app.config.from_object(settings)
if "public_config" not in app.config:
app.config.public_config = {}
for key in settings.exports:
app.config.public_config[key] = app.config[key]
# Configure Flask-SQLAlchemy for app
db.app = app
parser = argparse.ArgumentParser()
parser.add_argument("action", default=action_default, nargs=action_nargs, choices=actions, help="Action to perform.")
parser.add_argument("connectors", nargs='*', default=[], help="Connectors to run.")
parser.add_argument('--version', action='store_true', help="Show the connector version.")
parser.add_argument('--show-mappings', action='store_true', help="Show the mappings which would be used by the connector.")
parser.add_argument('--testmode', action='store_true', help="Run connectors in test mode.")
parser.add_argument('--save-data', action='store_true', help="Saves the data loaded from other system.")
# parser.add_argument('--load-data', default="", help="Directory from which to load data.")
parser.add_argument('--ini', type=str, default=relative_app_path("config.ini"), help="Config file to use.")
parser.add_argument('--logging-config', type=str, default=logging_setting_path, help="Use to override logging config file to use.")
parser.add_argument('--record-count', type=int, default=None, help="Number of records to pull and process from connection.")
args = parser.parse_args()
config.setup_logging(args)
LOG.info("Connector version: %s", version.VERSION)
if args.version:
exit()
if args.testmode:
LOG.info("Connector started in Test Mode.")
if args.action == 'generate-ini':
config.generate_ini_file(args)
elif args.action == 'gui' and HAVE_GUI:
if not os.path.exists(args.ini):
# ensure the gui has a config.ini file to load. Generate it if missing.
config.generate_ini_file(args)
gui_main(args)