python: update python examples to modern python, Gst, and Glib versions

[experiments/gstreamer.git] / python / gst-looping-video-2.py

diff --git a/python/gst-looping-video-2.py b/python/gst-looping-video-2.py
index eabb7a4..2976906 100755 (executable)
--- a/python/gst-looping-video-2.py
+++ b/python/gst-looping-video-2.py
@@ -1,47 +1,62 @@
-#!/usr/bin/env python
-
+#!/usr/bin/env python3
+#
+# A simple looping player.
+# Version 2, based on "about-to-finish" handling.
+#
 # Get a test sample with:
 # youtube-dl -t http://www.youtube.com/watch?v=yWa-YXiSk2Y
 
 import sys
 
-import gobject
-gobject.threads_init()
+import gi
+gi.require_version('Gst', '1.0')
+from gi.repository import Gst
+Gst.init(None)
 
-import gst
+from gi.repository import GLib
 
 
 class Player:
-    def __init__(self, filename):
-        self._filename = filename
-
-        self._player = gst.element_factory_make("playbin2", "player")
-        self._player.set_property("uri", filename)
+    def __init__(self, uri):
+        self._uri = uri
+        self._player = Gst.ElementFactory.make("playbin3", "player")
+        self._player.set_property("uri", uri)
         self._player.connect("about-to-finish", self.on_about_to_finish)
 
+        bus = self._player.get_bus()
+        bus.add_signal_watch()
+        bus.connect('message::error', self.on_error)
+
     def run(self):
-        self._player.set_state(gst.STATE_PLAYING)
-        loop = gobject.MainLoop()
-        loop.run()
+        self._player.set_state(Gst.State.PLAYING)
+        self.loop = GLib.MainLoop()
+        self.loop.run()
+
+    def quit(self):
+        self._player.set_state(Gst.State.NULL)
+        self.loop.quit()
 
     def on_about_to_finish(self, player):
         sys.stderr.write(".")
-        player.set_property("uri", self._filename)
+        player.set_property("uri", self._uri)
+
+    def on_error(self, bus, msg):
+        (err, debug) = msg.parse_error()
+        print("Error: %s" % err, debug)
+        self.quit()
 
 
 def main(args):
     def usage():
-        sys.stdout.write("usage: %s <URI-OF-MEDIA-FILE>\n" % args[0])
+        sys.stdout.write("usage: %s <filename>\n" % args[0])
 
     if len(args) != 2:
         usage()
         sys.exit(1)
 
-    if not gst.uri_is_valid(args[1]):
-        sys.stderr.write("Error: Invalid URI: %s\n" % args[1])
-        sys.exit(1)
+    uri = Gst.filename_to_uri(args[1])
 
-    player = Player(args[1])
+    player = Player(uri)
     player.run()
 
 if __name__ == '__main__':