Перенаправить пользователя на URL с WSGI (без фреймворка)

Я пытаюсь разработать небольшое веб-приложение с использованием Python WSGI. Например, если пользователь выберет Google, он будет перенаправлен на google.com, если он выберет Facebook, он будет перенаправлен на facebook.com и т. Д.

from wsgiref.simple_server import make_server
from cgi import parse_qs, escape

main_html = """
<html>
<head><title> Welcome to redirection test page </title> </head>
<body>
    <form method="get" action='/visit'>
        <input type=radio name='site' value=google> Google
        <input type=radio name='site' value=facebook> Facebook
        <input type=submit value=submit>
    </form>
</body>
</html>
"""


def main(environ, start_response):

    response_body = main_html
    print type(response_body)   
    status = '200 OK'

    response_headers = [
                        ('Content-Type','text/html'),
                        ('Content-Length', str(len(response_body)))
                       ]


    start_response(status, response_headers)
    return [response_body]


def visit(environ, start_response):
    qs = parse_qs(environ['QUERY_STRING'])
    dest = qs.ge('site')[0]
    if dest == 'google':
        start_response('301 Moved Permanently', [('Location','http://google.com')])
    else:
        start_response('301 Moved Permanently', [('Location','http://facebook.com')])

    return [1]


def app(environ, start_response):
    if environ['PATH_INFO'] == '/':
        return main(environ, start_response)
    elif environ['PATH_INFO'] == '/visit':
        return visit(environ, start_response)

httpd =  make_server('192.168.48.128',8052, app)
print 'Serving on port 8052'
httpd.serve_forever()

Однако когда я запускаю этот код, я получаю следующую ошибку:

Traceback (most recent call last):
  File "/usr/lib/python2.7/wsgiref/handlers.py", line 86, in run
    self.finish_response()
  File "/usr/lib/python2.7/wsgiref/handlers.py", line 131, in finish_response
    self.close()
  File "/usr/lib/python2.7/wsgiref/simple_server.py", line 33, in close
    self.status.split(' ',1)[0], self.bytes_sent
AttributeError: 'NoneType' object has no attribute 'split'
Источник

1 ответ

Решение

У вас есть две проблемы, обе в visit(), Первая - опечатка, а вторая - несоблюдение спецификации WSGI.

def visit(environ, start_response):
    qs = parse_qs(environ['QUERY_STRING'])
    dest = qs.ge('site')[0]
    if dest == 'google':
        start_response('301 Moved Permanently', [('Location','http://google.com')])
    else:
        start_response('301 Moved Permanently', [('Location','http://facebook.com')])

    return [1]

Посмотри на строчку dest = qs.ge('site')[0], qs это просто словарь, и не имеет метода ge, просто get, Более того, cgi.parse_qs устарела - используйте взамен urlparse.parse_qs(). Исправление, которое приводит нас к новой ошибке (мой адрес 127.0.0.1, потому что я просто запустил его из localhost):

127.0.0.1 - - [05/Feb/2015 16:45:17] "GET /visit?site=facebook HTTP/1.1" 302 0 
Traceback (most recent call last):
  File "C:\Python27\lib\wsgiref\handlers.py", line 86, in run
    self.finish_response()
  File "C:\Python27\lib\wsgiref\handlers.py", line 128, in finish_response
    self.write(data)
  File "C:\Python27\lib\wsgiref\handlers.py", line 204, in write
    assert type(data) is StringType, "write() argument must be string"
AssertionError: write() argument must be string

Это говорит нам, что что-то должно быть строкой, но это не так. Снова глядя на visit()На обратной линии выявляется проблема. Ты возвращаешься [1] вместо ['1'], Исправление этой ошибки заставляет все работать нормально. Вы обнаружите, что это поведение задокументировано в PEP 333

Вызываемый элемент start_response должен возвращать вызываемый объект write(body_data), который принимает один позиционный параметр: строку, которая должна быть записана как часть тела ответа HTTP.

Наконец, согласно этому ответу, вы, вероятно, должны использовать '302 Found' статус вместо '301 Moved Permanently' для перенаправления, но это работает в любом случае.

Замена функции этим должна исправить вашу проблему:

def visit(environ, start_response):
    qs = parse_qs(environ['QUERY_STRING'])
    dest = qs.get('site')[0]
    if dest == 'google':
        start_response('302 Found', [('Location','http://google.com')])
    else:
        start_response('302 Found', [('Location','http://facebook.com')])

    return ['1']
Источник

Тело ответа должно быть байтами, а не строкой. Поэтому вы должны вернуть [b"1"] вместо [1] или ["1"].

Источник
Другие вопросы по тегам