pythonでバイナリが書き込めない

Question

pythonのsocketでクライアントがスクリーンショットを取りバイナリをとりsocketでそのバイナリを送りサーバーがバイナリを受取、ファイルに書き込み保存するプログラムなのですが、保存されません。

#サーバー
# socket サーバを作成

import socket

# AF = IPv4 という意味
# TCP/IP の場合は、SOCK_STREAM を使う
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    s.bind(('127.0.0.1', 50007))
    s.listen(1)
    # connection するまで待つ
    while True:
        conn, addr = s.accept()
        with conn:
            while True:
                # データを受け取る
                data = conn.recv(8000)
                with open('./moka.png',mode='wb') as wb:
　　　　　　　　　　　　#moka.pngはあらかじめつくってあります
                    binary = data
                    wb.write(binary)
                if not data:
                    break

#クライアント
import pyautogui
import socket

num = 0
while True:
    s = pyautogui.screenshot()
    s.save("pict.png")
    print(num)
    num = num + 1
    with open("./pict.png",mode='rb') as bin:
        binary = bin.read()
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            # サーバを指定
            s.connect(('127.0.0.1', 50007))
            print(binary)
            s.sendall(binary)
            print("送信")

Comments

[otn]

実行した際の状況を書いてください。

Answer 1

データをrecvするたんびにファイルをopenしてますが、これではファイルは新規作成されてしまいます

Comments

[besides-poit]

他にファイルは新規作成されていないです...

[y_waiwai]

'./moka.png'という名前のファイルが、新規作成されます。それ以前に書き込まれたデータはクリアされますね

[besides-poit]

moka.pngというファイルはあらかじめつくってあります

[y_waiwai]

だから、openしたらその都度クリアされます

Answer 2

問題点 サーバー側

                # データを受け取る
                data = conn.recv(8000)
                with open('./moka.png',mode='wb') as wb:
　　　　　　　　　　　　#moka.pngはあらかじめつくってあります
                    binary = data
                    wb.write(binary)
                if not data:
                    break

• #moka.pngはあらかじめつくってあります の行のインデント
• conn.recv(8000) -> 画像ファイルのサイズは 8000 バイト (単位に注意) より小さい？

　　読み出しのバッファサイズを意図してのサイズでしょうか？

• data が空の時、moka.png を空データで上書きします

　　受信したデータが意図したものになっているか表示して確認して見て下さい。

以上３点の対処で、サーバー側でのファイルの保存を確認出来ました。(win10/py3.8)

ちなみに、recv() では状況によってはデータの取りこぼしが発生することがあります。
今回のケース、ローカル上での実行なら顕在しませんが。(今回の問題の原因ではありません)

recv(bufsize) は、最大サイズで、読み出しが保証されるわけではありません。
通常は何バイト読み込んだか確認してバッファリングが必要になります。
具体的には、socket.makefile() を使い、ファイルと同じように read() で読み出したりします。

# 要: import shutil (標準ライブラリ)

conn, addr = s.accept()
with conn, conn.makefile("rb") as rb, open("./moka.png", mode="wb") as wb:
    shutil.copyfileobj(rb, wb)

より効率良くデータを読みだそうとするなら、接続を維持したままにして、
最初にこれから送るバイナリーデータのデータサイズを送信するとよいです。
正常に受信できたかどうかの確認ができます。

Comments

[besides-poit]

天才ですね