Werkzeug: ํ˜ธ์ŠคํŠธ ์ด๋ฆ„์ด '.'๋กœ ์‹œ์ž‘ํ•˜๋Š” ๊ฒฝ์šฐ BaseRequest.trusted_hosts์˜ TypeError

์— ๋งŒ๋“  2017๋…„ 12์›” 11์ผ  ยท  1๋…ผํ‰  ยท  ์ถœ์ฒ˜: pallets/werkzeug

. ๋กœ ์‹œ์ž‘ํ•˜๋Š” ํ˜ธ์ŠคํŠธ ์ด๋ฆ„์„ wrappers.BaseRequest.trusted_hosts ์— ์ถ”๊ฐ€ํ•˜๋ฉด $#$4$#์—์„œ์™€ ๊ฐ™์ด TypeError: must be str, not bytes (Python 3.6) ๋˜๋Š” TypeError: Can't convert 'bytes' object to str implicitly (Python 3.4)๊ฐ€ ๋ฉ๋‹ˆ๋‹ค werkzeug.wsgi.host_is_trusted , _normalize ๋Š” ๋จผ์ € ref ๋ฅผ bytes ๋กœ ์ธ์ฝ”๋”ฉํ•˜๋Š” $ _encode_idna ๋ฅผ ํ˜ธ์ถœํ•œ ๋‹ค์Œ '.'๋กœ ์‹œ์ž‘ ํ•˜๋ฉด '.' ๋กœ ์—ฐ๊ฒฐ๋ฉ๋‹ˆ๋‹ค. , ๋”ฐ๋ผ์„œ str ์™€ bytes ๋ฅผ ํ˜ผํ•ฉํ•ฉ๋‹ˆ๋‹ค.

์ƒ˜ํ”Œ ์•ฑ

from flask import Flask, request

class TrustedHosts:
    """Flask extension setting `werkzeug.wrappers.BaseRequest.trusted_hosts`."""
    def __init__(self, app=None, trusted_hosts=None):
        self.app = app
        self.trusted_hosts = trusted_hosts
        if app is not None:
            self.init_app(app)

    def init_app(self, app):
        app.before_request(self.set_trusted_hosts)

    def set_trusted_hosts(self):
        # As host is a <strong i="20">@cached_property</strong> this should before its first access.
        request.trusted_hosts = self.trusted_hosts
        print(request.host)

app = Flask(__name__)
TrustedHosts(app, trusted_hosts=['.example.com'])

@app.route('/')
def index():
    return 'OK'

์ „์ฒด ์Šคํƒ ์ถ”์ 

Traceback (most recent call last):
  File "c:\program files\python36\lib\site-packages\flask\app.py", line 1982, in wsgi_app
    response = self.full_dispatch_request()
  File "c:\program files\python36\lib\site-packages\flask\app.py", line 1614, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "c:\program files\python36\lib\site-packages\flask\app.py", line 1517, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "c:\program files\python36\lib\site-packages\flask\_compat.py", line 33, in reraise
    raise value
  File "c:\program files\python36\lib\site-packages\flask\app.py", line 1610, in full_dispatch_request
    rv = self.preprocess_request()
  File "c:\program files\python36\lib\site-packages\flask\app.py", line 1831, in preprocess_request
    rv = func()
  File "D:\Dropbox\p\kb\python\web\flask\trusted_hosts.py", line 17, in set_trusted_hosts
    print(request.host)
  File "c:\program files\python36\lib\site-packages\werkzeug\local.py", line 347, in __getattr__
    return getattr(self._get_current_object(), name)
  File "c:\program files\python36\lib\site-packages\werkzeug\utils.py", line 73, in __get__
    value = self.func(obj)
  File "c:\program files\python36\lib\site-packages\werkzeug\wrappers.py", line 635, in host
    return get_host(self.environ, trusted_hosts=self.trusted_hosts)
  File "c:\program files\python36\lib\site-packages\werkzeug\wsgi.py", line 160, in get_host
    if not host_is_trusted(rv, trusted_hosts):
  File "c:\program files\python36\lib\site-packages\werkzeug\wsgi.py", line 132, in host_is_trusted
    if suffix_match and hostname.endswith('.' + ref):
TypeError: must be str, not bytes
127.0.0.1 - - [11/Dec/2017 17:11:18] "GET / HTTP/1.1" 500 -

๊ฐ€์žฅ ์œ ์šฉํ•œ ๋Œ“๊ธ€

์ฐธ๊ณ ๋กœ ๋ชจ๋“  ์š”์ฒญ์„ ์ฒ˜๋ฆฌํ•˜๋Š” before_request ํ•ธ๋“ค๋Ÿฌ๋Š” ๋น„ํšจ์œจ์ ์ž…๋‹ˆ๋‹ค. ํ•˜์œ„ ํด๋ž˜์Šค Request , ์†์„ฑ์„ ์„ค์ •ํ•˜๊ณ  ํด๋ž˜์Šค๋ฅผ ์•ฑ์— ํ• ๋‹นํ•ฉ๋‹ˆ๋‹ค.

from flask.wrappers import Request

class TrustedRequest(Request):
    trusted_hosts = ['a', 'b', 'c']

app.request_class = TrustedRequest

>๋ชจ๋“  ๋Œ“๊ธ€

์ฐธ๊ณ ๋กœ ๋ชจ๋“  ์š”์ฒญ์„ ์ฒ˜๋ฆฌํ•˜๋Š” before_request ํ•ธ๋“ค๋Ÿฌ๋Š” ๋น„ํšจ์œจ์ ์ž…๋‹ˆ๋‹ค. ํ•˜์œ„ ํด๋ž˜์Šค Request , ์†์„ฑ์„ ์„ค์ •ํ•˜๊ณ  ํด๋ž˜์Šค๋ฅผ ์•ฑ์— ํ• ๋‹นํ•ฉ๋‹ˆ๋‹ค.

from flask.wrappers import Request

class TrustedRequest(Request):
    trusted_hosts = ['a', 'b', 'c']

app.request_class = TrustedRequest
์ด ํŽ˜์ด์ง€๊ฐ€ ๋„์›€์ด ๋˜์—ˆ๋‚˜์š”?
0 / 5 - 0 ๋“ฑ๊ธ‰

๊ด€๋ จ ๋ฌธ์ œ

asottile picture asottile  ยท  11์ฝ”๋ฉ˜ํŠธ

golf-player picture golf-player  ยท  10์ฝ”๋ฉ˜ํŠธ

davidism picture davidism  ยท  9์ฝ”๋ฉ˜ํŠธ

miki725 picture miki725  ยท  10์ฝ”๋ฉ˜ํŠธ

lepture picture lepture  ยท  6์ฝ”๋ฉ˜ํŠธ