adding signals to avoid tmp trash in fail executions and remove duplicate msteams

This commit is contained in:
rafael 2023-10-10 11:15:11 +02:00
parent c7d19c7036
commit 94328e200a
2 changed files with 13 additions and 82 deletions

View File

@ -1,6 +1,6 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import argparse, pymsteams import argparse, pymsteams, signal
parser = argparse.ArgumentParser(description='MS Teams connector') parser = argparse.ArgumentParser(description='MS Teams connector')
parser.add_argument('-d', '--data', help='Data in coma separate keypairs. Ex: test=5,house=2', required=True) parser.add_argument('-d', '--data', help='Data in coma separate keypairs. Ex: test=5,house=2', required=True)
@ -16,6 +16,18 @@ parser.add_argument('--button_desc', help='Pandora button description', default=
args = parser.parse_args() args = parser.parse_args()
# Define a function to handle the SIGINT signal
def sigint_handler(signal, frame):
print ('\nInterrupted by user')
sys.exit(0)
signal.signal(signal.SIGINT, sigint_handler)
# Define a function to handle the SIGTERM signal
def sigterm_handler(signum, frame):
print("Received SIGTERM signal.")
sys.exit(0)
signal.signal(signal.SIGTERM, sigterm_handler)
### Functions: ### Functions:
def parse_dic(cValues): def parse_dic(cValues):
"""convert coma separate keypairs into a dic. EX "test=5,house=8,market=2" wil return "{'test': '5', 'casa': '8', 'mercado': '2'}" """ """convert coma separate keypairs into a dic. EX "test=5,house=8,market=2" wil return "{'test': '5', 'casa': '8', 'mercado': '2'}" """

View File

@ -1,81 +0,0 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import argparse, pymsteams
parser = argparse.ArgumentParser(description='MsTeams connector')
parser.add_argument('-d', '--data', help='Data in coma separate keypairs. Ex: test=5,house=2', required=True)
parser.add_argument('-u', '--url', help='Teams webhook URL', required=True)
parser.add_argument('-t', '--alert_tittle', help='Alert tittle', default='PandoraFMS alert fired')
parser.add_argument('-D', '--alert_desc', help='Alert description', default='Alert Fired')
parser.add_argument('-m', '--message', help='Alert message', default='')
parser.add_argument('-T','--tittle_color', help='Alert tittle descripcion in HEX EX: 53e514', default="53e514")
parser.add_argument('--sub_desc', help='Alert sub description', default='Alert Fired')
parser.add_argument('--thumb', help='Custom thumbnail url', default="https://pandorafms.com/images/alerta_roja.png")
parser.add_argument('--button', help='Pandora button Url', default='https://pandorafms.com')
parser.add_argument('--button_desc', help='Pandora button description', default='Open web console')
args = parser.parse_args()
### Functions:
def parse_dic(cValues):
"""convert coma separate keypairs into a dic. EX "test=5,house=8,market=2" wil return "{'test': '5', 'casa': '8', 'mercado': '2'}" """
data={}
try :
for kv in cValues.split(","):
k,v = kv.strip().split("=")
data[k.strip()]=v.strip()
except Exception as e :
print(f"Warning, error parsing keypairs values: {e}")
return data
def add_embed_itmes(data):
"""iterate dictionary and set webhook fields, one for eacj keypair"""
for k, v in data.items() :
myMessageSection.addFact(f"{k}:", v)
##Main
# You must create the connectorcard object with the Microsoft Webhook URL
myTeamsMessage = pymsteams.connectorcard(args.url)
# Set Summary
myTeamsMessage.summary(args.message)
# Set Alert tittle
myTeamsMessage.title(args.alert_tittle)
# Set link buttom
myTeamsMessage.addLinkButton(args.button_desc, args.button)
# Set message color
myTeamsMessage.color(args.tittle_color)
# create the section
myMessageSection = pymsteams.cardsection()
# Section Title
myMessageSection.title(args.message)
# Activity Elements
myMessageSection.activityTitle(args.alert_desc)
myMessageSection.activitySubtitle(args.sub_desc)
myMessageSection.activityImage(args.thumb)
# Facts are key value pairs displayed in a list.
data = parse_dic(args.data)
add_embed_itmes(data)
# Section Text
# myMessageSection.text("This is my section text")
# Section Images
# myMessageSection.addImage("http://i.imgur.com/c4jt321l.png", ititle="This Is Fine")
# Add your section to the connector card object before sending
myTeamsMessage.addSection(myMessageSection)
# Then send the card
try:
myTeamsMessage.send()
except Exception as e :
print(f"Error sending to message: {e}")