def main():
"""Handle all options"""
parser = OptionParser(description="""WASKR Command Line Utility""", version=__version__)
parser.add_option(
"--server",
action="store_true",
help="""Runs the web server.
If no configuration file is passed localhost and port 8080 is used.""",
)
parser.add_option("--conf", help="""Pass a INI configuration file""")
parser.add_option("--add-user", help="""Adds a user email for the web interface""")
parser.add_option("--remove-user", help="""Removes a user email from the web interface""")
options, arguments = parser.parse_args()
if len(sys.argv) <= 1:
parser.print_help()
if options.server:
if options.conf and path.isfile(options.conf):
configuration = config.options(options.conf)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print "waskrc could not add user: %s" % e
def main():
"""Handle all options"""
parser = OptionParser(description="""WASKR Command Line Utility""",
version=__version__)
parser.add_option('--server',
action='store_true',
help="""Runs the web server.
If no configuration file is passed localhost and port 8080 is used.""")
parser.add_option('--conf', help="""Pass a INI configuration file""")
parser.add_option('--add-user',
help="""Adds a user email for the web interface""")
parser.add_option('--remove-user',
help="""Removes a user email from the web interface""")
options, arguments = parser.parse_args()
if len(sys.argv) <= 1:
parser.print_help()
if options.server:
if options.conf and path.isfile(options.conf):
configuration = config.options(options.conf)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print 'waskrc could not add user: %s' % e
def parseArgs(self, argv):
parser = OptionParser(description="""
A stats engine for WSGI applications
"""
,version='0.0.9')
parser.add_option('--server', action='store_true', help="""Runs the web server.
If no configuration file is passed localhost and port 8080 is used.""")
parser.add_option('--conf', help="""Pass a INI configuration file""")
parser.add_option('--add-user', help="""Adds a user email for the web interface""")
parser.add_option('--remove-user', help="""Removes a user email from the web interface""")
options, arguments = parser.parse_args()
if len(sys.argv) <= 1:
parser.print_help()
if options.server:
if options.conf and path.isfile(options.conf):
configuration = config.options(options.conf)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print 'waskrc could not add user: %s' % e
def __init__(self, config=None, test=False):
config = options(config)
try:
self.connection = Connection(
config['db_host'],
int(config['db_port']))
except Exception, error:
log.database.critical("Could not connect to MongoDB: %s" % error)
def config_values(self):
try:
config_list = [i for i in self.db.get_config_path()]
config_file = config_list[0][0]
config = config.options(config_file)
print "\nConfiguration file: %s\n" % config_file
for i in config.items():
print "%-15s= %-4s" % (i[0], i[1])
except Exception, error:
print "Could not complete command: %s" % error
def check_config(self):
# if any commands are run, check for a MASTER config file Location
config_db = self.db.get_config_path()
config_file = None
try:
config_list = [i for i in self.db.get_config_path()]
config_file = config_list[0][0]
config = config.options(config_file)
return config
except IndexError:
self.msg(msg=WARNING, std="err")
def check_config(self):
# if any commands are run, check for a MASTER config file Location
config_db = self.db.get_config_path()
config_file = None
try:
config_list = [i for i in self.db.get_config_path()]
config_file = config_list[0][0]
conf = options(config_file)
self.config = conf
return conf
except IndexError:
return False
def check_config(self):
# if any commands are run, check for a MASTER config file Location
config_db = self.db.get_config_path()
config_file = None
try:
config_list = [i for i in self.db.get_config_path()]
config_file = config_list[0][0]
conf = options(config_file)
self.config = conf
return conf
except IndexError:
return False
def __init__(self, config=None, test=False):
config = options(config)
self.db = self._location_verify(config['db_location'])
self.table = 'stats'
if test:
self.table = 'test_stats'
self.conn = sqlite3.connect(self.db)
self.c = self.conn.cursor()
self.c.execute(STATS_TABLE % self.table)
self.conn.commit()
log.database.debug("database connection initialized Sqlite3")
def __init__(self, app, config=None, framework_config=None):
"""Initialize the RequestStats Middleware
Every single request will get logged with a unix timestamp
the time it took to respond and the url that was requested.
The dictionary is passed on the RequestParser that deals
with building the cache and submitting the data to
the queue.
"""
self.app = app
self.config = options(config, framework_config)
self.parser = RequestParser(self.config)
setlogging(self.config)
log.middleware.debug('middleware Initialized')
def test_options_ini_no_webport(self):
actual = options('conf_seven.ini')
expected = {
'db_engine':'sqlite',
'db_location':'/tmp',
'web_password': False,
'plugins': None,
'plugin_path': False,
'server_id': '2',
'db_host': 'remote.example.com',
'db_port': '00000',
'application': 'secondary',
'web_host': 'web.example.com',
'cache': '10',
'web_port': '8080',
'log_level': 'DEBUG',
'log_format': '%(asctime)s %(levelname)s %(name)s %(message)s',
'log_datefmt' : '%H:%M:%S'
}
self.assertEqual(actual, expected)
def test_framework_config(self):
actual = options(framework_config=self.pylons_config)
expected ={
'server_id': 2,
'db_engine':'sqlite',
'db_location':'/tmp',
'db_host': 'remote.example.com',
'db_port': 4444,
'application': 'secondary',
'web_host': 'web.example.com',
'web_port': '8080',
'web_password': False,
'plugins': None,
'plugin_path': False,
'cache': 20,
'log_level': 'DEBUG',
'log_format': '%(asctime)s %(levelname)s %(name)s %(message)s',
'log_datefmt' : '%H:%M:%S'
}
self.assertEqual(actual, expected)
def test_custom_plugin_none(self):
"""When bad configured it will not display or get parsed"""
actual = options('conf_nine.ini')
expected = {
'db_engine':'sqlite',
'db_location':'/tmp',
'web_password': False,
'plugins': None,
'plugin_path': False,
'server_id': '2',
'db_host': 'remote.example.com',
'db_port': '00000',
'application': 'secondary',
'web_host': 'web.example.com',
'cache': '10',
'web_port': '8080',
'log_level': 'DEBUG',
'log_format': '%(asctime)s %(levelname)s %(name)s %(message)s',
'log_datefmt' : '%H:%M:%S'
}
self.assertEqual(actual, expected)
def test_options_empty(self):
actual = options()
expected = self.defaults
self.assertEqual(actual, expected)
def __init__(self, config=None, test=False):
self.config = options(config)
self.engine = self._load_engine()
self.stats = self.engine.Stats(config, test)
def __init__(self, config=None, test=False):
self.config = options(config)
self.engine = self._load_engine()
self.stats = self.engine.Stats(config, test)
from os import path
from bottle import response, request, redirect, route, run, view, send_file, local, TEMPLATE_PATH
from waskr import log
from waskr.config import options, setlogging
from waskr.database import Stats
# Fixes a call from a different path
CWD = path.abspath(__file__)
APP_CWD = path.dirname(CWD)
view_dir = APP_CWD+'/views'
TEMPLATE_PATH.append(view_dir)
# Initialize the CONF with proper values
CONF = options()
# Initialize logging
setlogging(CONF)
log.server.debug("initialized logging")
@route('/')
@view('index')
def index():
logged_in()
return dict(last_received=local.db.last_insert(),
apps=local.db.apps_nodes())
@route('/application/:name/:type')
@route('/application/:name/:type/:minutes')
def test_options_typeError(self):
actual = options(['a list should never be passed'])
expected = self.defaults
self.assertEqual(actual, expected)
from os import path
from bottle import response, request, redirect, route, run, view, send_file, local, TEMPLATE_PATH
from waskr import log
from waskr.config import options, setlogging
from waskr.database import Stats
# Fixes a call from a different path
CWD = path.abspath(__file__)
APP_CWD = path.dirname(CWD)
view_dir = APP_CWD + '/views'
TEMPLATE_PATH.append(view_dir)
# Initialize the CONF with proper values
CONF = options()
# Initialize logging
setlogging(CONF)
log.server.debug("initialized logging")
@route('/')
@view('index')
def index():
logged_in()
return dict(last_received=local.db.last_insert(),
apps=local.db.apps_nodes())
@route('/application/:name/:type')
@route('/application/:name/:type/:minutes')
if options.server:
if options.conf and path.isfile(options.conf):
configuration = config.options(options.conf)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print 'waskrc could not add user: %s' % e
if options.remove_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.remove_user(options.add_user)
print "User %s removed from DB" % options.remove_user
except Exception, e:
print 'waskrc could not remove user: %s' % e
if __name__ == "__main__":
main()
if options.server:
if options.conf and path.isfile(options.conf):
configuration = config.options(options.conf)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print "waskrc could not add user: %s" % e
if options.remove_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.remove_user(options.add_user)
print "User %s removed from DB" % options.remove_user
except Exception, e:
print "waskrc could not remove user: %s" % e
if __name__ == "__main__":
main()
def test_options_invalid_file(self):
actual = options('/path/to/invalid/file.txt')
expected = self.defaults
self.assertEqual(actual, expected)
def test_options_TypeError(self):
actual = options(config={})
expected = self.defaults
self.assertEqual(actual, expected)
server.CONF = configuration
server.main(configuration)
else:
server.main()
if options.add_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.add_user(options.add_user)
print "User %s was added to the DB" % options.add_user
except Exception, e:
print 'waskrc could not add user: %s' % e
if not options.conf:
config = config.options('app.ini')
db = database.Stats(config)
if options.remove_user and options.conf:
try:
config = config.options(options.conf)
db = database.Stats(config)
db.remove_user(options.add_user)
print "User %s removed from DB" % options.remove_user
except Exception, e:
print 'waskrc could not remove user: %s' % e
main = WaskrCommands
def test_options_Error_string(self):
actual = options('a string should never be passed')
expected = self.defaults
self.assertEqual(actual, expected)