ptp: return TV category if season/episode detected in title

This commit is contained in:
Bogdan 2023-10-20 00:21:09 +03:00
parent 6cc6d06e4b
commit 22f296b645
1 changed files with 11 additions and 1 deletions

View File

@ -5,6 +5,7 @@ using System.Diagnostics.CodeAnalysis;
using System.Globalization; using System.Globalization;
using System.Linq; using System.Linq;
using System.Net; using System.Net;
using System.Text.RegularExpressions;
using System.Threading.Tasks; using System.Threading.Tasks;
using Jackett.Common.Models; using Jackett.Common.Models;
using Jackett.Common.Models.IndexerConfig; using Jackett.Common.Models.IndexerConfig;
@ -162,6 +163,8 @@ namespace Jackett.Common.Indexers
results = await RequestWithCookiesAndRetryAsync(movieListSearchUrl, headers: authHeaders); results = await RequestWithCookiesAndRetryAsync(movieListSearchUrl, headers: authHeaders);
} }
var seasonRegex = new Regex(@"\bS\d{2,3}(E\d{2,3})?\b", RegexOptions.Compiled);
try try
{ {
//Iterate over the releases for each movie //Iterate over the releases for each movie
@ -226,6 +229,13 @@ namespace Jackett.Common.Indexers
var publishDate = DateTime.ParseExact((string)torrent["UploadTime"], "yyyy-MM-dd HH:mm:ss", CultureInfo.InvariantCulture, DateTimeStyles.AssumeUniversal).ToLocalTime(); var publishDate = DateTime.ParseExact((string)torrent["UploadTime"], "yyyy-MM-dd HH:mm:ss", CultureInfo.InvariantCulture, DateTimeStyles.AssumeUniversal).ToLocalTime();
var leechers = int.Parse((string)torrent["Leechers"]); var leechers = int.Parse((string)torrent["Leechers"]);
var categories = new List<int> { TorznabCatType.Movies.ID };
if (releaseName != null && seasonRegex.Match(releaseName).Success)
{
categories.Add(TorznabCatType.TV.ID);
}
var release = new ReleaseInfo var release = new ReleaseInfo
{ {
Guid = link, Guid = link,
@ -234,7 +244,7 @@ namespace Jackett.Common.Indexers
Title = releaseName, Title = releaseName,
Description = $"Title: {movieTitle}", Description = $"Title: {movieTitle}",
Year = int.Parse(year), Year = int.Parse(year),
Category = new List<int> { TorznabCatType.Movies.ID }, Category = categories,
Poster = poster, Poster = poster,
Imdb = movieImdbId, Imdb = movieImdbId,
Size = size, Size = size,