diff options
Diffstat (limited to 'src/consolidation_run')
-rwxr-xr-x | src/consolidation_run | 53 |
1 files changed, 38 insertions, 15 deletions
diff --git a/src/consolidation_run b/src/consolidation_run index 7ef1a5c..3c514a7 100755 --- a/src/consolidation_run +++ b/src/consolidation_run @@ -15,45 +15,68 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. -import argparse +import argparse +import ConfigParser import logging +import sys, os -from sugar_stats_consolidation.db import * -from sugar_stats_consolidation.rrd_files import * -from sugar_stats_consolidation.consolidation import * + +sys.path.append(os.getcwd()) + +from db import * +from rrd_files import * +from consolidation import * parser = argparse.ArgumentParser() -parser.add_argument('--db_user',required=True) -parser.add_argument('--db_pass',required=True) -parser.add_argument('--db_name',required=True) -parser.add_argument('--rrd_path',required=True) -parser.add_argument('--log_path',required=True) +parser.add_argument('--config_file', required=False) +#parser.add_argument('--db_user',required=True) +#parser.add_argument('--db_pass',required=True) +#parser.add_argument('--db_name',required=True) +#parser.add_argument('--rrd_path',required=True) +#parser.add_argument('--log_path',required=True) args = parser.parse_args() _LOG_FORMAT='[%(asctime)s]-%(levelname)s-\'%(name)s\': %(message)s' - -logging.basicConfig(filename=os.path.join(args.log_path,'sugar-stats-consolidation.log'), format=_LOG_FORMAT, level=logging.DEBUG) -log = logging.getLogger(__name__) +_LOG_LEVEL={'debug':logging.DEBUG, 'info':logging.INFO, 'warning':logging.WARNING, 'error':logging.ERROR} +#logging.basicConfig(filename=os.path.join(args.log_path,'sugar-stats-consolidation.log'), format=_LOG_FORMAT, level=logging.DEBUG) +#log = logging.getLogger(__name__) +log=None def main(): + config = ConfigParser.ConfigParser() + if (args.config_file == None): + config_file = "./stats-consolidation.conf" + else: + config_file = args.config_file + + config.read(config_file) + + log_path = config.get('main', 'log_path') + log_level = config.get('main', 'log_level') + db_name = config.get('main', 'db_name') + db_user = config.get('main', 'db_user') + db_pass = config.get('main', 'db_pass') + rrd_path = config.get('main', 'rrd_path') + + logging.basicConfig(filename=os.path.join(log_path,'sugar-stats-consolidation.log'), format=_LOG_FORMAT, level=_LOG_LEVEL[log_level]) + log = logging.getLogger(__name__) log.info('**************************************') log.info(' Starting consolidation ') log.info('**************************************') - try: - db = DB_Stats (args.db_name, args.db_user, args.db_pass) + db = DB_Stats (db_name, db_user, db_pass) db.create() except Exception as e: log.warning('Creating DB: %s', str(e)) try: - con = Consolidation(args.rrd_path, db) + con = Consolidation(rrd_path, db) con.process_rrds() db.close() |