简体   繁体   中英

kubectl cp in kubernetes python client

I have been trying to convert kubectl cp command to it's equivalent kubernetes python client program. I have following code for that:

from kubernetes import client, config
from kubernetes.stream import stream
import tarfile
from tempfile import TemporaryFile

# create an instance of the API class
config.load_kube_config()
api_instance = client.CoreV1Api()

exec_command = ['tar', 'xvf', '-', '-C', '/']
resp = stream(api_instance.connect_get_namespaced_pod_exec, "nginx-deployment-6bb6554bf-9sdtr", 'default',
              command=exec_command,
              stderr=True, stdin=True,
              stdout=True, tty=False,
              _preload_content=False)

source_file = '/tmp/abc.txt'

with TemporaryFile() as tar_buffer:
    with tarfile.open(fileobj=tar_buffer, mode='w') as tar:
        tar.add(source_file)

    tar_buffer.seek(0)
    commands = []
    commands.append(tar_buffer.read())

    while resp.is_open():
        resp.update(timeout=1)
        if resp.peek_stdout():
            print("STDOUT: %s" % resp.read_stdout())
        if resp.peek_stderr():
            print("STDERR: %s" % resp.read_stderr())
        if commands:
            c = commands.pop(0)
            # print("Running command... %s\n" % c)

            resp.write_stdin(c)
        else:
            break
    resp.close()

The above code gives me following error:

/home/velotio/venv/bin/python /home/velotio/PycharmProjects/k8sClient/testing.py
Traceback (most recent call last):
  File "/home/velotio/PycharmProjects/k8sClient/testing.py", line 38, in <module>
    resp.write_stdin(c)
  File "/usr/local/lib/python3.6/site-packages/kubernetes/stream/ws_client.py", line 160, in write_stdin
    self.write_channel(STDIN_CHANNEL, data)
  File "/usr/local/lib/python3.6/site-packages/kubernetes/stream/ws_client.py", line 114, in write_channel
    self.sock.send(chr(channel) + data)
TypeError: must be str, not bytes

I am using Python 3.6.3 and on kubernetes 1.13 version.

You have to convert bytes back to string which is what write_stdin method is expecting to get.

for example:

resp.write_stdin(c.decode())

Another example:

# Array with two byte objects
In [1]: a = [b'1234', b'3455']

# Pop one of them
In [2]: c = a.pop(0)

# it is of byte type
In [3]: c
Out[3]: b'1234'

# wrapping in str won't help if you don't provide decoding
In [4]: str(c)
Out[4]: "b'1234'"

# With decoding
In [5]: str(c, 'utf-8')
Out[5]: '1234'

# Or simply use the decode str method
In [6]: c.decode()
Out[6]: '1234'

More on byte to string conversion here: Convert bytes to a string?

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM