[Python] 使用 socket 模組基本教學

Last Updated on 2021-04-13 by Clay

我本來打算直接寫一篇使用 Socket 來建立簡單的、只會重複你說的話的 ChatBot，跟現在網路上還滿多的 LineBot 教學差不多。不過差別在於我使用的是 Socket 來建立兩端的連線，而 LineBot 我看過大部份都是使用 webhook 來完成的。

但後來想想，因為我還要拉出界面啥的，怕這樣敘述起來很亂，最後還是忍痛拆成兩偏了。（其實也沒忍什麼，只是覺得一篇解決比較簡單哈哈哈。）

那麼，這篇我先介紹一下 Socket，以及在 Python 中要怎麼使用吧！

Socket

Socket 最早源自於 Berkeley 大學所開發的，我們可以將其視為可以在網路上通訊的工具，只要使用這個工具、或者你習慣說是 API，我們便可以建立起不同主機的 Server 端以及 Client 端，並且讓彼此互相連接、發送訊息。

在 Python 中，本來就存在著 Socket 的 Package，所以我們並不需要使用指令另外安裝，直接 import 即可。

那麼，以下是一段簡單的 sample code。

Server.py

# -*- coding: utf-8 -*-
import socket
HOST = '127.0.0.1'
PORT = 8000

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((HOST, PORT))
server.listen(10)

while True:
    conn, addr = server.accept()
    clientMessage = str(conn.recv(1024), encoding='utf-8')

    print('Client message is:', clientMessage)

    serverMessage = 'I\'m here!'
    conn.sendall(serverMessage.encode())
    conn.close()

Output:

沒錯，在這邊的話，我們什麼結果都沒有印出來，我們只是啟動了 socker 的 server 端（伺服器） 在等待 client 端（客戶端）的訊息而已。

Client.py

import socket

HOST = '127.0.0.1'
PORT = 8000
clientMessage = 'Hello!'

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((HOST, PORT))
client.sendall(clientMessage.encode())

serverMessage = str(client.recv(1024), encoding='utf-8')
print('Server:', serverMessage)

client.close()

Output:

# Server
Client message is: Hello!

# Client
Server: I'm here!

稍微講解下程式碼吧。

一開始，在 Server 端，我們先設定好 Host 跟我們的 Port。在這邊，我們 Host 先設定為 127.0.0.1，也就是本機伺服器的地址。如果你有另外一台有固定 IP 的電腦，也可以在上面啟動伺服器（IP 記得填那台電腦的），再用另外一台電腦用 Client 端連線。

socket.AF_INET: IPv4 (Default)
socket.SOCK_STREAM: TCP (Default) 

這裡我們要設定這兩種選項，也是經典的 IPv4 的 TCP/IP 連線。這種連線機制比較可靠，不過傳輸速度不夠快。

除此之外還有個地方值得注意：那便是我們傳送的訊息是 byte 的資料型態，我們需要轉換成 string 顯示出來才是字串，詳細可以看我兩邊的程式碼。

Client 端傳送了一次訊息後便關閉了；相較之下，我讓 Server 端持續開著等待 Client 端的請求。

這篇是比較基本就能完成的互傳訊息，接下來我打算進展一些，可用這個套件來完成比較多的功能。

References

• https://docs.python.org/3/howto/sockets.html
• https://realpython.com/python-sockets/

Read More

• [Python] 使用 Socket 建立簡單的 ChatBot