Fix favicon (fixes #1)
* Include milla.ico in package_data in setup.py * Use pkg_resources API to load icon instead of milla.__file__ * Allow caller to specify icon content typemaster
parent
dd2634224c
commit
6154f8b6f8
3
setup.py
3
setup.py
|
@ -33,6 +33,9 @@ setup(
|
|||
install_requires=install_requires,
|
||||
packages=find_packages('src', exclude=['distribute_setup']),
|
||||
package_dir={'': 'src'},
|
||||
package_data={
|
||||
'milla': ['milla.ico'],
|
||||
},
|
||||
entry_points={
|
||||
'milla.request_validator': [
|
||||
'default = milla.auth:RequestValidator'
|
||||
|
|
|
@ -24,9 +24,8 @@ from one or more of these classes can make things significantly easier.
|
|||
'''
|
||||
|
||||
import milla
|
||||
import os
|
||||
import pkg_resources
|
||||
|
||||
__all__ = ['Controller', 'FaviconController']
|
||||
|
||||
class Controller(object):
|
||||
'''The base controller class
|
||||
|
@ -41,28 +40,36 @@ class Controller(object):
|
|||
def __after__(self, request):
|
||||
pass
|
||||
|
||||
|
||||
class FaviconController(Controller):
|
||||
'''A controller for the "favicon"
|
||||
|
||||
This controller is specifically suited to serve a site "favicon" or
|
||||
bookmark icon. By default, it will serve the *Milla* icon, but you
|
||||
can pass an alternate file time to the constructor.
|
||||
can pass an alternate filename to the constructor.
|
||||
|
||||
:param icon: Path to an icon to serve
|
||||
:param content_type: Internet media type describing the type of image
|
||||
used as the favicon, defaults to 'image/x-icon' (Windows ICO format)
|
||||
'''
|
||||
|
||||
def __init__(self, icon=None):
|
||||
if not icon:
|
||||
icon = os.path.join(os.path.dirname(milla.__file__), 'milla.ico')
|
||||
def __init__(self, icon=None, content_type='image/x-icon'):
|
||||
if icon:
|
||||
try:
|
||||
self.icon = open(icon)
|
||||
except (IOError, OSError):
|
||||
self.icon = None
|
||||
else:
|
||||
try:
|
||||
self.icon = pkg_resources.resource_stream('milla', 'milla.ico')
|
||||
except IOError:
|
||||
self.icon = None
|
||||
self.content_type = content_type
|
||||
|
||||
def __call__(self, request):
|
||||
if not self.icon:
|
||||
raise milla.HTTPNotFound
|
||||
response = milla.Response()
|
||||
response.app_iter = self.icon
|
||||
response.headers['Content-Type'] = 'image/x-icon'
|
||||
response.headers['Content-Type'] = self.content_type
|
||||
return response
|
||||
|
|
Loading…
Reference in New Issue