Added support for handling existing target account viewstate

Existing target account's viewstate can be added to or replaced

Author: cetteup

README.md

@@ -42,5 +42,4 @@ If you have Python >=3.6 installed on your system, you can use the Python script
 
 ## Known limitations
 - users/accounts only appear in database after connecting to the server and/or playing a media item from it at least once
-- script currently does not check whether target user already has viewstate information for a library item (may result in two conflicting viewstates)
 - Tautulli ([A Python based monitoring and tracking tool for Plex Media Server](https://tautulli.com)) will not pick up any of the transferred viewstate information

transfer-plex-user-viewstate.py

@@ -1,5 +1,6 @@
 import argparse
 import sqlite3
+import sys
 
 from tabulate import tabulate
 
@@ -21,7 +22,7 @@ def str_range(stop: int) -> list:
     return [str(v) for v in values]
 
 
-parser = argparse.ArgumentParser('Transfer Plex user viewstate between users')
+parser = argparse.ArgumentParser("Transfer Plex user viewstate between users")
 parser.add_argument("-p", "--db-path", help="Path to com.plexapp.plugins.library.db file")
 args = parser.parse_args()
 
@@ -45,34 +46,75 @@ def str_range(stop: int) -> list:
 ))
 
 # Ask user to select target user
-targetUserIndex = int(get_valid_input(
+targetAccountIndex = int(get_valid_input(
     "Now, please enter the _target_ account index",
     str_range(len(accounts)),
     "No account with that ID, please enter a valid account index"
 ))
 
 # Ask user to select copying/moving view state
 mode = get_valid_input(
-    "Do you want to copy or move the viewstate? (copy/move)",
+    "Do you want to copy or move the viewstate? ([c]opy/[m]move)",
     ["c", "copy", "m", "move"],
-    "Please select either (c)opy or (m)ove"
+    "Please select either [c]opy or [m]ove"
 )
 
 # Get source account viewstate
 sql = "SELECT * FROM metadata_item_settings WHERE account_id = :account_id"
-cursor.execute(sql, {"account_id": accounts[sourceAccountIndex]['id']})
+cursor.execute(sql, {"account_id": accounts[sourceAccountIndex]["id"]})
 sourceViewstate = cursor.fetchall()
 
+# Make sure source account has any viewstate items
+if len(sourceViewstate) == 0:
+    sys.exit("Source account has no viewstate items")
+
+# Check for existing viewstate items of target user
+sql = "SELECT id, guid FROM metadata_item_settings WHERE account_id = :account_id"
+cursor.execute(sql, {"account_id": accounts[targetAccountIndex]['id']})
+targetViewstate = cursor.fetchall()
+
+# Handle existing target account viewstates items if any were found
+if len(targetViewstate) > 0:
+    print(f"Found {len(targetViewstate)} existing viewstate items for target account")
+    handleExisting = get_valid_input(
+        "Do you want to add to or replace existing viewstate? ([a]dd/[r]eplace)",
+        ["a", "add", "r", "replace"],
+        "Please select either [a]dd/[r]eplace"
+    )
+
+    if handleExisting in ["a", "add"]:
+        # Remove source account's viewstate for any media items for which target user has an existing viewstate
+        sourceViewstateGuids = [e["guid"] for e in sourceViewstate]
+        targetViewstateGuids = [e["guid"] for e in targetViewstate]
+        for guid in targetViewstateGuids:
+            # Check if guid is also present in source viewstate
+            if guid in sourceViewstateGuids:
+                # Get index of media item
+                # (indexes are the same between guid list and overall source viewstate list,
+                # so we can use the index to remove the media item's source viewstate from the copy/move list)
+                index = sourceViewstateGuids.index(guid)
+                # Remove viewstate for media item from copy/move list
+                del sourceViewstate[index], sourceViewstateGuids[index]
+
+                # If we are moving the viewstate, remove viewstate items we don't need to move from source user
+                sql = "DELETE FROM metadata_item_settings WHERE account_id = :account_id AND guid = :guid"
+                cursor.execute(sql, {"account_id": accounts[sourceAccountIndex]["id"], "guid": guid})
+    elif handleExisting in ["r", "replace"]:
+        # Remove existing viewstate items
+        for entry in targetViewstate:
+            sql = "DELETE FROM metadata_item_settings WHERE id = :id"
+            cursor.execute(sql, {"id": entry["id"]})
+
 if mode in ["c", "copy"]:
     print(f"Copying viewstate ({len(sourceViewstate)} items) "
-          f"from {accounts[sourceAccountIndex]['name']} to {accounts[targetUserIndex]['name']}")
+          f"from {accounts[sourceAccountIndex]['name']} to {accounts[targetAccountIndex]['name']}")
 
     for entry in sourceViewstate:
         sql = "INSERT INTO metadata_item_settings (account_id, guid, rating, view_offset, view_count, last_viewed_at, " \
               "created_at, updated_at, skip_count, last_skipped_at, changed_at, extra_data, last_rated_at) VALUES (?, " \
               "?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) "
         cursor.execute(sql, (
-            accounts[targetUserIndex]["id"],
+            accounts[targetAccountIndex]["id"],
             entry["guid"],
             entry["rating"],
             entry["view_offset"],
@@ -90,11 +132,11 @@ def str_range(stop: int) -> list:
     connection.commit()
 elif mode in ["m", "move"]:
     print(f"Moving viewstate ({len(sourceViewstate)} items) "
-          f"from {accounts[sourceAccountIndex]['name']} to {accounts[targetUserIndex]['name']}")
+          f"from {accounts[sourceAccountIndex]['name']} to {accounts[targetAccountIndex]['name']}")
 
     sql = "UPDATE metadata_item_settings SET account_id = :target_account_id WHERE account_id = :source_account_id"
     cursor.execute(sql, {
-        "target_account_id": accounts[targetUserIndex]["id"],
+        "target_account_id": accounts[targetAccountIndex]["id"],
         "source_account_id": accounts[sourceAccountIndex]["id"]
     })