64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
import boto3
|
|
import base64
|
|
import hashlib
|
|
from datetime import datetime
|
|
import os
|
|
import sys
|
|
|
|
sys.path.append(os.getcwd())
|
|
from pullTrafficInfo import getTrafficInfo
|
|
|
|
|
|
class sendToSQS(object):
|
|
"""docstring for sendToSQS"""
|
|
|
|
def __init__(self, session, queueURL):
|
|
super(sendToSQS, self).__init__()
|
|
self.session = session
|
|
self.sqs = session.client('sqs')
|
|
self.queueURL = queueURL
|
|
|
|
@classmethod
|
|
def createSession(cls, profileName, queueURL):
|
|
session = boto3.Session(profile_name=profileName)
|
|
return cls(session, queueURL)
|
|
|
|
def sendMessage(self, message, messageGroupId, b64=True, dedup=False):
|
|
currentTime = datetime.now().strftime('%H:%M:%S.%f')
|
|
if b64:
|
|
message = (base64.b64encode(message.encode())).decode()
|
|
if not dedup:
|
|
dedupId = hashlib.md5((message + currentTime).encode()).hexdigest()
|
|
msg = self.sqs.send_message(
|
|
QueueUrl=self.queueURL,
|
|
MessageBody=message,
|
|
MessageGroupId=messageGroupId,
|
|
MessageDeduplicationId=dedupId,
|
|
)
|
|
else:
|
|
msg = self.sqs.send_message(
|
|
QueueUrl=self.queueURL,
|
|
MessageBody=message,
|
|
MessageGroupId=messageGroupId,
|
|
)
|
|
if msg is not None:
|
|
print(msg['MessageId'])
|
|
|
|
|
|
inst = sendToSQS.createSession(
|
|
profileName='plex-aws',
|
|
queueURL='https://sqs.eu-west-1.amazonaws.com'
|
|
'/745437999005/slack-bot.fifo',
|
|
)
|
|
|
|
instM = (
|
|
getTrafficInfo.getTrafficURL('M62')
|
|
.findIncidents()
|
|
.getIncidentInformation()
|
|
.generateOutput()
|
|
)
|
|
|
|
for _ in range(0, 5):
|
|
for item in instM.output:
|
|
inst.sendMessage(message=item, messageGroupId='slack-bot-M62')
|