基于python-httpserver上传下载共享

最新推荐文章于 2024-04-09 13:51:00 发布
小喜子898
最新推荐文章于 2024-04-09 13:51:00 发布
阅读量472
点赞数
分类专栏： python 文章标签： python 开发语言
本文链接：https://blog.csdn.net/weixin_38250126/article/details/123396170
版权
python 专栏收录该内容
18 篇文章 0 订阅
订阅专栏
说明：本代码基于python-httpserver，实现过滤隐藏文件，完成修改时间排序
import cgi
import html
import http.server
import mimetypes
import os
import platform
import posixpath
import re
import shutil
import socket
import subprocess
import sys
import threading
import time
import urllib.error
import urllib.parse
import urllib.request
import uuid
from socketserver import ThreadingMixIn

try:
    import numpy as np
except:
    os.system('pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy ')
    import numpy as np

try:
    import qrcode
except:
    os.system('pip install -i https://pypi.tuna.tsinghua.edu.cn/simple qrcode ')
    import qrcode

try:
    import PIL
except:
    os.system('pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pillow ')

try:
    from io import StringIO
except ImportError:
    from io import StringIO


class GetWanIp:
    def getip(self):
        import socket
        myname = socket.getfqdn(socket.gethostname())
        myaddr = socket.gethostbyname(myname)
        print(myaddr)
        return myaddr

    def visit(self, url):
        opener = urllib.request.urlopen(url, None, 3)
        if url == opener.geturl():
            str = opener.read()
        return re.search(r'(\d+\.){3}\d+', str).group(0)


def showTips():
    print("")
    print('----------------------------------------------------------------------->> ')
    try:
        port = int(sys.argv[1])
    except Exception as e:
        port = 8080

    if not 1024 < port < 65535:
        port = 9898
    print('-------->> 现在，在监听 ' + str(port) + ' 端口 ...')
    osType = platform.system()
    # data = 'http://172.16.8.88:' + str(port)
    data = 'http://127.0.0.1:' + str(port)

    if osType == "Windows":
        print('-------->> 您可以访问地址URL:http://' + str(GetWanIp().getip()) + ':' + str(port))
        data = 'http://' + str(GetWanIp().getip()) + ':' + str(port)
    else:
        print('-------->> 您可以访问地址URL:http://127.0.0.1:' + str(port))
    return ('', port)


serveraddr = showTips()


def sizeof_fmt(num):
    for x in ['bytes', 'KB', 'MB', 'GB']:
        if num < 1024.0:
            return "%3.1f%s" % (num, x)
        num /= 1024.0
    return "%3.1f%s" % (num, 'TB')


def modification_date(filename):
    # return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(os.path.getmtime(filename)))
    return time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(os.path.getmtime(filename)))



class SimpleHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
    def do_GET(self):
        f = self.send_head()
        if f:
            for i in f.readlines():
                if isinstance(i, str):
                    self.wfile.write(i.encode("utf-8"))
                else:
                    self.wfile.write(i)
            f.close()

    def do_HEAD(self):
        f = self.send_head()
        if f:
            f.close()

    def do_POST(self):
        r, info = self.deal_post_data()
        print(r, info, "by: ", self.client_address)
        f = StringIO()
        f.write('<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">')
        f.write(
            '<meta name="viewport" content="width=device-width" charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0, minimum-scale=1.0, maximum-scale=1.0, user-scalable=no">')
        f.write("<html>\n<title>上传结果</title>\n")
        f.write("<body>\n<h2>文件上传</h2>")
        if r:
            f.write('<strong style="color:#00FF00">成功</strong>\n')
        else:
            f.write('<strong style="color:#FF0000">失败</strong>\n')
        f.write("<hr>\n")
        f.write(info)
        f.write("</br><a href=\"%s\">点击返回</a>" % self.headers['referer'])
        f.write("</small></body>\n</html>\n")
        length = f.tell()
        f.seek(0)
        self.send_response(200)
        self.send_header("Content-type", "text/html")
        self.send_header("Content-Length", str(length))
        self.end_headers()
        if f:
            for i in f.readlines():
                self.wfile.write(i.encode("utf-8"))
            f.close()

    def deal_post_data(self):
        boundary = str(
            self.headers["Content-Type"].split("=")[1]).encode("utf-8")
        remainbytes = int(self.headers['Content-length'])
        line = self.rfile.readline()
        remainbytes -= len(line)
        if not boundary in line:
            return (False, "Content NOT begin with boundary")
        line = self.rfile.readline()
        remainbytes -= len(line)
        fn = re.findall(
            r'Content-Disposition.*name="file"; filename="(.*)"'.encode('utf-8'), line)

        if not fn:
            return (False, "Can't find out file name...")
        path = str(self.translate_path(self.path)).encode('utf-8')
        osType = platform.system()
        try:
            if osType == "Linux":
                fn = os.path.join(path, fn[0].decode('gbk').encode('utf-8'))

            else:
                fn = os.path.join(path, fn[0])
        except Exception as e:
            return (False, "文件名请不要用中文，或者使用IE上传中文名的文件。{}".format(e))
        while os.path.exists(fn):
            fn += "_".encode("utf-8")
        line = self.rfile.readline()
        remainbytes -= len(line)
        line = self.rfile.readline()
        remainbytes -= len(line)
        try:
            out = open(fn, 'wb')
        except IOError:
            return (False, "Can't create file to write, do you have permission to write?")

        preline = self.rfile.readline()
        remainbytes -= len(preline)
        while remainbytes > 0:
            line = self.rfile.readline()
            remainbytes -= len(line)
            if boundary in line:
                preline = preline[0:-1]
                if preline.endswith('\r'.encode("utf-8")):
                    preline = preline[0:-1]
                out.write(preline)
                out.close()
                # return (True, "文件 '%s' 上传成功" % fn)
                return (True, "文件上传成功")

            else:
                out.write(preline)
                preline = line
        return (False, "Unexpect Ends of data.")

    def send_head(self):
        path = self.translate_path(self.path)
        f = None
        if os.path.isdir(path):
            if not self.path.endswith('/'):
                self.send_response(301)
                self.send_header("Location", self.path + "/")
                self.end_headers()
                return None
            for index in "index.html", "index.htm":
                index = os.path.join(path, index)
                if os.path.exists(index):
                    path = index
                    break
            else:
                return self.list_directory(path)
        ctype = self.guess_type(path)
        try:
            f = open(path, 'rb')
        except IOError:
            self.send_error(404, "File not found")
            return None
        self.send_response(200)
        self.send_header("Content-type", ctype)
        fs = os.fstat(f.fileno())
        self.send_header("Content-Length", str(fs[6]))
        self.send_header("Last-Modified", self.date_time_string(fs.st_mtime))

        # self.send_header("Last-Modified", self.date_time_string(fs.st_ctime))
        self.end_headers()
        return f

    def list_directory(self, path):
        try:
            list = os.listdir(path)
            for filename in list:

                if  filename.startswith('.') or filename.startswith('_'):
                    list.remove(filename)


        except os.error:
            self.send_error(404, "No permission to list directory")
            return None
        # list.sort(key=lambda a: a.lower())

        f = StringIO()
        displaypath = html.escape(urllib.parse.unquote(self.path))
        f.write('<!DOCTYPE html>')
        f.write(
            '<meta name="viewport" content="width=device-width" charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0, minimum-scale=1.0, maximum-scale=1.0, user-scalable=no">')
        f.write("<html>\n<title>agent上传下载</title>\n")
        f.write("<hr>\n")
        f.write("<form ENCTYPE=\"multipart/form-data\" method=\"post\">")
        f.write("<input name=\"file\" type=\"file\"/>")
        f.write("<input type=\"submit\" value=\"上传\"/>")
        f.write(
            "&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp&nbsp")
        f.write("</form>\n")
        f.write('<hr>\n<ul>\n')
        f.write("<p><font size=\"5\" face=\"arial\" color=\"red\">文件下载地址为：http://172.16.8.88:9898/+下载文件名</font></p>")

        newdata = []
        for name in list:
            # print(">>>>>>name:"+name)
            newdata.append({
                'filename': name,
                'mtime': modification_date(name),
            })
        new_list = sorted(newdata, key=lambda item: item["mtime"], reverse=True)
        for ll in range(len(new_list)):
            newfile = new_list[ll]["filename"]
            print("<<<<<<<<newfile:" + newfile)
            fullname = os.path.join(path, newfile)
            colorName = displayname = linkname = newfile
            print(displayname)
            if os.path.isdir(fullname):
                colorName = '<span style="background-color: #CEFFCE;">' + newfile + '/</span>'
                displayname = newfile
                linkname = newfile + "/"
            if os.path.islink(fullname):
                colorName = '<span style="background-color: #FFBFFF;">' + newfile + '@</span>'
                displayname = newfile
                print(displaypath)
                filename = os.getcwd() + '/' + displaypath + displayname
                print("&&&&" + os.getcwd())
                print("当前工作目录 : %s" % os.getcwd())
            f.write(
                '<table><tr><td width="600"><a href="%s">%s</a></td><td ">%s</td></tr>\n'
                % (urllib.parse.quote(linkname), colorName, modification_date(newfile)))

        f.write("</table>\n<hr>\n</body>\n</html>\n")

        length = f.tell()
        f.seek(0)
        self.send_response(200)
        self.send_header("Content-type", "text/html")
        self.send_header("Content-Length", str(length))
        self.end_headers()
        return f

    def translate_path(self, path):
        path = path.split('?', 1)[0]
        path = path.split('#', 1)[0]
        path = posixpath.normpath(urllib.parse.unquote(path))
        words = path.split('/')
        words = [_f for _f in words if _f]
        path = os.getcwd()
        for word in words:
            drive, word = os.path.splitdrive(word)
            head, word = os.path.split(word)
            if word in (os.curdir, os.pardir):
                continue
            path = os.path.join(path, word)
        return path

    def copyfile(self, source, outputfile):
        shutil.copyfileobj(source, outputfile)

    def guess_type(self, path):

        base, ext = posixpath.splitext(path)
        if ext in self.extensions_map:
            return self.extensions_map[ext]
        ext = ext.lower()
        if ext in self.extensions_map:
            return self.extensions_map[ext]
        else:
            return self.extensions_map['']

    if not mimetypes.inited:
        mimetypes.init()  # try to read system mime.types
    extensions_map = mimetypes.types_map.copy()
    extensions_map.update({
        '': 'application/octet-stream',  # Default
        '.py': 'text/plain',
        '.c': 'text/plain',
        '.h': 'text/plain',
    })


class ThreadingServer(ThreadingMixIn, http.server.HTTPServer):
    pass


def test(HandlerClass=SimpleHTTPRequestHandler,
         ServerClass=http.server.HTTPServer):
    http.server.test(HandlerClass, ServerClass)


if __name__ == '__main__':
    srvr = ThreadingServer(serveraddr, SimpleHTTPRequestHandler)

    srvr.serve_forever()
小喜子898
关注
0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
基于python-httpserver上传下载共享

说明：本代码基于python-httpserver，实现过滤隐藏文件，完成修改时间排序import cgiimport htmlimport http.serverimport mimetypesimport osimport platformimport posixpathimport reimport shutilimport socketimport subprocessimport sysimport threadingimport timeimport urlli
复制链接

扫一扫