add socket examples

michaelliao · michaelliao · commit 347de0d3e801 · 2014-09-11T16:50:51.000+08:00
diff --git a/socket/echo_client.py b/socket/echo_client.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'a socket example which send echo message to server.'
+
+import socket
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+# 建立连接:
+s.connect(('127.0.0.1', 9999))
+# 接收欢迎消息:
+print s.recv(1024)
+for data in ['Michael', 'Tracy', 'Sarah']:
+    # 发送数据:
+    s.send(data)
+    print s.recv(1024)
+s.send('exit')
+s.close()
diff --git a/socket/echo_server.py b/socket/echo_server.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'a server example which send hello to client.'
+
+import time, socket, threading
+
+def tcplink(sock, addr):
+    print 'Accept new connection from %s:%s...' % addr
+    sock.send('Welcome!')
+    while True:
+        data = sock.recv(1024)
+        time.sleep(1)
+        if data == 'exit' or not data:
+            break
+        sock.send('Hello, %s!' % data)
+    sock.close()
+    print 'Connection from %s:%s closed.' % addr
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+# 监听端口:
+s.bind(('127.0.0.1', 9999))
+s.listen(5)
+print 'Waiting for connection...'
+while True:
+    # 接受一个新连接:
+    sock, addr = s.accept()
+    # 创建新线程来处理TCP连接:
+    t = threading.Thread(target=tcplink, args=(sock, addr))
+    t.start()
diff --git a/socket/http_client.py b/socket/http_client.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'a socket example which get html data from www.sina.com.cn'
+
+import socket
+
+s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+# 建立连接:
+s.connect(('www.sina.com.cn', 80))
+# 发送数据:
+s.send('GET / HTTP/1.1\r\nHost: www.sina.com.cn\r\nConnection: close\r\n\r\n')
+# 接收数据:
+buffer = []
+while True:
+    # 每次最多接收1k字节:
+    d = s.recv(1024)
+    if d:
+        buffer.append(d)
+    else:
+        break
+data = ''.join(buffer)
+# 关闭连接:
+s.close()
+
+header, html = data.split('\r\n\r\n', 1)
+print header
+# 把接收的数据写入文件:
+with open('sina.html', 'wb') as f:
+    f.write(html)
