diff options
| -rw-r--r-- | photoblaster/server.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/photoblaster/server.py b/photoblaster/server.py index 9082d05..c23db04 100644 --- a/photoblaster/server.py +++ b/photoblaster/server.py @@ -88,7 +88,11 @@ class Server(object): url = request.args.get("url") req = urllib2.Request(url=url) req = urllib2.urlopen(req) - return req.read() + header = req.headers.getheader('content-type') + if re.match(r'image', header, re.IGNORECASE): + return req.read() + else: + raise InvalidUsage('Improper Usage', status_code=410) @self.app.route('/im/data', methods=['GET']) def get_data(): |