Switch to an entry_point

trishnaguha · Oct 3, 2016 · 1ecc61f · 1ecc61f
1 parent 59efc6c
commit 1ecc61f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paasta_tools/cleanup_maintenance.py b/paasta_tools/cleanup_maintenance.py
@@ -72,7 +72,7 @@ def reserve_all_resources_on_draining_hosts():
     reserve_all_resources(hostnames=get_draining_hosts())
 
 
-def cleanup_maintenance():
+def main():
     log.debug("Cleaning up maintenance cruft")
     parse_args()
 
@@ -83,6 +83,6 @@ def cleanup_maintenance():
 
 
 if __name__ == "__main__":
-    if cleanup_maintenance():
+    if main():
         sys.exit(0)
     sys.exit(1)
diff --git a/setup.py b/setup.py
@@ -74,7 +74,6 @@
         'paasta_tools/autoscale_all_services.py',
         'paasta_tools/check_marathon_services_replication.py',
         'paasta_tools/cleanup_marathon_jobs.py',
-        'paasta_tools/cleanup_maintenance.py',
         'paasta_tools/paasta_deploy_chronos_jobs',
         'paasta_tools/deploy_marathon_services',
         'paasta_tools/generate_all_deployments',
@@ -102,6 +101,7 @@
         'paasta_list_chronos_jobs=paasta_tools.list_chronos_jobs:main',
         'paasta_setup_chronos_job=paasta_tools.setup_chronos_job:main',
         'paasta_chronos_rerun=paasta_tools.chronos_rerun:main',
+        'paasta_cleanup_maintenance=paasta_tools.cleanup_maintenance:main',
     ],
         'paste.app_factory': [
         'paasta-api-config=paasta_tools.api.api:make_app'