Jay/internet.py at master · ijinjay/Jay · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# coding:utf8
# import sys
# print sys.path
# sys.path.append('/usr/local/lib/python2.7/site-packages')
# print sys.path

import SocketServer
import subprocess
import string
import time
import os
import json

class serverInfo(SocketServer.BaseRequestHandler):
    """docstring for serverInfo"""
    def recvFile(self,filename):
        print "Start receive file"
        f = open(filename, 'wb')
        self.request.send('ready')
        while True:
            data = self.request.recv(4096)
            if data == 'EOF':
                print "receive file success!"
                break
            f.write(data)
        f.close()

    def sendFile(self, filename):
        print "start send file"
        self.request.send('ready')
        time.sleep(1)
        f = open(filename, 'rb')
        while True:
            data = f.read(4096)
            if not data:
                break
            self.request.send(data)
        f.close()
        time.sleep(1)
        self.request.send('EOF')
        print "send file success!"

    def sentLists(self):
        filelists = os.listdir(os.getcwd())
        listdata = json.dumps(filelists)
        print type(listdata)
        self.request.send(listdata)

    def handle(self):
        print "get connection from:",self.client_address
        while True:
            try:
                # send Dir info
                self.sentLists()
                # receive inf
                data = self.request.recv(4096)
                print "get Data:", data
                if not data:
                    print "break the connection!"
                    break
                else:
                    action, filename = data.split()
                    filename = os.path.basename(filename)
                    if action == "put":
                        self.recvFile(filename)
                        self.sentLists()
                    elif action == "get":
                        self.sendFile(filename)
                    else:
                        print "Get Error!"
                        continue
            except Exception, e:
                print "Error: ",e

if __name__ == "__main__":
    host = ''
    port = 13140
    s = SocketServer.ThreadingTCPServer((host,port),serverInfo)
    s.serve_forever()