Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/src/consolidation_run
diff options
context:
space:
mode:
Diffstat (limited to 'src/consolidation_run')
-rwxr-xr-xsrc/consolidation_run53
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()