Add __main__ module to Python's egg (#1418)

* Revert "Remove __main__ module" (commit a85e6cd).

* Add __main__ to Python's egg

__main__.py

@@ -0,0 +1,20 @@
+# Copyright 2018 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.6+ and Openssl 1.0+
+#
+
+import azurelinuxagent.agent as agent
+
+agent.main()

setup.py

@@ -221,6 +221,11 @@ def run(self):
 if float(sys.version[:3]) >= 3.7:
     requires = ['distro']
 
+modules = []
+
+if "bdist_egg" in sys.argv:
+    modules.append("__main__")
+
 setuptools.setup(
     name=AGENT_NAME,
     version=AGENT_VERSION,
@@ -231,6 +236,7 @@ def run(self):
     url='https://github.com/Azure/WALinuxAgent',
     license='Apache License Version 2.0',
     packages=find_packages(exclude=["tests*"]),
+    py_modules=modules,
     install_requires=requires,
     cmdclass={
         'install': install