From: Kaliko Jack <kaliko@azylum.org>
Date: Sat, 5 Dec 2020 16:52:02 +0000 (+0100)
Subject: Add readpicture command
X-Git-Tag: v0.5.0~5
X-Git-Url: https://git.kaliko.me/?a=commitdiff_plain;h=e645adfa14a04168a4d8a892e88569205f868c84;p=python-musicpd.git

Add readpicture command
---

diff --git a/doc/source/use.rst b/doc/source/use.rst
index 28f9413..a213247 100644
--- a/doc/source/use.rst
+++ b/doc/source/use.rst
@@ -107,14 +107,14 @@ Fetching album covers is possible with albumart, here is an example:
 
 .. code-block:: python
 
-    >>> fetched_cover_file = '/tmp/cover'
     >>> cli = musicpd.MPDClient()
     >>> cli.connect()
     >>> track = "Steve Reich/1978-Music for 18 Musicians"
-    >>> with open(fetched_cover_file, 'wb') as cover:
-    >>>     aart = cli.albumart(track, 0)
-    >>>     received = int(aart.get('binary'))
-    >>>     size = int(aart.get('size'))
+    >>> aart = cli.albumart(track, 0)
+    >>> received = int(aart.get('binary'))
+    >>> size = int(aart.get('size'))
+    >>> with open('/tmp/cover', 'wb') as cover:
+    >>>     # aart = {'size': 42, 'binary': 2051, data: bytes(...)}
     >>>     cover.write(aart.get('data'))
     >>>     while received < size:
     >>>         aart = cli.albumart(track, received)
@@ -124,6 +124,30 @@ Fetching album covers is possible with albumart, here is an example:
     >>>         print('something went wrong', file=sys.stderr)
     >>> cli.disconnect()
 
+A `CommandError` is raised if the album does not expose a cover.
+
+You can also use `readpicture` command to fetch embedded picture:
+
+.. code-block:: python
+
+    >>> cli = musicpd.MPDClient()
+    >>> cli.connect()
+    >>> track = 'muse/Amon Tobin/2011-ISAM/01-Amon Tobin - Journeyman.mp3'
+    >>> rpict = cli.readpicture(track, 0)
+    >>> if not rpict:
+    >>>     print('No embedded picture found', file=sys.stderr)
+    >>>     sys.exit(1)
+    >>> size = int(rpict['size'])
+    >>> done = int(rpict['binary'])
+    >>> with open('/tmp/cover', 'wb') as cover:
+    >>>     cover.write(rpict['data'])
+    >>>     while size > done:
+    >>>         rpict = cli.readpicture(track, done)
+    >>>         done += int(rpict['binary'])
+    >>>         print(f'writing {rpict["binary"]}, done {100*done/size:03.0f}%')
+    >>>         cover.write(rpict['data'])
+    >>> cli.disconnect()
+
 Refer to `MPD protocol documentation`_ for the meaning of `binary`, `size` and `data`.
 
 .. _MPD protocol documentation: http://www.musicpd.org/doc/protocol/
diff --git a/musicpd.py b/musicpd.py
index 3425a6f..008d9d8 100644
--- a/musicpd.py
+++ b/musicpd.py
@@ -233,6 +233,7 @@ class MPDClient:
             "listfiles":          self._fetch_database,
             "lsinfo":             self._fetch_database,
             "readcomments":       self._fetch_object,
+            "readpicture":        self._fetch_composite,
             "search":             self._fetch_songs,
             "searchadd":          self._fetch_nothing,
             "searchaddpl":        self._fetch_nothing,
@@ -542,6 +543,10 @@ class MPDClient:
             obj[key] = value
             if key == 'binary':
                 break
+        if not obj:
+            # If the song file was recognized, but there is no picture, the
+            # response is successful, but is otherwise empty.
+            return obj
         amount = int(obj['binary'])
         try:
             obj['data'] = self._read_binary(amount)