Merge pull request #433 from pkkid/allowMediaDeletion

allowMediaDeletion

Author: Hellowlol

plexapi/server.py

@@ -493,6 +493,25 @@ def refreshSync(self):
         self.refreshSynclist()
         self.refreshContent()
 
+    def _allowMediaDeletion(self, toggle=False):
+        """ Toggle allowMediaDeletion.
+            Parameters:
+                toggle (bool): True enables Media Deletion
+                               False or None disable Media Deletion (Default)
+        """
+        if self.allowMediaDeletion and toggle is False:
+            log.debug('Plex is currently allowed to delete media. Toggling off.')
+        elif self.allowMediaDeletion and toggle is True:
+            log.debug('Plex is currently allowed to delete media. Toggle set to allow, exiting.')
+            raise BadRequest('Plex is currently allowed to delete media. Toggle set to allow, exiting.')
+        elif self.allowMediaDeletion is None and toggle is True:
+            log.debug('Plex is currently not allowed to delete media. Toggle set to allow.')
+        else:
+            log.debug('Plex is currently not allowed to delete media. Toggle set to not allow, exiting.')
+            raise BadRequest('Plex is currently not allowed to delete media. Toggle set to not allow, exiting.')
+        value = 1 if toggle is True else 0
+        return self.query('/:/prefs?allowMediaDeletion=%s' % value, self._session.put)
+
 
 class Account(PlexObject):
     """ Contains the locally cached MyPlex account information. The properties provided don't

tests/test_server.py

@@ -269,3 +269,41 @@ def test_server_downloadLogs(tmpdir, plex):
 def test_server_downloadDatabases(tmpdir, plex):
     plex.downloadDatabases(savepath=str(tmpdir), unpack=True)
     assert len(tmpdir.listdir()) > 1
+
+def test_server_allowMediaDeletion(account):
+    plex = PlexServer(utils.SERVER_BASEURL, account.authenticationToken)
+    # Check server current allowMediaDeletion setting
+    if plex.allowMediaDeletion:
+        # If allowed then test disallowed
+        plex._allowMediaDeletion(False)
+        time.sleep(1)
+        plex = PlexServer(utils.SERVER_BASEURL, account.authenticationToken)
+        assert plex.allowMediaDeletion is None
+        # Test redundant toggle
+        with pytest.raises(BadRequest):
+            plex._allowMediaDeletion(False)
+
+        plex._allowMediaDeletion(True)
+        time.sleep(1)
+        plex = PlexServer(utils.SERVER_BASEURL, account.authenticationToken)
+        assert plex.allowMediaDeletion is True
+        # Test redundant toggle
+        with pytest.raises(BadRequest):
+            plex._allowMediaDeletion(True)
+    else:
+        # If disallowed then test allowed
+        plex._allowMediaDeletion(True)
+        time.sleep(1)
+        plex = PlexServer(utils.SERVER_BASEURL, account.authenticationToken)
+        assert plex.allowMediaDeletion is True
+        # Test redundant toggle
+        with pytest.raises(BadRequest):
+            plex._allowMediaDeletion(True)
+
+        plex._allowMediaDeletion(False)
+        time.sleep(1)
+        plex = PlexServer(utils.SERVER_BASEURL, account.authenticationToken)
+        assert plex.allowMediaDeletion is None
+        # Test redundant toggle
+        with pytest.raises(BadRequest):
+            plex._allowMediaDeletion(False)