Reformat project

Author: Hamuko

beetsplug/__init__.py

@@ -24,22 +24,22 @@
 from beets.plugins import BeetsPlugin
 
 
-class MusicPlayerDaemonClient():
+class MusicPlayerDaemonClient:
     """Simple socket client to provide connectivity to a Music Player Daemon
     server for updates and queue management.
     """
 
-    def __init__(self, host='localhost', port=6600, password=None):
+    def __init__(self, host="localhost", port=6600, password=None):
         self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.sock.connect((host, port))
         self.sock.settimeout(0.25)
         acknowledgement = self._read()[0]
-        if not acknowledgement.startswith('OK MPD'):
+        if not acknowledgement.startswith("OK MPD"):
             self.close()
         if password:
             self._send('password "{}"'.format(password))
             response = self._read()[0]
-            if not response.startswith('OK'):
+            if not response.startswith("OK"):
                 self.close()
 
     def _read(self):
@@ -48,7 +48,7 @@ def _read(self):
         This operation may return an empty list if the server does not send
         anything during the socket timeout period.
         """
-        data = b''
+        data = b""
         while True:
             try:
                 data_buffer = self.sock.recv(1024)
@@ -57,28 +57,27 @@ def _read(self):
             if not data_buffer:
                 break
             data += data_buffer
-        return data.decode().strip().split('\n')
+        return data.decode().strip().split("\n")
 
     def _send(self, data):
         """Sends a string to the server."""
-        if data[-1] != '\n':
-            data = ''.join([data, '\n'])
+        if data[-1] != "\n":
+            data = "".join([data, "\n"])
         self.sock.send(data.encode())
 
     def add(self, path):
         """Adds given path to MPD queue."""
         self._send('add "{}"'.format(path))
-        return self._read()[0] == 'OK'
+        return self._read()[0] == "OK"
 
     def close(self):
         """Closes the connection to the server."""
-        self._send('close')
+        self._send("close")
         self.sock.close()
 
     def status(self):
-        """Fetches status from Music Player Daemon. Returns a tuple of strings.
-        """
-        self._send('status')
+        """Fetches status from Music Player Daemon. Returns a tuple of strings."""
+        self._send("status")
         return self._read()
 
     def update(self, directory):
@@ -88,12 +87,12 @@ def update(self, directory):
         """
         self._send('update "{}"'.format(directory))
         response = self._read()
-        if response[-1] != 'OK':
+        if response[-1] != "OK":
             return
         while True:
             updating = False
             for line in self.status():
-                if line.startswith('updating_db'):
+                if line.startswith("updating_db"):
                     updating = True
             if not updating:
                 break
@@ -108,17 +107,19 @@ class MPDQueuePlugin(BeetsPlugin):
 
     def __init__(self):
         super(MPDQueuePlugin, self).__init__()
-        config['mpd'].add({
-            'host': os.environ.get('MPD_HOST', 'localhost'),
-            'port': 6600,
-            'password': '',
-        })
-        config['mpd']['password'].redact = True
+        config["mpd"].add(
+            {
+                "host": os.environ.get("MPD_HOST", "localhost"),
+                "port": 6600,
+                "password": "",
+            }
+        )
+        config["mpd"]["password"].redact = True
 
         self.files = []
 
-        self.register_listener('import_task_files', self.import_task_files)
-        self.register_listener('cli_exit', self.update_queue)
+        self.register_listener("import_task_files", self.import_task_files)
+        self.register_listener("cli_exit", self.update_queue)
 
     def import_task_files(self, task, session):
         """Track all the files added during an import operation so they can be
@@ -128,14 +129,14 @@ def import_task_files(self, task, session):
         property as it indicates a reimport of existing library files.
         """
         if not task.toppath:
-            self._log.debug(u'Skipping library re-import')
+            self._log.debug("Skipping library re-import")
             return
 
         tracks = []
         items = sorted(task.imported_items(), key=lambda x: x.track)
         for item in items:
             destination = item.destination(relative_to_libdir=True).decode()
-            self._log.debug(u'{0} will be added to queue', destination)
+            self._log.debug("{0} will be added to queue", destination)
             tracks.append(destination)
         self.files += tracks
 
@@ -149,27 +150,27 @@ def update_queue(self, lib):
         files are in the database when they are added.
         """
         if not self.files:
-            self._log.debug(u'No files to add to queue')
+            self._log.debug("No files to add to queue")
             return
 
-        host = config['mpd']['host'].as_str()
-        port = config['mpd']['port'].get(int)
-        password = config['mpd']['password'].as_str()
+        host = config["mpd"]["host"].as_str()
+        port = config["mpd"]["port"].get(int)
+        password = config["mpd"]["password"].as_str()
         client = MusicPlayerDaemonClient(host, port, password)
 
         directories = set()
         for file_ in self.files:
             directories.add(os.path.dirname(file_))
         for directory in directories:
-            self._log.debug(u'Updating directory {0}', directory)
+            self._log.debug("Updating directory {0}", directory)
             client.update(directory)
-            self._log.debug(u'Finished updating {0}', directory)
+            self._log.debug("Finished updating {0}", directory)
 
         for file_ in self.files:
-            self._log.debug(u'Adding {0} to queue', file_)
+            self._log.debug("Adding {0} to queue", file_)
             success = client.add(file_)
             if success:
-                self._log.debug(u'Added {0} to queue', file_)
+                self._log.debug("Added {0} to queue", file_)
             else:
-                self._log.warning(u'Failed to add {0} to queue', file_)
+                self._log.warning("Failed to add {0} to queue", file_)
         client.close()