personal-file-sharing-center/index.py at master · goace/personal-file-sharing-center · 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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/usr/bin/env python
import web
import os
import time
import config
from urllib import quote

# load config file
root = config.root

types = [
    ".h",".cpp",".cxx",".cc",".c",".cs",".html",".js",
    ".php",".java",".py",".rb",".as",".jpeg",".jpg",".png",
    ".gif",".ai",".psd",".mp3",".avi",".rmvb",".mp4",".wmv",
    ".mkv",".doc",".docx",".ppt",".pptx",".xls",".xlsx",
    ".zip",".tar",".gz",".7z",".rar",".pdf",".txt",".exe",
    ".apk"
]

render = web.template.render('template')

urls = (
    '/favicon.ico',"Ico",
    '/(.*)','Index',
)

class Ico:
    def GET(self):
        return open("static/img/favicon.ico").read()

class Index:
    def GET(self,path):
        # list all the files
        if path == '':
            list = []
            item = os.listdir(root)
            item = sorted(item, key = str.lower)

            for i in item:
                if i[0] == '.' or os.path.isdir(os.path.join(root, i)):
                    continue
                temp = {}
                temp['name'] = i
                temp['type'] = '.' + i.split('.')[-1]

                try:
                    types.index(temp['type'])
                except:
                    temp['type'] = "general"


                temp["time"] = time.strftime("%H:%M:%S %Y-%m-%d",
                        time.localtime(os.path.getmtime(os.path.join(root, i))))

                size = os.path.getsize(os.path.join(root, i))
                if size < 1024:
                    size = str(size) + ".0 B"
                elif size < 1024 * 1024:
                    size = "%0.1f KB" % (size/1024.0)
                elif size < 1024 * 1024 * 1024:
                    size = "%0.1f MB" % (size/1024.0/1024.0)
                else :
                    size = "%0.1f GB" % (size/1024.0/1024.0/1024.0)

                temp["size"] = size
                temp["encode"] = quote(i)

                list.append(temp)

            return render.layout(list)

        # return a file
        else:
            web.header('Content-Type','application/octet-stream')
            web.header('Content-disposition', 'attachment; filename=%s' % path)
            file = open(os.path.join(root,path))
            size = os.path.getsize(os.path.join(root,path))
            web.header('Content-Length','%s' % size)
            return file.read()

    def DELETE(self,filename):
        try:
            filename = filename.encode('utf-8')
            os.remove(os.path.join(root,filename))
        except:
            return "success"


    def POST(self,filename):

        # save a file to disk
        x = web.input(file={})

        if 'file' in x:
            filepath= x.file.filename.replace('\\','/')     # replaces the windows-style slashes with linux ones.
            filename = filepath.split('/')[-1]              # splits the and chooses the last part (the filename with extension)
            filename = unicode(filename, "utf8")
            fout = open(os.path.join(root,filename),'w')    # creates the file where the uploaded file should be stored
            fout.write(x.file.file.read())                  # writes the uploaded file to the newly created file.
            fout.close()                                    # closes the file, upload complete.

        return "<script>parent.location.reload()</script>"

# start the application
# it's adaptable to both uwsgi start & python start
app = web.application(urls,globals())
application = app.wsgifunc()

if __name__ == "__main__":
    app.run()