Add <video> element

Author: srittau

NEWS.md

@@ -2,6 +2,7 @@
 
 ## API Additions
 
+* Add `Video` element.
 * Add `Button.disabled`.
 * Add `autocomplete` attribute to `Input`, `TextArea`, `Select`, and `Form`.
 * Add `enum_attribute`.

elements.rst

@@ -61,3 +61,4 @@ htmlgen currently supports the following HTML elements:
 * <title> - Title
 * <tr> - TableRow
 * <ul> - UnorderedList
+* <video> - Video

htmlgen/__init__.py

@@ -96,3 +96,4 @@
     Column,
 )
 from .time import Time
+from .video import Preload, Video

htmlgen/__init__.pyi

@@ -11,3 +11,4 @@ from .list import *
 from .structure import *
 from .table import *
 from .time import *
+from .video import *

htmlgen/video.py

@@ -0,0 +1,20 @@
+from enum import Enum
+
+from .attribute import boolean_html_attribute, html_attribute, enum_attribute
+from .element import Element
+
+
+class Preload(Enum):
+    NONE = "none"
+    METADATA = "metadata"
+    AUTO = "auto"
+
+
+class Video(Element):
+    controls = boolean_html_attribute("controls")
+    poster = html_attribute("poster")
+    preload = enum_attribute("preload", Preload)
+
+    def __init__(self, src: str) -> None:
+        super().__init__("video")
+        self.set_attribute("src", src)

htmlgen/video.pyi

@@ -0,0 +1,16 @@
+from enum import Enum
+from typing import Optional
+
+from htmlgen import Element
+
+class Preload(Enum):
+    NONE: str
+    METADATA: str
+    AUTO: str
+
+
+class Video(Element):
+    controls: bool
+    poster: Optional[str]
+    preload: Optional[Preload]
+    def __init__(self, src: str) -> None: ...

test_htmlgen/video.py

@@ -0,0 +1,42 @@
+from unittest import TestCase
+
+from asserts import assert_equal, assert_true, assert_is_none, assert_false
+
+from htmlgen import Video, Preload
+
+
+class VideoTest(TestCase):
+    def test_src_only(self):
+        video = Video("foo.mp4")
+        assert_equal(
+            [b'<video src="foo.mp4">', b"</video>"], list(iter(video)),
+        )
+
+    def test_controls(self):
+        video = Video("foo.mp4")
+        assert_false(video.controls)
+        video.controls = True
+        assert_true(video.controls)
+        assert_equal(
+            [b'<video controls="controls" src="foo.mp4">', b"</video>"],
+            list(iter(video)),
+        )
+
+    def test_poster(self):
+        video = Video("foo.mp4")
+        assert_is_none(video.poster)
+        video.poster = "poster.png"
+        assert_equal("poster.png", video.poster)
+        assert_equal(
+            [b'<video poster="poster.png" src="foo.mp4">', b"</video>"],
+            list(iter(video)),
+        )
+
+    def test_preload(self):
+        video = Video("foo.mp4")
+        assert_is_none(video.preload)
+        video.preload = Preload.NONE
+        assert_equal(
+            [b'<video preload="none" src="foo.mp4">', b"</video>"],
+            list(iter(video)),
+        )