usr/bin/fm-agent/report.py000064400000003541147204511270011516 0ustar00import fcntl import time import sys import os, os.path import csv from datetime import datetime if 'freebsd' in sys.platform.lower(): report_file = "/usr/local/fm-agent/lib/report" elif 'darwin' == sys.platform.lower(): register_file = '/usr/local/FortiMonitor/agent/share/fm-agent/report' else: report_file = '/var/lib/fm-agent/report' if not os.path.exists(report_file): home = os.path.expanduser("~") report_file = os.path.join(home, "lib", "fm-agent", "report") unlock_attempts = 10 def report_metric(textkey, value, timestamp): csvfile = open(report_file, 'a') # Acquire lock locked = True for i in range(unlock_attempts): try: fcntl.flock(csvfile, fcntl.LOCK_EX | fcntl.LOCK_NB) locked = False break except: time.sleep(1.0) if locked: print("Could not unlock %s after %s attempts" % (report_file, unlock_attempts)) sys.exit(1) # Append new metric value csvwriter = csv.writer(csvfile) csvwriter.writerow([textkey, value, timestamp.strftime("%Y-%m-%d %H:%M:%S")]) # Release lock fcntl.flock(csvfile, fcntl.LOCK_UN) csvfile.close() if __name__ == "__main__": if len(sys.argv) < 3: print("Please specify textkey, value, and optionally UTC timestamp in ISO format") print('Example: python report.py metric.key 3.5 "2015-03-24 10:11:12"') sys.exit(0) else: if len(sys.argv) >= 4: try: timestamp = datetime.strptime(sys.argv[3], "%Y-%m-%d %H:%M:%S") except: print("Cannot parse timestamp") sys.exit(0) else: timestamp = datetime.now() try: value = float(sys.argv[2]) except: print("Value should be a number") sys.exit(0) report_metric(sys.argv[1], value, timestamp)