Separated function for assembling SMS and the main program

This commit is contained in:
root 2017-02-04 12:30:34 +01:00
parent 1745ad0be4
commit 6334bbda32

View File

@ -4,16 +4,20 @@ import logging
import os
from pydbus import SystemBus
from gi.repository import GLib
from datetime import datetime
import sms_functions
def assemble_sms():
since_date = datetime.fromtimestamp(int(os.environ.get('LASTSTATECHANGE')))
str_list = []
str_list.append("Icinga {}\n".format(os.environ.get('NOTIFICATIONTYPE')))
str_list.append("{} is {} [{}]\n".format(os.environ.get('HOSTALIAS'), os.environ.get('HOSTSTATE'), os.environ.get('HOSTADDRESS')))
str_list.append("since {}\n".format(os.environ.get('LASTSTATECHANGE')))
str_list.append("since {}\n".format(since_date.strftime('%Y-%m-%dT%H:%M:%S')))
str_list.append("Comment: [{}] {}".format(os.environ.get('NOTIFICATIONAUTHORNAME'), os.environ.get('NOTIFICATIONCOMMENT')))
return ''.join(str_list)
def main():
# Setup logging
logfile = "/opt/sms/sms.log"
FORMAT = '%(asctime)-15s %(message)s'
@ -43,3 +47,6 @@ while not success and retries < max_retries:
logger.error('Error when sending SMS: {0}'.format(e.value))
else:
success = True
if __name__ == "__main__":
main()