def __call__(self, target=None):
if target is None:
return self.logger is not None
if self.logger is not None:
for handler in tuple(self.logger.handlers):
self.logger.removeHandler(handler)
if target in ('off', False):
if self.state:
self.logger.setLevel(0)
self.logger.addHandler(logging.NullHandler())
return
if target == 'on':
handler = logging.StreamHandler()
elif isinstance(target, basestring):
url = urlparse(target)
if not url.scheme and url.path:
handler = logging.FileHandler(url.path)
elif url.scheme == 'syslog':
handler = logging.SysLogHandler(address=url.netloc.split(':'))
else:
raise ValueError('logging scheme not supported')
else:
handler = target
fmt = '%(asctime)s %(levelname)8s %(name)s: %(message)s'
formatter = logging.Formatter(fmt)
handler.setFormatter(formatter)
self.logger.addHandler(handler)
self.logger.setLevel(logging.DEBUG)
def __call__(self, target=None):
if target is None:
return self.logger is not None
if target == 'off':
if self.logger is not None:
self.logger.setLevel(logging.INFO)
self.logger.removeHandler(self.handler)
self.logger = None
self.handler = None
return
if target == 'on':
handler = logging.StreamHandler()
elif isinstance(target, basestring):
url = urlparse(target)
if not url.scheme and url.path:
handler = logging.FileHandler(url.path)
elif url.scheme == 'syslog':
handler = logging.SysLogHandler(address=url.netloc.split(':'))
else:
raise ValueError('logging scheme not supported')
else:
handler = target
self.handler = handler
self.logger = logging.getLogger('')
self.logger.addHandler(self.handler)
self.logger.setLevel(logging.DEBUG)
def main():
parser = ArgumentParser()
parser.add_argument('-c', '--config-file', default='config.yml')
parser.add_argument('--description', required=True,
help='Description (use "-" to use stdin)' )
parser.add_argument('--notification_type', required=True,
help='Notification type (PROBLEM|RECOVERY|CUSTOM). Nagios variable - $NOTIFICATIONTYPE$" ')
parser.add_argument('--host_name', required=True, help='Host name. Nagios variable - $HOSTNAME$')
parser.add_argument('--service_description', required=False, help='Service Description. Nagios variable - $SERVICEDESC$')
parser.add_argument('--long_date_time', required=True, help='Date and time. Nagios variable - $LONGDATETIME$' )
parser.add_argument('--syslog', action='store_true', default=False,
help='Log to syslog')
parser.add_argument('--debug', action='store_true', default=False,
help='Enable debug log level')
parser.add_argument('--log_file', default=sys.stdout,
help='Log file. default: stdout. Ignored if logging configured to syslog')
args = parser.parse_args()
LOG = logging.getLogger()
if args.syslog:
handler = logging.SysLogHandler()
elif (args.log_file != sys.stdout ):
handler = logging.FileHandler(args.log_file)
else:
handler = logging.StreamHandler(sys.stdout)
if args.debug:
log_level = logging.DEBUG
else:
log_level = logging.INFO
formatter = logging.Formatter(
'{} nagios_to_sfdc %(asctime)s %(process)d %(levelname)s %(name)s '
'[-] %(message)s'.format(socket.getfqdn()),
datefmt='%Y-%m-%d %H:%M:%S'
)
handler.setFormatter(formatter)
LOG.setLevel(log_level)
LOG.addHandler(handler)
# Read from stdin if desctiption defined as '-'
if args.description == '-':
args.description = ''.join(sys.stdin.readlines())
nagios_data = {
'notification_type': args.notification_type,
'description': args.description,
'host_name': args.host_name,
'long_date_time': args.long_date_time,
}
if args.service_description:
nagios_data['service_description'] = args.service_description
LOG.debug('Nagios data: {} '.format(nagios_data))
with open(args.config_file) as fp:
config = yaml.load(fp)
if 'sfdc_organization_id' in config:
organizationId = config['sfdc_organization_id']
else:
organizationId = None
sfdc_oauth2 = OAuth2(client_id=config['sfdc_client_id'],
client_secret=config['sfdc_client_secret'],
username=config['sfdc_username'],
password=config['sfdc_password'],
auth_url=config['sfdc_auth_url'],
organizationId = organizationId )
environment = config['environment']
# Alert ID shoud be uniq for env
Alert_ID = '{}--{}'.format(environment,args.host_name)
if args.service_description:
nagios_data['service_description'] = args.service_description
Alert_ID = '{}--{}'.format(Alert_ID, args.service_description)
LOG.debug('Alert_Id: {} '.format(Alert_ID))
sfdc_client = Client(sfdc_oauth2)
data = {
'Payload__c': json.dumps(nagios_data),
'Alert_ID__c': Alert_ID,
'Cloud__c': environment,
}
comment_data = {
'related_id__c': "-1",
'Comment__c': json.dumps(nagios_data),
'Alert_Id__c': Alert_ID,
'MOS_Alert_Name__c': "MA-0" ,
'MosAlertId__c': "-1",
'Cloud__c': environment,
}
try:
new_alert = sfdc_client.create_mos_alert(data)
except Exception as E:
LOG.debug(E)
sys.exit(1)
LOG.debug('New MOS_Alert status code {} '.format(new_alert.status_code))
LOG.debug('New MOS_Alert: {} '.format(new_alert.text))
if (new_alert.status_code == 400) and (new_alert.json()[0]['errorCode'] == 'DUPLICATE_VALUE'):
# Mos Alert exists
LOG.debug('Code: {}, Error message: {} '.format(new_alert.status_code, new_alert.text))
# Find Alert ID
Id = new_alert.json()[0]['message'].split(" ")[-1]
LOG.debug('MOS_Alert_Id: {} '.format(Id))
# Get Alert name from current alart
current_alert = sfdc_client.get_mos_alert(Id).json()
comment_data['MOS_Alert_Name__c'] = current_alert['Name']
LOG.debug('Existing MOS_Alert_Id: {} '.format(current_alert))
# Update Alert (alert contailns LAST status)
u = sfdc_client.update_mos_alert(id=Id, data=data)
LOG.debug('Upate status code: {} '.format(u.status_code))
# Add comment to updated alert
comment_data['related_id__c'] = Id
comment_data['MosAlertId__c'] = Id
add_comment = sfdc_client.create_mos_alert_comment(comment_data)
LOG.debug('Add Comment status code: {} '.format(add_comment.status_code))
LOG.debug('Add Comment data: {} '.format(add_comment.text))
elif (new_alert.status_code == 201):
# Add commnet, because MOS_Alert is LAST data and will be overriden on update
Id = new_alert.json()['id']
comment_data['related_id__c'] = Id
comment_data['MosAlertId__c'] = Id
current_alert = sfdc_client.get_mos_alert(Id).json()
comment_data['MOS_Alert_Name__c'] = current_alert['Name']
add_comment = sfdc_client.create_mos_alert_comment(comment_data)
LOG.debug('Add Comment status code: {} '.format(add_comment.status_code))
LOG.debug('Add Comment data: {} '.format(add_comment.text))
def main():
parser = ArgumentParser()
parser.add_argument('--webhook-url',
required=True,
help="Slack webhook url")
parser.add_argument('--desc', required=True, help='use "-" to use stdin')
parser.add_argument('--notification-type',
required=True,
help='PROBLEM|RECOVERY|etc ($NOTIFICATIONTYPE$)')
parser.add_argument('--state',
required=True,
help='OK|WARNING|CRITICAL|UNKNOWN|UP|DOWN|UNREACHABLE'
' $SERVICESTATE$ or $HOSTSTATE$')
parser.add_argument('--hostname', required=True, help='$HOSTNAME$')
parser.add_argument('--service-desc', required=False, help='$SERVICEDESC$')
parser.add_argument('--syslog',
action='store_true',
default=False,
help='Log to syslog')
parser.add_argument('--debug', action='store_true', default=False)
parser.add_argument('--log-file',
default=sys.stdout,
help='default stdout')
args = parser.parse_args()
LOG = logging.getLogger()
if args.syslog:
handler = logging.SysLogHandler()
elif (args.log_file != sys.stdout):
handler = logging.FileHandler(args.log_file)
else:
handler = logging.StreamHandler(sys.stdout)
if args.debug:
log_level = logging.DEBUG
else:
log_level = logging.INFO
formatter = logging.Formatter(
'nagios_to_slack %(asctime)s %(process)d %(levelname)s %(name)s '
'[-] %(message)s',
datefmt='%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
LOG.setLevel(log_level)
LOG.addHandler(handler)
if args.desc == '-':
args.desc = ''.join(sys.stdin.readlines())
state_color = {
'OK': 'good',
'UNKNOWN': 'gray',
'WARNING': 'warning',
'CRITICAL': 'danger',
'UP': 'good',
'DOWN': 'danger',
'UNREACHABLE': 'danger',
}
if args.service_desc:
title = '{:s} {:s}/{:s}'.format(args.state, args.hostname,
args.service_desc)
else:
title = '{:s} {:s}'.format(args.state, args.hostname)
slack_data = {
'color':
state_color[args.state],
'icon_emoji':
':ghost:',
'fields': [{
'title': args.notification_type,
'value': title
}, {
'title': 'Description',
'value': args.desc
}]
}
LOG.debug('Nagios data: {} '.format(slack_data))
r = requests.post(args.webhook_url, data=json.dumps(slack_data))
LOG.debug("Response: {:d} {:s}".format(r.status_code, r.text))
def main():
parser = ArgumentParser()
parser.add_argument('-c', '--config-file', default='config.yml')
parser.add_argument('--syslog',
action='store_true',
default=False,
help='Log to syslog')
parser.add_argument('--debug',
action='store_true',
default=False,
help='Enable debug log level')
parser.add_argument(
'--log_file',
default=sys.stdout,
help=
'Log file. default: stdout. Ignored if logging configured to syslog')
args = parser.parse_args()
LOG = logging.getLogger()
if args.syslog:
handler = logging.SysLogHandler()
elif (args.log_file != sys.stdout):
handler = logging.FileHandler(args.log_file)
else:
handler = logging.StreamHandler(sys.stdout)
if args.debug:
log_level = logging.DEBUG
else:
log_level = logging.INFO
formatter = logging.Formatter(
'{} nagios_to_sfdc %(asctime)s %(process)d %(levelname)s %(name)s '
'[-] %(message)s'.format(socket.getfqdn()),
datefmt='%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
LOG.setLevel(log_level)
LOG.addHandler(handler)
with open(args.config_file) as fp:
config = yaml.load(fp)
if 'sfdc_organization_id' in config:
organizationId = config['sfdc_organization_id']
else:
organizationId = None
sfdc_oauth2 = OAuth2(client_id=config['sfdc_client_id'],
client_secret=config['sfdc_client_secret'],
username=config['sfdc_username'],
password=config['sfdc_password'],
auth_url=config['sfdc_auth_url'],
organizationId=organizationId)
sfdc_client = Client(sfdc_oauth2)
# Serach example
# a=sfdc_client.search("SELECT Id from Case")
# for b in a:
# print(b['Id'])
# sfdc_client.get_case(b['Id'])
a = sfdc_client.search("SELECT Id from MOS_Alerts__c")
for b in a:
print(b['Id'])
alert = sfdc_client.get_mos_alert(b['Id'])
print(
json.dumps(alert.json(),
sort_keys=True,
indent=4,
separators=(',', ': ')))
sfdc_client.del_mos_alert(b['Id'])
a = sfdc_client.search("SELECT Id from MOS_Alert_Comment__c")
for b in a:
print(b['Id'])
a = sfdc_client.del_mos_alert_comment(b['Id'])
# alert = sfdc_client.get_mos_alert_comment(b['Id'])
# print(json.dumps(alert.json(), sort_keys=True, indent=4, separators=(',', ': ') ))
#print(a.status_code)
print(a.text)
format = "%(asctime)s %(filename)s:%(lineno)-3d %(levelname)s %(message)s"
formatter = logging.Formatter(format)
run_time = t.asctime(t.localtime()).replace(' ', '_')
file_handler = logging.FileHandler('simple_' + run_time + '.log')
file_handler.setLevel(logging.WARNING)
file_handler.setFormatter(formatter)
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.DEBUG)
console_handler.setFormatter(formatter)
server_format = "%(filename)s:%(lineno)-3d %(levelname)s %(message)s"
server_formatter = logging.Formatter(server_format)
server_handler = logging.SysLogHandler()
server_handler.setLevel(logging.ERROR)
server_handler.setFormatter(server_formatter)
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
logger.addHandler(file_handler)
logger.addHandler(console_handler)
def my_fun(n):
for i in range(0, n):
logging.debug(i)
if i == 50:
logging.warning("The value of i is 50.")
try:
def main():
parser = ArgumentParser()
parser.add_argument('-c', '--config-file', default='config.yml')
parser.add_argument('--description', required=True,
help='Description (use "-" to use stdin)')
parser.add_argument('--notification_type', required=True,
help='Notification type (PROBLEM|RECOVERY|CUSTOM). '
'Nagios variable - $NOTIFICATIONTYPE$')
parser.add_argument('--state', required=True,
help='(OK|UP|WARNING|CRITICAL|UNKNOWN|UNREACHABLE). '
'Nagios variable - $SERVICESTATE$ or $HOSTSTATE$')
parser.add_argument('--host_name', required=True,
help='Nagios variable - $HOSTNAME$')
parser.add_argument('--service_description', required=False,
help='Nagios variable - $SERVICEDESC$')
parser.add_argument('--long_date_time', required=True,
help='Nagios variable - $LONGDATETIME$')
parser.add_argument('--syslog', action='store_true', default=False,
help='Log to syslog')
parser.add_argument('--debug', action='store_true', default=False,
help='Enable debug log level')
parser.add_argument('--log_file', default=sys.stdout,
help='Log file. default: stdout. Ignored if '
'logging configured to syslog')
args = parser.parse_args()
LOG = logging.getLogger()
if args.syslog:
handler = logging.SysLogHandler()
elif (args.log_file != sys.stdout):
handler = logging.FileHandler(args.log_file)
else:
handler = logging.StreamHandler(sys.stdout)
if args.debug:
log_level = logging.DEBUG
else:
log_level = logging.INFO
formatter = logging.Formatter(
'{} nagios_to_sfdc %(asctime)s %(process)d %(levelname)s %(name)s '
'[-] %(message)s'.format(socket.getfqdn()),
datefmt='%Y-%m-%d %H:%M:%S'
)
handler.setFormatter(formatter)
LOG.setLevel(log_level)
LOG.addHandler(handler)
# Read from stdin if desctiption defined as '-'
if args.description == '-':
args.description = ''.join(sys.stdin.readlines())
# state are mapped to priority
state = {
'OK': '060 Informational',
'UP': '060 Informational',
'UNKNOWN': '070 Unknown',
'WARNING': '080 Warning',
'CRITICAL': '090 Critical',
'DOWN': '090 Critical',
'UNREACHABLE': '090 Critical',
}
nagios_data = {
'state': state[str(args.state).upper()],
'notification_type': args.notification_type,
'description': args.description,
'host_name': args.host_name,
'long_date_time': args.long_date_time,
}
if args.service_description:
nagios_data['service_description'] = args.service_description
else:
nagios_data['service_description'] = ''
LOG.debug('Nagios data: {} '.format(nagios_data))
with open(args.config_file) as fp:
config = yaml.load(fp)
if 'sfdc_organization_id' in config:
organizationId = config['sfdc_organization_id']
else:
organizationId = None
sfdc_oauth2 = OAuth2(client_id=config['sfdc_client_id'],
client_secret=config['sfdc_client_secret'],
username=config['sfdc_username'],
password=config['sfdc_password'],
auth_url=config['sfdc_auth_url'],
organizationId=organizationId)
environment = config['environment']
# Alert ID shoud be uniq for env
Alert_ID = '{}--{}'.format(environment, args.host_name)
if args.service_description:
nagios_data['service_description'] = args.service_description
Alert_ID = '{}--{}'.format(Alert_ID, args.service_description)
subject = "{}/{}".format(args.host_name, args.service_description)
else:
subject = "{}/host_status".format(args.host_name)
LOG.debug('Alert_Id: {} '.format(Alert_ID))
sfdc_client = Client(sfdc_oauth2)
# read cached token if it exists
try:
with open(config['token_cache_file'], 'r') as fp:
cached_tokens = yaml.load(fp)
except IOError:
cached_tokens = None
if cached_tokens:
sfdc_client.access_token = cached_tokens['access']
sfdc_client.instance_url = cached_tokens['instance_url']
test_response = sfdc_client.get_case('case_id_that_doesnt_exist')
if test_response.status_code == 401:
# If auth fails, reset tokens to None to force re-auth.
sfdc_client.access_token = None
sfdc_client.instance_url = None
LOG.debug('Cached access token expired. Going to re-auth.')
else:
LOG.debug('Using cached access token.')
#payload = {
# 'notification_type': args.notification_type,
# 'description': args.description,
# 'long_date_time': args.long_date_time,
#}
payload = args.description
data = {
'IsMosAlert__c': 'true',
'Description': payload,
'Alert_ID__c': Alert_ID,
'Subject': subject,
'Environment2__c': environment,
'Alert_Priority__c': nagios_data['state'],
'Alert_Host__c': nagios_data['host_name'],
'Alert_Service__c': nagios_data['service_description'],
}
feed_data_body = {
'Description': payload,
#'Alert_Id': Alert_ID,
#'Cloud_ID': environment,
#'Alert_Priority': nagios_data['state'],
'Status': "New",
}
#feed_data_body = payload
try:
new_case = sfdc_client.create_case(data)
except Exception as E:
LOG.debug(E)
sys.exit(1)
LOG.debug('New Case status code: {} '.format(new_case.status_code))
LOG.debug('New Case data: {} '.format(new_case.text))
# If Case exist
if (new_case.status_code == 400) and \
(new_case.json()[0]['errorCode'] == 'DUPLICATE_VALUE'):
LOG.debug('Code: {}, Error message: {} '.format(new_case.status_code,
new_case.text))
# Find Case ID
ExistingCaseId = new_case.json()[0]['message'].split(" ")[-1]
LOG.debug('ExistingCaseId: {} '.format(ExistingCaseId))
# Get Case
current_case = sfdc_client.get_case(ExistingCaseId).json()
LOG.debug("Existing Case: \n {}".format(json.dumps(current_case,
sort_keys=True, indent=4)))
LastModifiedDate = current_case['LastModifiedDate']
ExistingCaseStatus = current_case['Status']
feed_data_body['Status'] = ExistingCaseStatus
Now = datetime.now().replace(tzinfo=None)
delta = Now - dateutil.parser.parse(LastModifiedDate).replace(tzinfo=None)
LOG.debug("Check if Case should be marked as OUTDATED. Case "
"modification date is: {} , Now: {} , Delta(sec): {}, "
"OutdateDelta(sec): {}".format(LastModifiedDate, Now,
delta.seconds, DELTA_SECONDS))
if (delta.seconds > DELTA_SECONDS):
# Old Case is outdated
tmp_date = datetime.strftime(datetime.now(), "%Y.%m.%d-%H:%M:%S")
new_data = {
'Alert_Id__c': '{}_closed_at_{}'.format(current_case['Alert_ID__c'],
tmp_date),
'Alert_Priority__c': '000 OUTDATED',
}
u = sfdc_client.update_case(id=ExistingCaseId, data=new_data)
LOG.debug('Update status code: {} \n\nUpdate content: {}'
'\n\n Update headers: {}\n\n'.format(u.status_code,
u.content,
u.headers))
# Try to create new case again
try:
new_case = sfdc_client.create_case(data)
except Exception as E:
LOG.debug(E)
sys.exit(1)
else:
# Case was outdated an new was created
CaseId = new_case.json()['id']
LOG.debug("Case was just created, old one marked as Outdated")
# Add comment, because Case head should contains LAST data
# overwritten on any update
CaseId = new_case.json()['id']
feeditem_data = {
'ParentId': CaseId,
'Visibility': 'AllUsers',
#'Body': json.dumps(feed_data_body, sort_keys=True, indent=4),
'Body': format_feed_body(feed_data_body),
}
LOG.debug("FeedItem Data: {}".format(json.dumps(feeditem_data,
sort_keys=True,
indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug('Add FeedItem status code: {}\nAdd FeedItem '
'reply: {}'.format(add_feed_item.status_code,
add_feed_item.text))
else:
# Update Case
# If ok, mark case as solved.
if args.state in ("OK", "UP"):
data['Status'] = 'Auto-solved'
feed_data_body['Status'] = 'Auto-solved'
u = sfdc_client.update_case(id=ExistingCaseId, data=data)
LOG.debug('Upate status code: {} '.format(u.status_code))
feeditem_data = {
'ParentId': ExistingCaseId,
'Visibility': 'AllUsers',
#'Body': json.dumps(feed_data_body, sort_keys=True, indent=4),
'Body': format_feed_body(feed_data_body),
}
LOG.debug("FeedItem Data: {}".format(json.dumps(feeditem_data,
sort_keys=True,
indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug('Add FeedItem status code: {}\nAdd FeedItem '
'reply: {} '.format(add_feed_item.status_code,
add_feed_item.text))
# Else If Case did not exist before and was just created
elif (new_case.status_code == 201):
LOG.debug("Case was just created")
# Add comment, because Case head should contain LAST data
# overwritten on any update
CaseId = new_case.json()['id']
# If OK, ensure "Auto-solved" is in the first feed.
if args.state in ("OK", "UP"):
feed_data_body['Status'] = 'Auto-solved'
feeditem_data = {
'ParentId': CaseId,
'Visibility': 'AllUsers',
#'Body': json.dumps(feed_data_body, sort_keys=True, indent=4),
'Body': format_feed_body(feed_data_body),
}
LOG.debug("FeedItem Data: {}".format(json.dumps(feeditem_data,
sort_keys=True,
indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug('Add FeedItem status code: {}\nAdd FeedItem '
'reply: {} '.format(add_feed_item.status_code,
add_feed_item.text))
# If OK, mark case as solved.
if args.state in ("OK", "UP"):
data['Status'] = 'Auto-solved'
u = sfdc_client.update_case(id=CaseId, data=data)
LOG.debug('Update status code: {} '.format(u.status_code))
else:
LOG.debug("Unexpected error: Case was not created (code !=201) "
"and Case does not exist (code != 400)")
sys.exit(1)
# Write out token/instance_url
with open(config['token_cache_file'], 'w') as fp:
fp.write("access: {}\n".format(sfdc_client.access_token))
fp.write("instance_url: {}\n".format(sfdc_client.instance_url))
def main():
parser = ArgumentParser()
parser.add_argument('-c', '--config-file', default='config.yml')
parser.add_argument('--description',
required=True,
help='Description (use "-" to use stdin)')
parser.add_argument(
'--notification_type',
required=True,
help=
'Notification type (PROBLEM|RECOVERY|CUSTOM). Nagios variable - $NOTIFICATIONTYPE$" '
)
parser.add_argument(
'--state',
required=True,
help=
'Service ot Host (OK|WARNING|CRITICAL|UNCKNOWN). Nagios variable - $SERVICESTATE$" or $HOSTSTATE$ '
)
parser.add_argument('--host_name',
required=True,
help='Host name. Nagios variable - $HOSTNAME$')
parser.add_argument(
'--service_description',
required=False,
help='Service Description. Nagios variable - $SERVICEDESC$')
parser.add_argument('--long_date_time',
required=True,
help='Date and time. Nagios variable - $LONGDATETIME$')
parser.add_argument('--syslog',
action='store_true',
default=False,
help='Log to syslog')
parser.add_argument('--debug',
action='store_true',
default=False,
help='Enable debug log level')
parser.add_argument(
'--log_file',
default=sys.stdout,
help=
'Log file. default: stdout. Ignored if logging configured to syslog')
args = parser.parse_args()
LOG = logging.getLogger()
if args.syslog:
handler = logging.SysLogHandler()
elif (args.log_file != sys.stdout):
handler = logging.FileHandler(args.log_file)
else:
handler = logging.StreamHandler(sys.stdout)
if args.debug:
log_level = logging.DEBUG
else:
log_level = logging.INFO
formatter = logging.Formatter(
'{} nagios_to_sfdc %(asctime)s %(process)d %(levelname)s %(name)s '
'[-] %(message)s'.format(socket.getfqdn()),
datefmt='%Y-%m-%d %H:%M:%S')
handler.setFormatter(formatter)
LOG.setLevel(log_level)
LOG.addHandler(handler)
# Read from stdin if desctiption defined as '-'
if args.description == '-':
args.description = ''.join(sys.stdin.readlines())
# state are mapped to priority
state = {
'OK': '060 Informational',
'UNKNOWN': '070 Unknown',
'WARNING': '080 Warning',
'CRITICAL': '090 Critical',
}
nagios_data = {
'state': state[str(args.state).upper()],
'notification_type': args.notification_type,
'description': args.description,
'host_name': args.host_name,
'long_date_time': args.long_date_time,
}
if args.service_description:
nagios_data['service_description'] = args.service_description
else:
nagios_data['service_description'] = ''
LOG.debug('Nagios data: {} '.format(nagios_data))
with open(args.config_file) as fp:
config = yaml.load(fp)
if 'sfdc_organization_id' in config:
organizationId = config['sfdc_organization_id']
else:
organizationId = None
sfdc_oauth2 = OAuth2(client_id=config['sfdc_client_id'],
client_secret=config['sfdc_client_secret'],
username=config['sfdc_username'],
password=config['sfdc_password'],
auth_url=config['sfdc_auth_url'],
organizationId=organizationId)
environment = config['environment']
# Alert ID shoud be uniq for env
Alert_ID = '{}--{}'.format(environment, args.host_name)
if args.service_description:
nagios_data['service_description'] = args.service_description
Alert_ID = '{}--{}'.format(Alert_ID, args.service_description)
LOG.debug('Alert_Id: {} '.format(Alert_ID))
sfdc_client = Client(sfdc_oauth2)
payload = {
'notification_type': args.notification_type,
'description': args.description,
'long_date_time': args.long_date_time,
}
data = {
'IsMosAlert__c': 'true',
'Description': json.dumps(payload, sort_keys=True, indent=4),
'Alert_ID__c': Alert_ID,
'Subject': Alert_ID,
'Environment2__c': environment,
'Alert_Priority__c': nagios_data['state'],
'Alert_Host__c': nagios_data['host_name'],
'Alert_Service__c': nagios_data['service_description'],
# 'sort_marker__c': sort_marker,
}
feed_data_body = {
'Description': payload,
'Alert_Id': Alert_ID,
'Cloud_ID': environment,
'Alert_Priority': nagios_data['state'],
'Status': "New",
}
try:
new_case = sfdc_client.create_case(data)
except Exception as E:
LOG.debug(E)
sys.exit(1)
# print(new_case)
LOG.debug('New Caset status code: {} '.format(new_case.status_code))
LOG.debug('New Case data: {} '.format(new_case.text))
# If Case exist
if (new_case.status_code == 400) and (new_case.json()[0]['errorCode']
== 'DUPLICATE_VALUE'):
LOG.debug('Code: {}, Error message: {} '.format(
new_case.status_code, new_case.text))
# Find Case ID
ExistingCaseId = new_case.json()[0]['message'].split(" ")[-1]
LOG.debug('ExistingCaseId: {} '.format(ExistingCaseId))
# Get Case
current_case = sfdc_client.get_case(ExistingCaseId).json()
LOG.debug("Existing Case: \n {}".format(
json.dumps(current_case, sort_keys=True, indent=4)))
LastModifiedDate = current_case['LastModifiedDate']
ExistingCaseStatus = current_case['Status']
feed_data_body['Status'] = ExistingCaseStatus
Now = datetime.now().replace(tzinfo=None)
delta = Now - dateutil.parser.parse(LastModifiedDate).replace(
tzinfo=None)
LOG.debug(
"Check if Case should be marked as OUTDATED. Case modification date is: {} , Now: {} , Delta(sec): {}, OutdateDelta(sec): {}"
.format(LastModifiedDate, Now, delta.seconds, DELTA_SECONDS))
if (delta.seconds > DELTA_SECONDS):
# Old Case is outdated
new_data = {
'Alert_Id__c':
'{}_closed_at_{}'.format(
current_case['Alert_ID__c'],
datetime.strftime(datetime.now(), "%Y.%m.%d-%H:%M:%S")),
'Alert_Priority__c':
'000 OUTDATED',
}
u = sfdc_client.update_case(id=ExistingCaseId, data=new_data)
LOG.debug(
'Upate status code: {} \n\nUpate content: {}\n\n Upate headers: {}\n\n'
.format(u.status_code, u.content, u.headers))
# Try to create new caset again
try:
new_case = sfdc_client.create_case(data)
except Exception as E:
LOG.debug(E)
sys.exit(1)
else:
# Case was outdated an new was created
CaseId = new_case.json()['id']
LOG.debug(
"Case was just created, old one was marked as Outdated")
# Add commnet, because Case head should conains LAST data overriden on any update
CaseId = new_case.json()['id']
feeditem_data = {
'ParentId': CaseId,
'Visibility': 'AllUsers',
'Body': json.dumps(feed_data_body,
sort_keys=True,
indent=4),
}
LOG.debug("FeedItem Data: {}".format(
json.dumps(feeditem_data, sort_keys=True, indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug(
'Add FeedItem status code: {} \n Add FeedItem reply: {} '.
format(add_feed_item.status_code, add_feed_item.text))
else:
# Update Case
u = sfdc_client.update_case(id=ExistingCaseId, data=data)
LOG.debug('Upate status code: {} '.format(u.status_code))
feeditem_data = {
'ParentId': ExistingCaseId,
'Visibility': 'AllUsers',
'Body': json.dumps(feed_data_body, sort_keys=True, indent=4),
}
LOG.debug("FeedItem Data: {}".format(
json.dumps(feeditem_data, sort_keys=True, indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug(
'Add FeedItem status code: {} \n Add FeedItem reply: {} '.
format(add_feed_item.status_code, add_feed_item.text))
# Else If Case did not exist before and was just created
elif (new_case.status_code == 201):
LOG.debug("Case was just created")
# Add commnet, because Case head should conains LAST data overriden on any update
CaseId = new_case.json()['id']
feeditem_data = {
'ParentId': CaseId,
'Visibility': 'AllUsers',
'Body': json.dumps(feed_data_body, sort_keys=True, indent=4),
}
LOG.debug("FeedItem Data: {}".format(
json.dumps(feeditem_data, sort_keys=True, indent=4)))
add_feed_item = sfdc_client.create_feeditem(feeditem_data)
LOG.debug(
'Add FeedItem status code: {} \n Add FeedItem reply: {} '.format(
add_feed_item.status_code, add_feed_item.text))
else:
LOG.debug(
"Unexpected error: Case was not created (code !=201) and Case does not exist (code != 400)"
)
sys.exit(1)
