shpaq/Prowlarr-Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr.git synced 2025-10-01 07:55:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

Support categories with Transmission

Browse Source
This commit is contained in:
Bogdan
2023-07-17 19:39:51 +03:00
parent d153746a98
commit efd5e92ca5
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/NzbDrone.Core/Download/Clients/Transmission/Transmission.cs
View File

@@ -38,6 +38,6 @@ namespace NzbDrone.Core.Download.Clients.Transmission
}
public override string Name => "Transmission";
public override bool SupportsCategories => false;
public override bool SupportsCategories => true;
}
}

16
src/NzbDrone.Core/Download/Clients/Transmission/TransmissionBase.cs
View File

@@ -28,7 +28,10 @@ namespace NzbDrone.Core.Download.Clients.Transmission
protected override string AddFromMagnetLink(TorrentInfo release, string hash, string magnetLink)
{
_proxy.AddTorrentFromUrl(magnetLink, GetDownloadDirectory(), Settings);
var category = GetCategoryForRelease(release) ?? Settings.Category;
var downloadDirectory = GetDownloadDirectory(category);
_proxy.AddTorrentFromUrl(magnetLink, downloadDirectory, Settings);
_proxy.SetTorrentSeedingConfiguration(hash, release.SeedConfiguration, Settings);
if (Settings.Priority == (int)TransmissionPriority.First)
@@ -41,7 +44,10 @@ namespace NzbDrone.Core.Download.Clients.Transmission
protected override string AddFromTorrentFile(TorrentInfo release, string hash, string filename, byte[] fileContent)
{
_proxy.AddTorrentFromData(fileContent, GetDownloadDirectory(), Settings);
var category = GetCategoryForRelease(release) ?? Settings.Category;
var downloadDirectory = GetDownloadDirectory(category);
_proxy.AddTorrentFromData(fileContent, downloadDirectory, Settings);
_proxy.SetTorrentSeedingConfiguration(hash, release.SeedConfiguration, Settings);
if (Settings.Priority == (int)TransmissionPriority.First)
@@ -73,14 +79,14 @@ namespace NzbDrone.Core.Download.Clients.Transmission
return outputPath + torrent.Name.Replace(":", "_");
}
protected string GetDownloadDirectory()
protected string GetDownloadDirectory(string category)
{
if (Settings.Directory.IsNotNullOrWhiteSpace())
{
return Settings.Directory;
}
if (!Settings.Category.IsNotNullOrWhiteSpace())
if (category.IsNullOrWhiteSpace())
{
return null;
}
@@ -88,7 +94,7 @@ namespace NzbDrone.Core.Download.Clients.Transmission
var config = _proxy.GetConfig(Settings);
var destDir = config.DownloadDir;
return $"{destDir.TrimEnd('/')}/{Settings.Category}";
return $"{destDir.TrimEnd('/')}/{category}";
}
protected ValidationFailure TestConnection()