def set_verbose_log(self, verbose):
if (verbose == "1" or verbose == 1):
self.log("Enable verbose log")
LoggerInit(self._context._log_file, '/dev/stdout', verbose=True)
else:
self.log("Disable verbose log")
LoggerInit(self._context._log_file, '/dev/stdout', verbose=False)
def main():
#Global Variables definition
global waagent
waagent = imp.load_source('waagent', '/usr/sbin/waagent')
from waagent import LoggerInit
LoggerInit('/var/log/waagent.log', '/dev/stdout')
global DownloadDirectory
DownloadDirectory = 'download'
global ExtensionShortName
ExtensionShortName = 'CustomScript'
global Util
Util = imp.load_source('HandlerUtil', './resources/HandlerUtil.py')
waagent.Log("%s started to handle." % (ExtensionShortName))
for a in sys.argv[1:]:
if re.match("^([-/]*)(disable)", a):
disable()
elif re.match("^([-/]*)(uninstall)", a):
uninstall()
elif re.match("^([-/]*)(install)", a):
install()
elif re.match("^([-/]*)(enable)", a):
enable()
elif re.match("^([-/]*)(update)", a):
update()
def _change_log_file(self):
self.log("Change log file to " + self._context._log_file)
LoggerInit(self._context._log_file, '/dev/stdout')
self._log = waagent.Log
self._error = waagent.Error
"""
Example Azure Handler script for Linux IaaS
Install example
Reads port from Public Config if present.
Creates service_port.txt in resources dir.
"""
import os
import imp
import time
waagent = imp.load_source('waagent', '/usr/sbin/waagent')
from waagent import LoggerInit
hutil = imp.load_source('HandlerUtil', './resources/HandlerUtil.py')
LoggerInit('/var/log/waagent.log', '/dev/stdout')
waagent.Log("install.py starting.")
logfile = waagent.Log
name, seqNo, version, config_dir, log_dir, settings_file, status_file, heartbeat_file, config = hutil.doParse(
logfile, 'Install')
LoggerInit('/var/log/' + name + '_Install.log', '/dev/stdout')
waagent.Log(name + " - install.py starting.")
logfile = waagent.Log
hutil.doStatusReport(name, seqNo, version, status_file,
time.strftime("%Y-%M-%dT%H:%M:%SZ", time.gmtime()),
time.strftime("%Y-%M-%dT%H:%M:%SZ",
"""
Example Azure Handler script for Linux IaaS
Diable example
"""
import os
import imp
import time
import json
waagent=imp.load_source('waagent','/usr/sbin/waagent')
from waagent import LoggerInit
hutil=imp.load_source('HandlerUtil','./resources/HandlerUtil.py')
LoggerInit('/var/log/waagent.log','/dev/stdout')
waagent.Log("disable.py starting.")
logfile=waagent.Log
name,seqNo,version,config_dir,log_dir,settings_file,status_file,heartbeat_file,config=hutil.doParse(logfile,'Disable')
LoggerInit('/var/log/'+name+'_Disable.log','/dev/stdout')
waagent.Log(name+" - disable.py starting.")
logfile=waagent.Log
hutil.doStatusReport(name,seqNo,version,status_file,time.strftime("%Y-%M-%dT%H:%M:%SZ", time.gmtime()),
time.strftime("%Y-%M-%dT%H:%M:%SZ", time.gmtime()),name,
import os
import os.path
import re
import string
import subprocess
import sys
import imp
import shlex
import traceback
import urllib
from azure.storage import BlobService
waagent = imp.load_source('waagent', '/usr/sbin/waagent')
from waagent import LoggerInit
LoggerInit('/var/log/waagent.log', '/dev/stdout')
#Global Variables definition
DownloadDirectory = 'download'
ExtensionShortName = 'CustomScript'
Util = imp.load_source('HandlerUtil', './resources/HandlerUtil.py')
waagent.Log("%s started to handle." % (ExtensionShortName))
def main():
for a in sys.argv[1:]:
if re.match("^([-/]*)(disable)", a):
disable()
elif re.match("^([-/]*)(uninstall)", a):
uninstall()
elif re.match("^([-/]*)(install)", a):
#!/usr/bin/env python
"""
Example Azure Handler script for Linux IaaS
Heartbeat example
"""
import os
import imp
import time
waagent = imp.load_source('waagent', '/usr/sbin/waagent')
from waagent import LoggerInit
hutil = imp.load_source('HandlerUtil', './resources/HandlerUtil.py')
LoggerInit('/var/log/waagent.log', '/dev/stdout')
waagent.Log("hearbeat.py starting.")
logfile = waagent.Log
name, seqNo, version, config_dir, log_dir, settings_file, status_file, heartbeat_file, config = hutil.doParse(
logfile, 'Hearbeat')
LoggerInit('/var/log/' + name + '_Hearbeat.log', '/dev/stdout')
waagent.Log(name + " - hearbeat.py starting.")
logfile = waagent.Log
pid = None
pidfile = './service_pid.txt'
retries = 5
waagent.SetFileContents('./heartbeat.pid', str(os.getpid()))
Reads port from Public Config if present.
Creates service_port.txt in resources dir.
Copies the service to /usr/bin and updates it
with the resource path.
"""
import os
import sys
import imp
import time
waagent = imp.load_source('waagent', '/usr/sbin/waagent')
from waagent import LoggerInit
hutil = imp.load_source('HandlerUtil', './resources/HandlerUtil.py')
LoggerInit('/var/log/waagent.log', '/dev/stdout')
waagent.Log("update.py starting.")
waagent.MyDistro = waagent.GetMyDistro()
logfile = waagent.Log
name, seqNo, version, config_dir, log_dir, settings_file, status_file, heartbeat_file, config = hutil.doParse(
logfile, 'Update')
LoggerInit('/var/log/' + name + '_Update.log', '/dev/stdout')
waagent.Log(name + " - update.py starting.")
logfile = waagent.Log
hutil.doStatusReport(name, seqNo, version, status_file,
time.strftime("%Y-%M-%dT%H:%M:%SZ", time.gmtime()),
elif '--test_vm_acct_pass' == sys.argv[i] : provisionedVMpass=sys.argv[i+1]
elif '--test_vm_acct_cert' == sys.argv[i] : provisionedVMCert=sys.argv[i+1]
elif '--azure_location' == sys.argv[i] : location=sys.argv[i+1]
elif '--mount_opts' == sys.argv[i] : mountOptions=sys.argv[i+1]
elif '--part_num' == sys.argv[i] : partNum=sys.argv[i+1]
elif '--retries' == sys.argv[i] : provision_retries=int(sys.argv[i+1])
elif '--fs_type' == sys.argv[i] : fstype=sys.argv[i+1]
elif '--keep_test_vm_vhd' == sys.argv[i] : keep_vhd=sys.argv[i+1]
elif '--teardown_test_vm' == sys.argv[i] : teardown_test_vm=sys.argv[i+1]
elif '--teardown_stable_vm' == sys.argv[i] : teardown_stable_vm=sys.argv[i+1]
elif '--prompt' == sys.argv[i] : prompt=sys.argv[i+1]
elif '--stable_vm_image' == sys.argv[i] : stable_vm_image=sys.argv[i+1]
elif '--stable_vm_vhd' == sys.argv[i] : stable_vm_vhd=sys.argv[i+1]
elif '--logfile' == sys.argv[i] : logfile=sys.argv[i+1]
LoggerInit(logfile,'')
waagent.Log("User: "******" Running Command :\n" + reduce(lambda x, y: x+' '+y,sys.argv))
if len(stableVM) == 0 and not ( stable_vm_image or stable_vm_vhd ):
print '--vm <stable vm> must be provided unless --stable_vm_image or --stable_vm_vhd'
waagent.Log( '--vm <stable vm> must be provided!')
sys.exit(1)
else:
if stable_vm_image:
sourceVHD=createStableVMFromVMImage(stable_vm_image,sourceVHD)
stableVM=testname+'-stable-vm'
elif stable_vm_vhd:
stableVMImageName=testname+'-stable-vi'
sourceVHD=createStableVMFromVHD(stableVMImageName,stable_vm_vhd)
stableVM=testname+'-stable-vm'
p = False
