
import 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)