import fcntl
import time
import sys
import os, os.path
update_config_file = '/etc/panopta-agent/panopta_agent.cfg'
if not os.path.exists(update_config_file):
home = os.path.expanduser("~")
update_config_file = os.path.join(home, "lib", "fm-agent", "update_config")
unlock_attempts = 10
def update_config(manifest_file):
if not os.path.exists(manifest_file):
print "Manifest file %s does not exist" % manifest_file
sys.exit(0)
try:
mf = open(manifest_file).read().strip().split('\n')
props = dict([map(str.strip, line.split("=")) for line in mf])
except:
print "Error reading manifest file %s" % manifest_file
sys.exit(0)
manfile = open(update_config_file, 'wt')
# Acquire lock
locked = True
for i in range(unlock_attempts):
try:
fcntl.flock(manfile, fcntl.LOCK_EX | fcntl.LOCK_NB)
locked = False
break
except:
time.sleep(1.0)
if locked:
print "Could not unlock %s after %s attempts" % (update_config_file, unlock_attempts)
sys.exit(1)
# Write new properties
manfile.write(open(manifest_file).read())
# Release lock
fcntl.flock(manfile, fcntl.LOCK_UN)
manfile.close()
# Change perms to make sure the agent user can read the config file
os.system("chmod 666 %s" % update_config_file)
if __name__ == "__main__":
if len(sys.argv) < 2:
print "Please specify the manifest file"
sys.exit(0)
else:
update_config(sys.argv[1])