def start_drivers(sensor, pin, best_temp, best_hum):
dht = DHTDriver(sensor, pin)
mntr = Monitor(best_temp, best_hum)
while 1:
humidity, temp = dht.get_reading()
# TODO: how to handle regulating things?
mntr.check(humidity, temp)
time.sleep(5)
def random_tester(): monitor = Monitor() while True: user = random.randint(1,30) if not monitor.check(str(user)): print "malicious user detected" time.sleep(0.05)
def test_good_content(self):
"""Confirm that everything works right if the content is good"""
monitor = Monitor('http://fake', {"content": ["fake"]})
monitor._fetch_contents = Mock(return_value="Some fake content")
result = monitor.check()
assert result is not False
def test_ioerror(self):
"""Confirm correct behavior on network error"""
monitor = Monitor('http://fake', {})
monitor._fetch_contents = Mock(side_effect=IOError)
result = monitor.check()
assert result is False
def test_invalid_content(self):
"""Confirm correct behavior on invalid content"""
monitor = Monitor('http://fake', {})
monitor._fetch_contents = Mock(return_value="Some fake content")
monitor._check_contents = Mock(return_value=False)
result = monitor.check()
assert result is False
def main():
logging.basicConfig(level=logging.INFO,
format="%(asctime)s %(levelname)s MODULE:%(module)s LineNO:%(lineno)d %(message)s",
stream=sys.stdout)
timeout = getattr(settings, "TIMEOUT", DEFAULT_TIMEOUT)
check_inv = getattr(settings, "CHECK_INTERVAL", DEFAULT_INTERVAL)
# callback_url = getattr(settings, "CALLBACK_URL", None)
monitor = Monitor(settings.NOTICATION, settings.HANDLER, timeout)
try:
while True:
monitor.check(settings.CHECK_URL, settings.RETRY_COUNT)
time.sleep(check_inv)
except KeyboardInterrupt:
pass
except Exception, e:
logging.error(e)
def main():
from conf import settings
logging.basicConfig(level=logging.INFO,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stdout);
timeout = getattr(settings, 'TIMEOUT', DEFAULT_TIMEOUT)
check_time = getattr(settings, 'CHECK_INTERVAL', DEFAULT_INTERVAL)
monitor = Monitor(settings.ALERT_BACKENDS, timeout)
try:
while True:
monitor.check(settings.URLS_TO_CHECK)
time.sleep(check_time)
except KeyboardInterrupt:
pass
except Exception, e:
logging.error(e)
def main():
from conf import settings
logging.basicConfig(level=logging.INFO,
format='%(asctime)s %(levelname)s %(message)s',
stream=sys.stdout)
timeout = getattr(settings, 'TIMEOUT', DEFAULT_TIMEOUT)
check_time = getattr(settings, 'CHECK_INTERVAL', DEFAULT_INTERVAL)
callBackUri = getattr(settings, 'ALERT_CALLBACK', None)
monitor = Monitor(settings.ALERT_BACKENDS, timeout)
try:
while True:
monitor.check(settings.URLS_TO_CHECK, callBackUri)
time.sleep(check_time)
except KeyboardInterrupt:
pass
except Exception, e:
logging.error(e)
def main():
argc = len(sys.argv)
if argc < 2:
print "usage: python monitor_server port"
return
else:
server_port = int(sys.argv[1])
monitor = Monitor()
context = zmq.Context()
# monitor server
server = context.socket(zmq.ROUTER)
server.bind("tcp://*:" + str(server_port))
# heartbeat
heartbeat = context.socket(zmq.ROUTER)
heartbeat.bind("tcp://*:" + str(server_port+1))
poller = zmq.Poller()
poller.register(server, zmq.POLLIN)
poller.register(heartbeat, zmq.POLLIN)
while True:
socks = dict(poller.poll())
if socks.get(server) == zmq.POLLIN:
addr = server.recv()
tmp = server.recv()
data = server.recv()
print data
server.send(addr, zmq.SNDMORE)
server.send("", zmq.SNDMORE)
if monitor.check(data):
server.send("pass")
else:
server.send("fail")
elif socks.get(heartbeat) == zmq.POLLIN:
addr = heartbeat.recv()
tmp = heartbeat.recv()
data = heartbeat.recv()
print "heartbeat"
heartbeat.send(addr, zmq.SNDMORE)
heartbeat.send("", zmq.SNDMORE)
heartbeat.send("live")
def load_tester(): count = 1 monitor = Monitor() while True: monitor.check(str(count)) count = count + 1
