paasta-tools 1.14.3__tar.gz → 1.14.4__tar.gz

{paasta-tools-1.14.3/paasta_tools.egg-info → paasta-tools-1.14.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.2
 Name: paasta-tools
-Version: 1.14.3
+Version: 1.14.4
 Summary: Tools for Yelps SOA infrastructure
 Home-page: UNKNOWN
 Author: Compute Infrastructure @ Yelp

{paasta-tools-1.14.3 → paasta-tools-1.14.4}/paasta_tools/__init__.py

@@ -17,4 +17,4 @@
 # setup phase, the dependencies may not exist on disk yet.
 #
 # Don't bump version manually. See `make release` docs in ./Makefile
-__version__ = "1.14.3"
+__version__ = "1.14.4"

{paasta-tools-1.14.3 → paasta-tools-1.14.4}/paasta_tools/kubernetes/bin/paasta_cleanup_remote_run_resources.py

@@ -13,6 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import argparse
+import logging
 from datetime import datetime
 from datetime import timedelta
 from datetime import timezone
@@ -30,15 +31,19 @@ from paasta_tools.kubernetes_tools import KubeClient
 
 ListingFuncType = Callable[[KubeClient, str], Sequence[Any]]
 DeletionFuncType = Callable[[str, str], Any]
+log = logging.getLogger(__name__)
 
 
-def clean_namespace(kube_client: KubeClient, namespace: str, age_limit: datetime):
+def clean_namespace(
+    kube_client: KubeClient, namespace: str, age_limit: datetime, dry_run: bool = False
+):
     """Clean ephemeral remote-run resource in a namespace
 
     :param KubeClient kube_client: kubernetes client
     :param str namepsace: kubernetes namespace
     :param datetime age_limit: expiration time for resources
     """
+    dry_run_msg = " (dry_run)" if dry_run else ""
     cleanup_actions: Sequence[Tuple[DeletionFuncType, ListingFuncType]] = (
         (
             kube_client.core.delete_namespaced_service_account,
@@ -54,7 +59,9 @@ def clean_namespace(kube_client: KubeClient, namespace: str, age_limit: datetime
                 or entity.metadata.creation_timestamp > age_limit
             ):
                 continue
-            delete_func(entity.metadata.name, namespace)
+            log.info(f"Deleting {entity.metadata.name} in {namespace}{dry_run_msg}")
+            if not dry_run:
+                delete_func(entity.metadata.name, namespace)
 
 
 def parse_args() -> argparse.Namespace:
@@ -68,15 +75,30 @@ def parse_args() -> argparse.Namespace:
         default=600,
         help="Maximum age, in seconds, resources are allowed to have",
     )
+    parser.add_argument(
+        "-n",
+        "--dry-run",
+        action="store_true",
+        default=False,
+        help="Do not actually delete resources",
+    )
+    parser.add_argument(
+        "-v",
+        "--verbose",
+        action="store_true",
+        default=False,
+        help="More verbose logging",
+    )
     return parser.parse_args()
 
 
 def main():
     args = parse_args()
+    logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
     kube_client = KubeClient()
     age_limit = datetime.now(tz=timezone.utc) - timedelta(seconds=args.max_age)
     for namespace in get_all_managed_namespaces(kube_client):
-        clean_namespace(kube_client, namespace, age_limit)
+        clean_namespace(kube_client, namespace, age_limit, args.dry_run)
 
 
 if __name__ == "__main__":

{paasta-tools-1.14.3 → paasta-tools-1.14.4}/paasta_tools/kubernetes/remote_run.py

@@ -372,7 +372,7 @@ def create_pod_scoped_role(
             labels={POD_OWNER_LABEL: user},
         ),
     )
-    kube_client.core.create_namespaced_role(namespace=namespace, body=role)
+    kube_client.rbac.create_namespaced_role(namespace=namespace, body=role)
     return role_name
 
 

{paasta-tools-1.14.3 → paasta-tools-1.14.4/paasta_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.2
 Name: paasta-tools
-Version: 1.14.3
+Version: 1.14.4
 Summary: Tools for Yelps SOA infrastructure
 Home-page: UNKNOWN
 Author: Compute Infrastructure @ Yelp