488 lines
14 KiB
Python
488 lines
14 KiB
Python
import sys
|
|
import os
|
|
|
|
# getting the name of the directory
|
|
# where the this file is present.
|
|
current = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
# Getting the parent directory name
|
|
# where the current directory is present.
|
|
parent = os.path.dirname(current)
|
|
|
|
# adding the parent directory to
|
|
# the sys.path.
|
|
sys.path.append(parent)
|
|
|
|
from src.watched import cleanup_watched, combine_watched_dicts
|
|
|
|
tv_shows_watched_list_1 = {
|
|
frozenset(
|
|
{
|
|
("tvdb", "75710"),
|
|
("title", "Criminal Minds"),
|
|
("imdb", "tt0452046"),
|
|
("locations", ("Criminal Minds",)),
|
|
("tmdb", "4057"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt0550489",
|
|
"tmdb": "282843",
|
|
"tvdb": "176357",
|
|
"title": "Extreme Aggressor",
|
|
"locations": (
|
|
"Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"imdb": "tt0550487",
|
|
"tmdb": "282861",
|
|
"tvdb": "300385",
|
|
"title": "Compulsion",
|
|
"locations": ("Criminal Minds S01E02 Compulsion WEBDL-720p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
},
|
|
frozenset({("title", "Test"), ("locations", ("Test",))}): {
|
|
"Season 1": [
|
|
{
|
|
"title": "S01E01",
|
|
"locations": ("Test S01E01.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "S01E02",
|
|
"locations": ("Test S01E02.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "S01E04",
|
|
"locations": ("Test S01E04.mkv",),
|
|
"status": {"completed": False, "time": 5},
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
tv_shows_watched_list_2 = {
|
|
frozenset(
|
|
{
|
|
("tvdb", "75710"),
|
|
("title", "Criminal Minds"),
|
|
("imdb", "tt0452046"),
|
|
("locations", ("Criminal Minds",)),
|
|
("tmdb", "4057"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt0550487",
|
|
"tmdb": "282861",
|
|
"tvdb": "300385",
|
|
"title": "Compulsion",
|
|
"locations": ("Criminal Minds S01E02 Compulsion WEBDL-720p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"imdb": "tt0550498",
|
|
"tmdb": "282865",
|
|
"tvdb": "300474",
|
|
"title": "Won't Get Fooled Again",
|
|
"locations": (
|
|
"Criminal Minds S01E03 Won't Get Fooled Again WEBDL-720p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
},
|
|
frozenset({("title", "Test"), ("locations", ("Test",))}): {
|
|
"Season 1": [
|
|
{
|
|
"title": "S01E02",
|
|
"locations": ("Test S01E02.mkv",),
|
|
"status": {"completed": False, "time": 10},
|
|
},
|
|
{
|
|
"title": "S01E03",
|
|
"locations": ("Test S01E03.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "S01E04",
|
|
"locations": ("Test S01E04.mkv",),
|
|
"status": {"completed": False, "time": 10},
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
expected_tv_show_watched_list_1 = {
|
|
frozenset(
|
|
{
|
|
("tvdb", "75710"),
|
|
("title", "Criminal Minds"),
|
|
("imdb", "tt0452046"),
|
|
("locations", ("Criminal Minds",)),
|
|
("tmdb", "4057"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt0550489",
|
|
"tmdb": "282843",
|
|
"tvdb": "176357",
|
|
"title": "Extreme Aggressor",
|
|
"locations": (
|
|
"Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
},
|
|
frozenset({("title", "Test"), ("locations", ("Test",))}): {
|
|
"Season 1": [
|
|
{
|
|
"title": "S01E01",
|
|
"locations": ("Test S01E01.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "S01E02",
|
|
"locations": ("Test S01E02.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
expected_tv_show_watched_list_2 = {
|
|
frozenset(
|
|
{
|
|
("tvdb", "75710"),
|
|
("title", "Criminal Minds"),
|
|
("imdb", "tt0452046"),
|
|
("locations", ("Criminal Minds",)),
|
|
("tmdb", "4057"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt0550498",
|
|
"tmdb": "282865",
|
|
"tvdb": "300474",
|
|
"title": "Won't Get Fooled Again",
|
|
"locations": (
|
|
"Criminal Minds S01E03 Won't Get Fooled Again WEBDL-720p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
},
|
|
frozenset({("title", "Test"), ("locations", ("Test",))}): {
|
|
"Season 1": [
|
|
{
|
|
"title": "S01E03",
|
|
"locations": ("Test S01E03.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "S01E04",
|
|
"locations": ("Test S01E04.mkv",),
|
|
"status": {"completed": False, "time": 10},
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
movies_watched_list_1 = [
|
|
{
|
|
"imdb": "tt2380307",
|
|
"tmdb": "354912",
|
|
"title": "Coco",
|
|
"locations": ("Coco (2017) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"tmdbcollection": "448150",
|
|
"imdb": "tt1431045",
|
|
"tmdb": "293660",
|
|
"title": "Deadpool",
|
|
"locations": ("Deadpool (2016) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
|
|
movies_watched_list_2 = [
|
|
{
|
|
"imdb": "tt2380307",
|
|
"tmdb": "354912",
|
|
"title": "Coco",
|
|
"locations": ("Coco (2017) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"imdb": "tt0384793",
|
|
"tmdb": "9788",
|
|
"tvdb": "9103",
|
|
"title": "Accepted",
|
|
"locations": ("Accepted (2006) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
|
|
|
|
expected_movie_watched_list_1 = [
|
|
{
|
|
"tmdbcollection": "448150",
|
|
"imdb": "tt1431045",
|
|
"tmdb": "293660",
|
|
"title": "Deadpool",
|
|
"locations": ("Deadpool (2016) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
|
|
expected_movie_watched_list_2 = [
|
|
{
|
|
"imdb": "tt0384793",
|
|
"tmdb": "9788",
|
|
"tvdb": "9103",
|
|
"title": "Accepted",
|
|
"locations": ("Accepted (2006) Remux-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
|
|
# Test to see if objects get deleted all the way up to the root.
|
|
tv_shows_2_watched_list_1 = {
|
|
frozenset(
|
|
{
|
|
("tvdb", "75710"),
|
|
("title", "Criminal Minds"),
|
|
("imdb", "tt0452046"),
|
|
("locations", ("Criminal Minds",)),
|
|
("tmdb", "4057"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt0550489",
|
|
"tmdb": "282843",
|
|
"tvdb": "176357",
|
|
"title": "Extreme Aggressor",
|
|
"locations": (
|
|
"Criminal Minds S01E01 Extreme Aggressor WEBDL-720p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
}
|
|
}
|
|
|
|
|
|
def test_simple_cleanup_watched():
|
|
user_watched_list_1 = {
|
|
"user1": {
|
|
"TV Shows": tv_shows_watched_list_1,
|
|
"Movies": movies_watched_list_1,
|
|
"Other Shows": tv_shows_2_watched_list_1,
|
|
},
|
|
}
|
|
user_watched_list_2 = {
|
|
"user1": {
|
|
"TV Shows": tv_shows_watched_list_2,
|
|
"Movies": movies_watched_list_2,
|
|
"Other Shows": tv_shows_2_watched_list_1,
|
|
}
|
|
}
|
|
|
|
expected_watched_list_1 = {
|
|
"user1": {
|
|
"TV Shows": expected_tv_show_watched_list_1,
|
|
"Movies": expected_movie_watched_list_1,
|
|
}
|
|
}
|
|
|
|
expected_watched_list_2 = {
|
|
"user1": {
|
|
"TV Shows": expected_tv_show_watched_list_2,
|
|
"Movies": expected_movie_watched_list_2,
|
|
}
|
|
}
|
|
|
|
return_watched_list_1 = cleanup_watched(user_watched_list_1, user_watched_list_2)
|
|
return_watched_list_2 = cleanup_watched(user_watched_list_2, user_watched_list_1)
|
|
|
|
assert return_watched_list_1 == expected_watched_list_1
|
|
assert return_watched_list_2 == expected_watched_list_2
|
|
|
|
|
|
def test_mapping_cleanup_watched():
|
|
user_watched_list_1 = {
|
|
"user1": {
|
|
"TV Shows": tv_shows_watched_list_1,
|
|
"Movies": movies_watched_list_1,
|
|
"Other Shows": tv_shows_2_watched_list_1,
|
|
},
|
|
}
|
|
user_watched_list_2 = {
|
|
"user2": {
|
|
"Shows": tv_shows_watched_list_2,
|
|
"Movies": movies_watched_list_2,
|
|
"Other Shows": tv_shows_2_watched_list_1,
|
|
}
|
|
}
|
|
|
|
expected_watched_list_1 = {
|
|
"user1": {
|
|
"TV Shows": expected_tv_show_watched_list_1,
|
|
"Movies": expected_movie_watched_list_1,
|
|
}
|
|
}
|
|
|
|
expected_watched_list_2 = {
|
|
"user2": {
|
|
"Shows": expected_tv_show_watched_list_2,
|
|
"Movies": expected_movie_watched_list_2,
|
|
}
|
|
}
|
|
|
|
user_mapping = {"user1": "user2"}
|
|
library_mapping = {"TV Shows": "Shows"}
|
|
|
|
return_watched_list_1 = cleanup_watched(
|
|
user_watched_list_1,
|
|
user_watched_list_2,
|
|
user_mapping=user_mapping,
|
|
library_mapping=library_mapping,
|
|
)
|
|
return_watched_list_2 = cleanup_watched(
|
|
user_watched_list_2,
|
|
user_watched_list_1,
|
|
user_mapping=user_mapping,
|
|
library_mapping=library_mapping,
|
|
)
|
|
|
|
assert return_watched_list_1 == expected_watched_list_1
|
|
assert return_watched_list_2 == expected_watched_list_2
|
|
|
|
|
|
def test_combine_watched_dicts():
|
|
input_watched = [
|
|
{
|
|
"test3": {
|
|
"Anime Movies": [
|
|
{
|
|
"title": "Ponyo",
|
|
"tmdb": "12429",
|
|
"imdb": "tt0876563",
|
|
"locations": ("Ponyo (2008) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "Spirited Away",
|
|
"tmdb": "129",
|
|
"imdb": "tt0245429",
|
|
"locations": ("Spirited Away (2001) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "Castle in the Sky",
|
|
"tmdb": "10515",
|
|
"imdb": "tt0092067",
|
|
"locations": ("Castle in the Sky (1986) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
]
|
|
}
|
|
},
|
|
{"test3": {"Anime Shows": {}}},
|
|
{"test3": {"Cartoon Shows": {}}},
|
|
{
|
|
"test3": {
|
|
"Shows": {
|
|
frozenset(
|
|
{
|
|
("tmdb", "64464"),
|
|
("tvdb", "301824"),
|
|
("tvrage", "45210"),
|
|
("title", "11.22.63"),
|
|
("locations", ("11.22.63",)),
|
|
("imdb", "tt2879552"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt4460418",
|
|
"title": "The Rabbit Hole",
|
|
"locations": (
|
|
"11.22.63 S01E01 The Rabbit Hole Bluray-1080p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{"test3": {"Subbed Anime": {}}},
|
|
]
|
|
expected = {
|
|
"test3": {
|
|
"Anime Movies": [
|
|
{
|
|
"title": "Ponyo",
|
|
"tmdb": "12429",
|
|
"imdb": "tt0876563",
|
|
"locations": ("Ponyo (2008) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "Spirited Away",
|
|
"tmdb": "129",
|
|
"imdb": "tt0245429",
|
|
"locations": ("Spirited Away (2001) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
{
|
|
"title": "Castle in the Sky",
|
|
"tmdb": "10515",
|
|
"imdb": "tt0092067",
|
|
"locations": ("Castle in the Sky (1986) Bluray-1080p.mkv",),
|
|
"status": {"completed": True, "time": 0},
|
|
},
|
|
],
|
|
"Anime Shows": {},
|
|
"Cartoon Shows": {},
|
|
"Shows": {
|
|
frozenset(
|
|
{
|
|
("tmdb", "64464"),
|
|
("tvdb", "301824"),
|
|
("tvrage", "45210"),
|
|
("title", "11.22.63"),
|
|
("locations", ("11.22.63",)),
|
|
("imdb", "tt2879552"),
|
|
}
|
|
): {
|
|
"Season 1": [
|
|
{
|
|
"imdb": "tt4460418",
|
|
"title": "The Rabbit Hole",
|
|
"locations": (
|
|
"11.22.63 S01E01 The Rabbit Hole Bluray-1080p.mkv",
|
|
),
|
|
"status": {"completed": True, "time": 0},
|
|
}
|
|
]
|
|
}
|
|
},
|
|
"Subbed Anime": {},
|
|
}
|
|
}
|
|
|
|
assert combine_watched_dicts(input_watched) == expected
|