Switching the hook 'get_media_manager' to a more "directed" tuple-hook

[mediagoblin.git] / mediagoblin / media_types / pdf / __init__.py

diff --git a/mediagoblin/media_types/pdf/__init__.py b/mediagoblin/media_types/pdf/__init__.py
index c180507cadf18f5e72a34acfd781610a98b4a3e3..67509ddc5196835c1f111c05dfc9e396f84aced9 100644 (file)
--- a/mediagoblin/media_types/pdf/__init__.py
+++ b/mediagoblin/media_types/pdf/__init__.py
@@ -34,18 +34,14 @@ class PDFMediaManager(MediaManagerBase):
     default_thumb = "images/media_thumbs/pdf.jpg"
 
 
-def get_media_manager(media_type):
-    if media_type == MEDIA_TYPE:
-        return PDFMediaManager
-
-
 def get_media_type_and_manager(ext):
     if ext in ACCEPTED_EXTENSIONS:
         return MEDIA_TYPE, PDFMediaManager
 
+
 hooks = {
     'setup': setup_plugin,
     'get_media_type_and_manager': get_media_type_and_manager,
     'sniff_handler': sniff_handler,
-    'get_media_manager': get_media_manager,
+    ('media_manager', MEDIA_TYPE): lambda: PDFMediaManager,
 }