Застрял с Python HTTP Server с базовой аутентификацией с использованием BaseHTTP - PullRequest
Новая
       28

Застрял с Python HTTP Server с базовой аутентификацией с использованием BaseHTTP

10 голосов
/ 26 ноября 2010

Я застрял, пытаясь заставить работать веб-сервер на основе Python.

Я хочу выполнить базовую аутентификацию (отправка заголовка 401) и аутентификацию по списку пользователей. У меня нет проблем с отправкой ответа 401 с заголовком «WWW-Authorize», я могу проверить ответ пользователя (имя пользователя и пароль в кодировке base64), однако после успешного подтверждения окно ввода пароля продолжает появляться. 

import SimpleHTTPServer
import SocketServer
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer

class Handler(BaseHTTPRequestHandler):
    ''' Main class to present webpages and authentication. '''
    def do_HEAD(self):
        print "send header"
        self.send_response(401)
        self.send_header('WWW-Authenticate', 'Basic realm=\"Test\"')
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        ''' Present frontpage with user authentication. '''
        self.do_HEAD()

        if self.headers.getheader('Authorization') == None:
            self.wfile.write('no auth header received')
            pass
        elif self.headers.getheader('Authorization') == 'Basic dGVzdDp0ZXN0':
            self.wfile.write(self.headers.getheader('Authorization'))
            self.wfile.write('authenticated!')
            pass
        else:
            self.wfile.write(self.headers.getheader('Authorization'))
            self.wfile.write('not authenticated')
            pass

httpd = SocketServer.TCPServer(("", 10001), Handler)

httpd.serve_forever()

if __name__ == '__main__':
    main()

При первой загрузке (http://localhost:10001) всплывающее окно входа в систему, я вхожу в тест, тестовый (правильный пользователь) пользователь проверен нормально, но окно снова появляется, если я нажимаю кнопку отмены, я получаю на проверенную страницу ...

Кто-нибудь может помочь? Я подозреваю, что это как-то связано с тем, что авторизация происходит в do_GET, который запускается каждый раз при загрузке страницы.

Ответы [ 2 ]

17 голосов
/ 16 ноября 2011

Попробуйте для размера: 

import SimpleHTTPServer
import SocketServer
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer

class Handler(BaseHTTPRequestHandler):
    ''' 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=\"Test\"')
        self.send_header('Content-type', 'text/html')
        self.end_headers()

    def do_GET(self):
        ''' 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 dGVzdDp0ZXN0':
            self.do_HEAD()
            self.wfile.write(self.headers.getheader('Authorization'))
            self.wfile.write('authenticated!')
            pass
        else:
            self.do_AUTHHEAD()
            self.wfile.write(self.headers.getheader('Authorization'))
            self.wfile.write('not authenticated')
            pass

httpd = SocketServer.TCPServer(("", 10001), Handler)

httpd.serve_forever()

if __name__ == '__main__':
    main()
6 голосов
/ 26 ноября 2010

Это потому, что вы безоговорочно отправляете в ответ заголовок 401 и WWW-Authenticate. Это необходимо делать только в том случае, если в запросе отсутствуют допустимые учетные данные для аутентификации. Если вы удовлетворены запросом, отправьте 200 (или любой другой подходящий) и больше не запрашивайте аутентификацию.

...