First step towards a MediaManager class: Compat one.

To get us moving towards a MediaManager class, the first
idea is to create a class that wraps our current dict based
manager and makes all users happy.

1 files changed, 29 insertions, 2 deletions

diff --git a/mediagoblin/media_types/__init__.py b/mediagoblin/media_types/__init__.py
index 0abb38d3..745e05ef 100644
--- a/mediagoblin/media_types/__init__.py
+++ b/mediagoblin/media_types/__init__.py
@@ -20,6 +20,7 @@ import logging
 import tempfile
 
 from mediagoblin import mg_globals
+from mediagoblin.tools.common import import_component
 from mediagoblin.tools.translate import lazy_pass_to_ugettext as _
 
 _log = logging.getLogger(__name__)
@@ -31,6 +32,29 @@ class InvalidFileType(Exception):
     pass
 
 
+class CompatMediaManager(object):
+    def __init__(self, mm_dict, entry=None):
+        self.mm_dict = mm_dict
+        self.entry = entry
+	
+    def __call__(self, entry):
+        "So this object can look like a class too, somehow"
+        assert self.entry is None
+        return self.__class__(self.mm_dict, entry)
+
+    def __getitem__(self, i):
+        return self.mm_dict[i]
+
+    def __contains__(self, i):
+        return (i in self.mm_dict)
+
+    def get(self, *args, **kwargs):
+        return self.mm_dict.get(*args, **kwargs)
+
+    def __getattr__(self, i):
+        return self.mm_dict[i]
+
+
 def sniff_media(media):
     '''
     Iterate through the enabled media types and find those suited
@@ -74,9 +98,12 @@ def get_media_managers():
     Generator, yields all enabled media managers
     '''
     for media_type in get_media_types():
-        __import__(media_type)
+        mm = import_component(media_type + ":MEDIA_MANAGER")
+
+        if isinstance(mm, dict):
+            mm = CompatMediaManager(mm)
 
-        yield media_type, sys.modules[media_type].MEDIA_MANAGER
+        yield media_type, mm
 
 
 def get_media_type_and_manager(filename):