Python調用遠程Socket接口

Web應用通信通常都喜歡用HTTP接口，但不排除直接socket通信的情況。

socket除了server端構建麻煩些（需要考慮很多實際情況），對於調用者來說構建個client端其實不比http麻煩多少。

1. #!/usr/bin/env python
2. # -*- coding:utf-8 -*-
3. # Auther: linvo
5. import socket
8. class Client(object):
9. """
10. 調用遠程Socket接口
11. <code>
12. try:
13. obj = Client(host, port)
14. ret = obj.send(data)
15. except Exception, e:
16. ret = e
17. </code>
18. """
19. def __init__(self, host, port, timeout = 5):
20. """
21. 鏈接遠程接口服務
22. """
23. self.sock = None
24. try:
25. socket.setdefaulttimeout(timeout)
26. self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
27. self.sock.connect((host, port))
28. except socket.error, e:
29. raise Exception('SOCKET ERROR:' + str(e))
30. except Exception, e:
31. raise Exception('CONNECT ERROR:' + str(e))
35. def send(self, data):
36. """
37. socket通信
38. 發送和接收
39. data: 發送的數據
40. ret: 接收的數據
41. """
42. ret = None
43. # 鏈接成功，開始傳輸
44. if self.sock:
45. data = str(data)
46. try:
47. result = self.sock.sendall(data)
48. except Exception, e:
49. result = str(e)
51. if result is not None:
52. raise Exception('SEND ERROR:' + str(result))
53. else:
54. # 接收
55. ret = ''
56. try:
57. while True:
58. buffer = self.sock.recv(2048)
59. if buffer:
60. ret += buffer
61. else:
62. break
63. except Exception, e:
64. raise Exception('RECV ERROR:' + str(e))
65. return ret

順便給個簡易的server端，以便測試：

1. import socket
2. sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
3. sock.bind(('localhost', 9801))
4. sock.listen(5)
5. while True:
6. connection, address = sock.accept()
7. try:
8. connection.settimeout(5)
9. buf = connection.recv(2048)
10. connection.send(buf)
11. except socket.timeout:
12. print 'time out'
13. connection.close()