paasta-tools 1.16.3__tar.gz → 1.17.0__tar.gz

{paasta-tools-1.16.3/paasta_tools.egg-info → paasta-tools-1.17.0}/PKG-INFO

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

{paasta-tools-1.16.3 → paasta-tools-1.17.0}/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.16.3"
+__version__ = "1.17.0"

{paasta-tools-1.16.3 → paasta-tools-1.17.0}/paasta_tools/cli/cmds/remote_run.py

@@ -13,13 +13,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import argparse
-import pty
-import shlex
 import time
 
 from paasta_tools.cli.utils import get_paasta_oapi_api_clustername
 from paasta_tools.cli.utils import get_paasta_oapi_client_with_auth
 from paasta_tools.cli.utils import lazy_choices_completer
+from paasta_tools.cli.utils import run_interactive_cli
 from paasta_tools.paastaapi.model.remote_run_start import RemoteRunStart
 from paasta_tools.paastaapi.model.remote_run_stop import RemoteRunStop
 from paasta_tools.utils import get_username
@@ -96,7 +95,7 @@ def paasta_remote_run_start(
         pod=poll_response.pod_name,
         token=token_response.token,
     )
-    pty.spawn(shlex.split(exec_command))
+    run_interactive_cli(exec_command)
     return 0
 
 

{paasta-tools-1.16.3 → paasta-tools-1.17.0}/paasta_tools/cli/utils.py

@@ -18,8 +18,10 @@ import getpass
 import hashlib
 import logging
 import os
+import pty
 import random
 import re
+import shutil
 import socket
 import subprocess
 from collections import defaultdict
@@ -1150,3 +1152,20 @@ def get_paasta_oapi_client_with_auth(
         http_res=http_res,
         auth_token=get_sso_service_auth_token(),
     )
+
+
+def run_interactive_cli(cmd: str, shell: str = "bash", term: str = "xterm-256color"):
+    """Runs interactive command in a pseudo terminal, handling terminal size management
+
+    :param str cmd: shell command
+    :param str shell: shell utility to use as wrapper
+    :param str term: terminal type
+    """
+    cols, rows = shutil.get_terminal_size()
+    wrapped_cmd = (
+        f"export SHELL={shell};"
+        f"export TERM={term};"
+        f"stty columns {cols} rows {rows};"
+        f"exec {cmd}"
+    )
+    pty.spawn([shell, "-c", wrapped_cmd])

{paasta-tools-1.16.3 → paasta-tools-1.17.0}/paasta_tools/kubernetes_tools.py

@@ -1198,6 +1198,13 @@ class KubernetesDeploymentConfig(LongRunningServiceConfig):
             "git_sha", ""
         )
 
+        # Our workloads interact with AWS quite a lot, so it comes handy to
+        # propagate an "application ID" in the user-agent of API requests
+        # for debugging purposes (max length is 50 chars from AWS docs).
+        # TODO: temporarily setting this here rather than in `InstanceConfig`
+        # to only affect long running services.
+        env["AWS_SDK_UA_APP_ID"] = f"{self.service}.{self.instance}"[:50]
+
         # We drop PAASTA_CLUSTER here because it will be added via `get_kubernetes_environment()`
         env.pop("PAASTA_CLUSTER", None)
 

{paasta-tools-1.16.3 → paasta-tools-1.17.0/paasta_tools.egg-info}/PKG-INFO

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