Bug fixes in graphite and statsd output

2 files changed, 6 insertions, 2 deletions

diff --git a/logster/outputs/GraphiteOutput.py b/logster/outputs/GraphiteOutput.py
index fb7c7f5..c2ff501 100644
--- a/logster/outputs/GraphiteOutput.py
+++ b/logster/outputs/GraphiteOutput.py
@@ -1,4 +1,5 @@
 from logster.logster_helper import LogsterOutput
+import re
 import socket
 
 
@@ -48,7 +49,7 @@ class GraphiteOutput(LogsterOutput):
                 self.logger.debug("Submitting Graphite metric: %s" % metric_string)
 
                 if (not self.dry_run):
-                    s.sendall(bytes("%s\n" % metric_string))
+                    s.sendall(bytes("%s\n" % metric_string, 'ascii'))
                 else:
                     print("%s %s" % (self.graphite_host, metric_string))
         finally:
diff --git a/logster/outputs/StatsdOutput.py b/logster/outputs/StatsdOutput.py
index d4c7453..64cd2f8 100644
--- a/logster/outputs/StatsdOutput.py
+++ b/logster/outputs/StatsdOutput.py
@@ -14,6 +14,9 @@ class StatsdOutput(LogsterOutput):
 
     def __init__(self, parser, options, logger):
         super(StatsdOutput, self).__init__(parser, options, logger)
+        if not options.statsd_host:
+            parser.print_help()
+            parser.error("You must supply --statsd-host when using 'statsd' as an output type.")
         self.statsd_host = options.statsd_host
 
 
@@ -28,6 +31,6 @@ class StatsdOutput(LogsterOutput):
 
             if (not self.dry_run):
                 udp_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-                udp_sock.sendto(metric_string, (host[0], int(host[1])))
+                udp_sock.sendto(bytes(metric_string, 'ascii'), (host[0], int(host[1])))
             else:
                 print("%s %s" % (self.statsd_host, metric_string))