create test-run script
This commit is contained in:
@@ -15,7 +15,7 @@ class sendToSQS(object):
|
||||
def __init__(self, session, queueURL):
|
||||
super(sendToSQS, self).__init__()
|
||||
self.session = session
|
||||
self.sqs = session.client("sqs")
|
||||
self.sqs = session.client('sqs')
|
||||
self.queueURL = queueURL
|
||||
|
||||
@classmethod
|
||||
@@ -24,7 +24,7 @@ class sendToSQS(object):
|
||||
return cls(session, queueURL)
|
||||
|
||||
def sendMessage(self, message, messageGroupId, b64=True, dedup=False):
|
||||
currentTime = datetime.now().strftime("%H:%M:%S.%f")
|
||||
currentTime = datetime.now().strftime('%H:%M:%S.%f')
|
||||
if b64:
|
||||
message = (base64.b64encode(message.encode())).decode()
|
||||
if not dedup:
|
||||
@@ -42,17 +42,17 @@ class sendToSQS(object):
|
||||
MessageGroupId=messageGroupId,
|
||||
)
|
||||
if msg is not None:
|
||||
print(msg["MessageId"])
|
||||
print(msg['MessageId'])
|
||||
|
||||
|
||||
inst = sendToSQS.createSession(
|
||||
profileName="plex-aws",
|
||||
queueURL="https://sqs.eu-west-1.amazonaws.com"
|
||||
"/745437999005/slack-bot.fifo",
|
||||
profileName='plex-aws',
|
||||
queueURL='https://sqs.eu-west-1.amazonaws.com'
|
||||
'/745437999005/slack-bot.fifo',
|
||||
)
|
||||
|
||||
instM = (
|
||||
getTrafficInfo.getTrafficURL("M62")
|
||||
getTrafficInfo.getTrafficURL('M62')
|
||||
.findIncidents()
|
||||
.getIncidentInformation()
|
||||
.generateOutput()
|
||||
@@ -60,4 +60,4 @@ instM = (
|
||||
|
||||
for _ in range(0, 5):
|
||||
for item in instM.output:
|
||||
inst.sendMessage(message=item, messageGroupId="slack-bot-M62")
|
||||
inst.sendMessage(message=item, messageGroupId='slack-bot-M62')
|
||||
|
||||
Reference in New Issue
Block a user