summaryrefslogtreecommitdiff
path: root/azurelinuxagent/agent.py
diff options
context:
space:
mode:
authorBen Howard <ben.howard@ubuntu.com>2015-08-14 16:40:41 -0600
committerusd-importer <ubuntu-server@lists.ubuntu.com>2015-08-15 14:33:21 +0000
commitf78b9650d0e7b008d430673a075aad95dda863be (patch)
treea6749619e78483d45a66d4bad4d6e922391541fc /azurelinuxagent/agent.py
parent0afc048f2a6ff3638ecfa33e7ded5dc8dddf041a (diff)
downloadvyos-walinuxagent-f78b9650d0e7b008d430673a075aad95dda863be.tar.gz
vyos-walinuxagent-f78b9650d0e7b008d430673a075aad95dda863be.zip
Import patches-unapplied version 2.1.1-0ubuntu1 to ubuntu/wily-proposed
Imported using git-ubuntu import. Changelog parent: 0afc048f2a6ff3638ecfa33e7ded5dc8dddf041a New changelog entries: * New upstream release for Ubuntu. - Switch to Python3 - Applies Ubuntu specific patches
Diffstat (limited to 'azurelinuxagent/agent.py')
-rw-r--r--azurelinuxagent/agent.py141
1 files changed, 141 insertions, 0 deletions
diff --git a/azurelinuxagent/agent.py b/azurelinuxagent/agent.py
new file mode 100644
index 0000000..5e61a6c
--- /dev/null
+++ b/azurelinuxagent/agent.py
@@ -0,0 +1,141 @@
+# Windows Azure Linux Agent
+#
+# Copyright 2014 Microsoft Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Requires Python 2.4+ and Openssl 1.0+
+#
+
+"""
+Module agent
+"""
+
+import os
+import sys
+import re
+import subprocess
+from azurelinuxagent.metadata import AGENT_NAME, AGENT_LONG_VERSION, \
+ DISTRO_NAME, DISTRO_VERSION, \
+ PY_VERSION_MAJOR, PY_VERSION_MINOR, \
+ PY_VERSION_MICRO
+from azurelinuxagent.utils.osutil import OSUTIL
+from azurelinuxagent.handler import HANDLERS
+
+
+def init(verbose):
+ """
+ Initialize agent running environment.
+ """
+ HANDLERS.init_handler.init(verbose)
+
+def run():
+ """
+ Run agent daemon
+ """
+ HANDLERS.main_handler.run()
+
+def deprovision(force=False, deluser=False):
+ """
+ Run deprovision command
+ """
+ HANDLERS.deprovision_handler.deprovision(force=force, deluser=deluser)
+
+def parse_args(sys_args):
+ """
+ Parse command line arguments
+ """
+ cmd = "help"
+ force = False
+ verbose = False
+ for a in sys_args:
+ if re.match("^([-/]*)deprovision\\+user", a):
+ cmd = "deprovision+user"
+ elif re.match("^([-/]*)deprovision", a):
+ cmd = "deprovision"
+ elif re.match("^([-/]*)daemon", a):
+ cmd = "daemon"
+ elif re.match("^([-/]*)start", a):
+ cmd = "start"
+ elif re.match("^([-/]*)register-service", a):
+ cmd = "register-service"
+ elif re.match("^([-/]*)version", a):
+ cmd = "version"
+ elif re.match("^([-/]*)verbose", a):
+ verbose = True
+ elif re.match("^([-/]*)force", a):
+ force = True
+ elif re.match("^([-/]*)(help|usage|\\?)", a):
+ cmd = "help"
+ else:
+ cmd = "help"
+ break
+ return cmd, force, verbose
+
+def version():
+ """
+ Show agent version
+ """
+ print(("{0} running on {1} {2}".format(AGENT_LONG_VERSION, DISTRO_NAME,
+ DISTRO_VERSION)))
+ print("Python: {0}.{1}.{2}".format(PY_VERSION_MAJOR, PY_VERSION_MINOR,
+ PY_VERSION_MICRO))
+def usage():
+ """
+ Show agent usage
+ """
+ print("")
+ print((("usage: {0} [-verbose] [-force] [-help]"
+ "-deprovision[+user]|-register-service|-version|-daemon|-start]"
+ "").format(sys.argv[0])))
+ print("")
+
+def start():
+ """
+ Start agent daemon in a background process and set stdout/stderr to
+ /dev/null
+ """
+ devnull = open(os.devnull, 'w')
+ subprocess.Popen([sys.argv[0], '-daemon'], stdout=devnull, stderr=devnull)
+
+def register_service():
+ """
+ Register agent as a service
+ """
+ print("Register {0} service".format(AGENT_NAME))
+ OSUTIL.register_agent_service()
+ print("Start {0} service".format(AGENT_NAME))
+ OSUTIL.start_agent_service()
+
+def main():
+ """
+ Parse command line arguments, exit with usage() on error.
+ Invoke different methods according to different command
+ """
+ command, force, verbose = parse_args(sys.argv[1:])
+ if command == "version":
+ version()
+ elif command == "help":
+ usage()
+ else:
+ init(verbose)
+ if command == "deprovision+user":
+ deprovision(force, deluser=True)
+ elif command == "deprovision":
+ deprovision(force, deluser=False)
+ elif command == "start":
+ start()
+ elif command == "register-service":
+ register_service()
+ elif command == "daemon":
+ run()