Fixes #10

vladiate/logs.py

@@ -0,0 +1,9 @@
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("vlad_logger")
+sh = logging.StreamHandler()
+sh.setLevel(logging.INFO)
+sh.setFormatter(logging.Formatter('%(message)s'))
+logger.addHandler(sh)
+logger.propagate = False

vladiate/main.py

@@ -1,21 +1,13 @@
 import vladiate
 from vladiate import Vlad
+from vladiate import logs
 
 import os
 import sys
 import inspect
-import logging
 from optparse import OptionParser
 from pkg_resources import get_distribution
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("vlad_logger")
-sh = logging.StreamHandler()
-sh.setLevel(logging.INFO)
-sh.setFormatter(logging.Formatter('%(message)s'))
-logger.addHandler(sh)
-logger.propagate = False
-
 
 def parse_options():
     """
@@ -142,6 +134,7 @@ def load_vladfile(path):
 
 def main():
     parser, options, arguments = parse_options()
+    logger = logs.logger
 
     if options.show_version:
         print "Vladiate %s" % (get_distribution('vladiate').version, )

vladiate/vlad.py

@@ -1,15 +1,15 @@
 import sys
 import csv
-import logging
 from collections import defaultdict
 from vladiate.exceptions import ValidationException
 from vladiate.validators import EmptyValidator
+from vladiate import logs
 
 
 class Vlad(object):
 
     def __init__(self, source, validators={}, default_validator=EmptyValidator):
-        self.logger = logging.getLogger("vlad_logger")
+        self.logger = logs.logger
         self.failures = defaultdict(lambda: defaultdict(list))
         self.missing_validators = None
         self.missing_fields = None