VideoIframe from YouTubeVideo (simplified) + VimeoVideo

IPython/lib/display.py

@@ -6,7 +6,27 b' from os.path import exists, isfile, splitext, abspath, join, isdir'
 from os import walk, sep
 
 
-class YouTubeVideo(object):
+class VideoIframe(object):
+    """
+    Generic class to embed videos in a iframe
+    """
+
+    def __init__(self, id, width=400, height=300, **kwargs):
+        self.id = id
+        self.width = width
+        self.height = height
+        self.params = kwargs
+
+    def _repr_html_(self):
+        """return the embed iframe for this video id"""
+        if self.params:
+            from urllib import urlencode
+            params = "?" + urlencode(self.params)
+        else:
+            params = ""
+        return self.iframe % (self.width, self.height, self.id, params)
+
+class YouTubeVideo(VideoIframe):
     """Class for embedding a YouTube Video in an IPython session, based on its video id.
 
     e.g. to embed the video on this page:
@@ -30,20 +50,7 b' class YouTubeVideo(object):'
     https://developers.google.com/youtube/player_parameters#parameter-subheader
     """
 
-    def __init__(self, id, width=400, height=300, **kwargs):
-        self.id = id
-        self.width = width
-        self.height = height
-        self.params = kwargs
-
-    def _repr_html_(self):
-        """return YouTube embed iframe for this video id"""
-        if self.params:
-            from urllib import urlencode  # Deferred import
-            params = "?" + urlencode(self.params)
-        else:
-            params = ""
-        return """
+    iframe = """
             <iframe
                 width="%i"
                 height="%i"
@@ -51,7 +58,25 b' class YouTubeVideo(object):'
                 frameborder="0"
                 allowfullscreen
             ></iframe>
-        """ % (self.width, self.height, self.id, params)
+        """
+
+class VimeoVideo(VideoIframe):
+    """
+    Class for embedding a Vimeo video in an IPython session, based on its video id.
+    """
+
+    iframe = """
+        <iframe
+            width = "%i"
+            height = "%i"
+            src="http://player.vimeo.com/video/%s%s"
+            frameborder="0"
+            webkitAllowFullScreen
+            mozallowfullscreen
+            allowFullScreen
+        ></iframe>
+        """
+
 
 class FileLink(object):
     """Class for embedding a local file link in an IPython session, based on path