将Web服务器从Python 2转换为Python 3

时间:2018-07-14 10:59:05

标签: python-3.x

我一直在使用此脚本,在线发现不是我的作品!

在python2中,它可以完美运行,但是在python3中(使用2到3),我们得到错误:

  

需要一个类似字节的对象,而不是'str'

如果我用静态凭据sys.argv[2]代替b'user:pass',我将不知道如何编码,该错误将消失。实际上这不是问题,我希望这样做,但这会带来另一个问题:设置端口。 sys.argv[1](端口号)的实际使用方式/位置?任何帮助表示赞赏。

import os
import BaseHTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler
import sys
import base64

os.chdir("H:\\")

key = ""

class AuthHandler(SimpleHTTPRequestHandler):
    ''' Main class to present webpages and authentication. '''
    def do_HEAD(self):
        print "send header"
        self.send_response(200)
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_AUTHHEAD(self):
        print "send header"
        self.send_response(401)
        self.send_header('WWW-Authenticate', 'Basic realm=\"Marradio\"')
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        global key
        ''' Present frontpage with user authentication. '''
        if self.headers.getheader('Authorization') == None:
            self.do_AUTHHEAD()
            self.wfile.write('no auth header received')
            pass
        elif self.headers.getheader('Authorization') == 'Basic '+key:
            SimpleHTTPRequestHandler.do_GET(self)
            pass
        else:
            self.do_AUTHHEAD()
            self.wfile.write(self.headers.getheader('Authorization'))
            self.wfile.write('not authenticated')
            pass

def test(HandlerClass = AuthHandler,
         ServerClass = BaseHTTPServer.HTTPServer):
    BaseHTTPServer.test(HandlerClass, ServerClass)


if __name__ == '__main__':
    if len(sys.argv)<3:
        print "usage SimpleAuthServer.py [port] [username:password]"
        sys.exit()
    key = base64.b64encode(sys.argv[2])

    test()

1 个答案:

答案 0 :(得分:0)

将字符串替换为字节值:

base64string = base64.b64encode( bytes(s, "utf-8") )