/
receiver.py
73 lines (60 loc) · 2.01 KB
/
receiver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#!usr/local/bin python
import pika
from settings import config
from rabbitmessenger.emailsender import EmailSender
from rabbitmessenger.SchedSms import SchedSms
import ast
from rabbitmessenger.RabbitThreadedConsumers import RabbitThreadedConsumer
import logging
import time
#logging.basicConfig(level=logging.DEBUG,format='[%(levelname)s] (%(threadName)-10s) %(message)s',)
class receiver():
def __init__(self):
self.emailer = EmailSender(
config["SMTPServer"],
config["SMTPPort"],
config["SMTPUserName"],
config["SMTPPassword"]
)
self.smser = SchedSms(config["PauseSMS"])
def messageHandler(self,ch,method,properties,body):
message = None
#parse the body message
try:
message = ast.literal_eval(body)
except ValueError:
print "[Error] : This reciever does not accept non dictionary messages."
ch.basic_ack(delivery_tag=method.delivery_tag) #thrash the message from the queue
return
isSuccess = False
#by now we recieved a message.
if message["type"] == "email":
print "[Info] : recieved email message"
isSuccess = self.emailer.Send(message)
elif message["type"] == "sms":
print "[Info] : recieved sms message"
isSuccess = self.smser.Send(message);
else:
print "[Error] : Unkown message type. System can handle only emails and smses yet."
if (isSuccess):
ch.basic_ack(delivery_tag=method.delivery_tag)
return
def start_listening(self,host,queue):
self.__rabbit_consumer = RabbitThreadedConsumer(host,queue,self.messageHandler)
self.__rabbit_consumer.setDaemon(True)
self.__rabbit_consumer.start()
def stop_listening(self):
if self.__rabbit_consumer != None:
self.__rabbit_consumer.stop_consuming()
self.emailer.Stop();
self.smser.Stop();
if __name__ == "__main__":
current_reciever = receiver()
current_reciever.start_listening('localhost','jrd_main_messages')
try:
print "Messenger started. press ctrl + c to exit."
while True:
time.sleep(1)
except KeyboardInterrupt:
print "stopping the app."
current_reciever.stop_listening()