a58f3cb7df
Change-Id: I1f39f73b899dadc04e9c592de8b087a5545da9ce Former-commit-id: 18d692c17ea11537d0eb832fbe63a04b59b0040c
115 lines
4.1 KiB
Python
115 lines
4.1 KiB
Python
##############################################################################
|
|
# Notifies EDEX of an ACK or NACK from MHS
|
|
#
|
|
# SOFTWARE HISTORY
|
|
#
|
|
# Date Ticket# Engineer Description
|
|
# ------------ ---------- ----------- --------------------------
|
|
# 04/06/2012 10388 D. Friedman Initial version
|
|
# 10/09/12 DR 13901 D. Friedman Limit execution time
|
|
# 06/13/2013 DR 16242 D. Friedman Add Qpid authentication info
|
|
# 03/06/2014 DR 17907 D. Friedman Workaround for issue QPID-5569
|
|
##############################################################################
|
|
|
|
import getopt
|
|
import os
|
|
import os.path
|
|
import sys
|
|
|
|
from lib.Util import doWithinTime
|
|
|
|
import qpid
|
|
from qpid.util import connect
|
|
from qpid.connection import Connection
|
|
from qpid.datatypes import Message, uuid4
|
|
|
|
DESTINATION = 'amq.topic'
|
|
TOPIC_NAME = 'mhs.ackmgr'
|
|
QPID_USERNAME = 'guest'
|
|
QPID_PASSWORD = 'guest'
|
|
|
|
class MhsAckNotification:
|
|
def __init__(self):
|
|
self.messageId = None
|
|
self.sender = None
|
|
self.response = None
|
|
|
|
def remove_file(*parts):
|
|
path = os.path.join(*parts)
|
|
if os.path.exists(path):
|
|
os.remove(path)
|
|
|
|
def get_qpid_connection(broker_addr):
|
|
try:
|
|
socket = connect(broker_addr, 5672)
|
|
connection = Connection (sock=socket, username=QPID_USERNAME, password=QPID_PASSWORD)
|
|
connection.start()
|
|
return connection
|
|
except:
|
|
sys.stderr.write("mhsAckNotify: connect to %s: %s\n" % (broker_addr, sys.exc_info()[1],))
|
|
return None
|
|
|
|
def send_message(connection, notif):
|
|
session = connection.session(str(uuid4()))
|
|
|
|
props = session.delivery_properties(routing_key=TOPIC_NAME)
|
|
head = session.message_properties(application_headers={'sender':notif.sender,
|
|
'response':notif.response},
|
|
user_id=QPID_USERNAME) # For issue QPID-5569. Fixed in Qpid 0.27
|
|
session.message_transfer(destination=DESTINATION, message=Message(props, head, notif.messageId))
|
|
session.close(timeout=10)
|
|
connection.close()
|
|
|
|
def run():
|
|
mhs_data_dir = os.getenv('MHS_DATA', '/data/fxa/mhs')
|
|
notif = MhsAckNotification()
|
|
|
|
opts, args = getopt.getopt(sys.argv[1:], 'm:r:s:')
|
|
for k, v in opts:
|
|
if k == '-m':
|
|
notif.messageId = v
|
|
elif k == '-r':
|
|
notif.response = v
|
|
elif k == '-s':
|
|
notif.sender = v
|
|
|
|
if notif.messageId:
|
|
# Delete the .doc file
|
|
if mhs_data_dir:
|
|
try:
|
|
fn = '%s-%s.doc' %(notif.messageId, notif.sender)
|
|
if notif.response == 'ACK':
|
|
remove_file(mhs_data_dir, 'ackq', fn)
|
|
elif notif.response == 'NACK':
|
|
# A1 just uses message ID, but that does not look right...
|
|
# Do both to be safe
|
|
remove_file(mhs_data_dir, 'nackq', fn)
|
|
remove_file(mhs_data_dir, 'nackq', '%s.doc' % (notif.messageId, ))
|
|
except:
|
|
sys.stderr.write("mhsAckNotify: error removing MHS file: %s\n" % (sys.exc_info()[1],))
|
|
|
|
# TODO: Should have BROKER_ADDR in CLI setup.env.
|
|
broker_addr = os.getenv('BROKER_ADDR')
|
|
if broker_addr is None:
|
|
broker_addr = os.getenv('DEFAULT_HOST')
|
|
if broker_addr == 'ec':
|
|
broker_addr = 'cp1f'
|
|
if broker_addr is None:
|
|
broker_addr = 'localhost'
|
|
|
|
try:
|
|
connection = doWithinTime(get_qpid_connection, args=(broker_addr,))
|
|
if connection:
|
|
doWithinTime(send_message, max_tries=1, args=(connection, notif))
|
|
except:
|
|
sys.stderr.write("mhsAckNotify: error sending message: %s\n" % (sys.exc_info()[1],))
|
|
return 1
|
|
else:
|
|
sys.stderr.write("mhsAckNotify: message ID not specified\n")
|
|
sys.stderr.write("usage: mhsAckNotify -m <messageID> -s <sender> -r <response>\n")
|
|
return 1
|
|
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(run())
|