def setUp(self):
self.app = create_app(config=self.AFM_CONFIG)
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
self.test_app = TestApp(self.app)
def main():
parser = argparse.ArgumentParser(description='SLAMon Agent Fleet Manager')
parser.add_argument(
'--database-uri',
type=str,
default=None,
help='Set the AFM database URI, defaults to in memory sqlite')
parser.add_argument('--config',
'-c',
type=str,
default=None,
help='Load AFM configuration from a file')
subparsers = parser.add_subparsers()
run_parser = subparsers.add_parser(
'run',
help='Run AFM',
description='Run an instance of an Agent Fleet Manager '
'that listens to given host address')
run_parser.add_argument(
'host',
type=str,
help='Host name or address e.g. localhost or 127.0.0.1')
run_parser.add_argument('port',
type=int,
default=8080,
nargs='?',
help='Listening port, defaults to 8080')
run_parser.set_defaults(func=run_afm)
create_parser = subparsers.add_parser(
'create-tables',
help='Create SQL tables',
description='Create required database tables to PostgreSQL')
create_parser.set_defaults(func=create)
drop_parser = subparsers.add_parser(
'drop-tables',
help='Drop SQL tables',
description='Drop created database tables from PostgreSQL')
drop_parser.set_defaults(func=drop)
args = parser.parse_args()
if not hasattr(args, 'func'):
# No function defined, print usage and exit
parser.print_help()
exit(1)
app = create_app(
config_file=os.path.abspath(args.config) if args.config else None)
if args.database_uri:
# override the database URI
app.config.update(SQLALCHEMY_DATABASE_URI=args.database_uri)
args.func(app, args)
def main():
parser = argparse.ArgumentParser(description='SLAMon Agent Fleet Manager')
parser.add_argument('--database-uri', type=str, default=None,
help='Set the AFM database URI, defaults to in memory sqlite')
parser.add_argument('--config', '-c', type=str, default=None,
help='Load AFM configuration from a file')
subparsers = parser.add_subparsers()
run_parser = subparsers.add_parser('run', help='Run AFM',
description='Run an instance of an Agent Fleet Manager '
'that listens to given host address')
run_parser.add_argument('host', type=str, help='Host name or address e.g. localhost or 127.0.0.1')
run_parser.add_argument('port', type=int, default=8080, nargs='?', help='Listening port, defaults to 8080')
run_parser.set_defaults(func=run_afm)
create_parser = subparsers.add_parser('create-tables', help='Create SQL tables',
description='Create required database tables to PostgreSQL')
create_parser.set_defaults(func=create)
drop_parser = subparsers.add_parser('drop-tables', help='Drop SQL tables',
description='Drop created database tables from PostgreSQL')
drop_parser.set_defaults(func=drop)
args = parser.parse_args()
if not hasattr(args, 'func'):
# No function defined, print usage and exit
parser.print_help()
exit(1)
app = create_app(config_file=os.path.abspath(args.config) if args.config else None)
if args.database_uri:
# override the database URI
app.config.update(SQLALCHEMY_DATABASE_URI=args.database_uri)
args.func(app, args)