paasta-tools 1.21.3__py3-none-any.whl

paasta_tools/list_kubernetes_service_instances.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python
+# Copyright 2015-2016 Yelp Inc.
+#
+# 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.
+"""Usage: ./list_kubernetes_service_instances.py [options]
+
+Enumerates all kubernetes instances for services in the soa directory that
+are for the current cluster (defined by the kubernetes configuration file).
+
+Outputs (to stdout) a list of service.instance (one per line)
+for each instance found in kubernetes-<CLUSTER>.yaml for every folder
+in the SOA Configuration directory.
+
+Command line options:
+
+- -d <SOA_DIR>, --soa-dir <SOA_DIR>: Specify a SOA config dir to read from
+- -c <CLUSTER>, --cluster <CLUSTER>: Specify which cluster of services to read
+"""
+import argparse
+import random
+import sys
+from typing import List
+
+from paasta_tools import kubernetes_tools
+from paasta_tools.utils import compose_job_id
+from paasta_tools.utils import DEFAULT_SOA_DIR
+from paasta_tools.utils import get_services_for_cluster
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description="Lists kubernetes instances for a service."
+    )
+    parser.add_argument(
+        "-c",
+        "--cluster",
+        dest="cluster",
+        metavar="CLUSTER",
+        default=None,
+        help="define a specific cluster to read from",
+    )
+    parser.add_argument(
+        "-d",
+        "--soa-dir",
+        dest="soa_dir",
+        metavar="SOA_DIR",
+        default=DEFAULT_SOA_DIR,
+        help="define a different soa config directory",
+    )
+    parser.add_argument(
+        "--sanitise",
+        action="store_true",
+        help=(
+            "Whether or not to sanitise service instance names before displaying "
+            "them. Kubernetes apps created by PaaSTA use sanitised names."
+        ),
+    )
+    parser.add_argument(
+        "-t",
+        "--instance-type",
+        dest="instance_type",
+        default="kubernetes",
+        help="Instance type to list, default %(default)s",
+    )
+    parser.add_argument(
+        "--shuffle",
+        action="store_true",
+        help="Shuffle the instances output.",
+    )
+    parser.add_argument(
+        "--group-lines",
+        type=int,
+        dest="group_lines",
+        help=(
+            "Groups instances output into a desired number of lines with each instance separated by a space"
+        ),
+    )
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    args = parse_args()
+    soa_dir = args.soa_dir
+    cluster = args.cluster
+    instances = get_services_for_cluster(
+        cluster=cluster, instance_type=args.instance_type, soa_dir=soa_dir
+    )
+    service_instances = []
+    for name, instance in instances:
+        if args.sanitise:
+            app_name = kubernetes_tools.get_kubernetes_app_name(name, instance)
+        else:
+            app_name = compose_job_id(name, instance)
+        service_instances.append(app_name)
+    if args.shuffle:
+        random.shuffle(service_instances)
+
+    if args.group_lines:
+        group_lines(service_instances, args.group_lines)
+    else:
+        print("\n".join(service_instances))
+    sys.exit(0)
+
+
+def group_lines(service_instances: List[str], num_lines: int) -> None:
+    output: List[List[str]] = [[] for _ in range(num_lines)]
+    curr_index = 0
+    for instance in service_instances:
+        output[curr_index].append(instance)
+        curr_index = (curr_index + 1) % num_lines
+
+    for group in output:
+        print(" ".join(group))
+
+
+if __name__ == "__main__":
+    main()

paasta_tools/list_tron_namespaces.py

@@ -0,0 +1,60 @@
+#!/usr/bin/env python
+# Copyright 2015-2018 Yelp Inc.
+# 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.
+import argparse
+
+from paasta_tools import tron_tools
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description="Lists Tron namespaces for a cluster, excluding MASTER"
+    )
+    parser.add_argument(
+        "-c",
+        "--cluster",
+        dest="cluster",
+        default=None,
+        help="Use a different Tron cluster",
+    )
+    parser.add_argument(
+        "-d",
+        "--soa-dir",
+        dest="soa_dir",
+        default=tron_tools.DEFAULT_SOA_DIR,
+        help="Use a different soa config directory",
+    )
+    parser.add_argument(
+        "-e",
+        "--executors",
+        default=tron_tools.EXECUTOR_TYPES,
+        nargs="+",
+        help="tron executor types to list namespaces for",
+        choices=tron_tools.EXECUTOR_TYPES,
+    )
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    args = parse_args()
+    namespaces = tron_tools.get_tron_namespaces(
+        cluster=args.cluster,
+        soa_dir=args.soa_dir,
+        tron_executors=args.executors,
+    )
+    print("\n".join(namespaces))
+
+
+if __name__ == "__main__":
+    main()