import socket
def main():
  # 1. 創建tcp的套接字
  tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 2. 鏈接服務器
  # tcp_socket.connect(("192.168.33.11", 7890))
  server_ip = input("請輸入要鏈接的服務器的ip:")
  server_port = int(input("請輸入要鏈接的服務器的port:"))
  server_addr = (server_ip, server_port)
  tcp_socket.connect(server_addr)
  # 3. 發送數據/接收數據
  send_data = input("請輸入要發送的數據:")
  tcp_socket.send(send_data.encode("utf-8"))
  # 接收數據。 會阻塞代碼,直到接收到數據
  recv_data = tcp_socket.recv(1024) # 1024表示本次接收的最大字節數。
  print(recv_data.decode("utf-8"))
  # 4. 關閉套接字
  tcp_socket.close()
if __name__ == "__main__":
  main()

import socket
def main():
  # 1. 創建套接字 socket
  tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 設置當服務器先close 即服務器端4次揮手之后資源能夠立即釋放，這樣就保證了，下次運行程序時 可以立即綁定7890端口 (如果服務端先close,再啟動服務端就不會報端口被占用的錯誤。)
  tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  # 2. 綁定本機ip和端口號 bind
  tcp_server_socket.bind(("", 7890))
  # 3. 監聽客戶端的鏈接請求 listen (由主動去鏈接服務器,變成被客戶端鏈接)
  tcp_server_socket.listen(128) # 128影響鏈接客戶端的數量(并發量)(和操作系統也有關)
  # 4. 接受客戶端的鏈接 accept。 會阻塞代碼,直到有客戶端鏈接
  new_client_socket, client_addr = tcp_server_socket.accept()
  print(client_addr) # ("192.168.33.109",53766) 客戶端的ip和端口
  # 接收客戶端發送過來的數據。 會阻塞代碼,直到接收到數據
  recv_data = new_client_socket.recv(1024) # 1024表示本次接收的最大字節數。
  print(recv_data.decode("utf-8")) # 如果recv_data為空,表示客戶端斷開鏈接
  # 寫數據給客戶端
  new_client_socket.send("hahahghai-----ok-----".encode("utf-8"))
  # 關閉套接字
  new_client_socket.close() # 與客戶端通信的套接字
  tcp_server_socket.close() # 服務器套接字
if __name__ == "__main__":
  main()