New: Auto tag series based on tags present/absent on series
Closes #6236
This commit is contained in:
parent
5061dc4b5e
commit
f4c19a384b
|
@ -22,6 +22,7 @@ import PasswordInput from './PasswordInput';
|
||||||
import PathInputConnector from './PathInputConnector';
|
import PathInputConnector from './PathInputConnector';
|
||||||
import QualityProfileSelectInputConnector from './QualityProfileSelectInputConnector';
|
import QualityProfileSelectInputConnector from './QualityProfileSelectInputConnector';
|
||||||
import RootFolderSelectInputConnector from './RootFolderSelectInputConnector';
|
import RootFolderSelectInputConnector from './RootFolderSelectInputConnector';
|
||||||
|
import SeriesTagInput from './SeriesTagInput';
|
||||||
import SeriesTypeSelectInput from './SeriesTypeSelectInput';
|
import SeriesTypeSelectInput from './SeriesTypeSelectInput';
|
||||||
import TagInputConnector from './TagInputConnector';
|
import TagInputConnector from './TagInputConnector';
|
||||||
import TagSelectInputConnector from './TagSelectInputConnector';
|
import TagSelectInputConnector from './TagSelectInputConnector';
|
||||||
|
@ -87,6 +88,9 @@ function getComponent(type) {
|
||||||
case inputTypes.DYNAMIC_SELECT:
|
case inputTypes.DYNAMIC_SELECT:
|
||||||
return EnhancedSelectInputConnector;
|
return EnhancedSelectInputConnector;
|
||||||
|
|
||||||
|
case inputTypes.SERIES_TAG:
|
||||||
|
return SeriesTagInput;
|
||||||
|
|
||||||
case inputTypes.SERIES_TYPE_SELECT:
|
case inputTypes.SERIES_TYPE_SELECT:
|
||||||
return SeriesTypeSelectInput;
|
return SeriesTypeSelectInput;
|
||||||
|
|
||||||
|
|
|
@ -27,6 +27,8 @@ function getType({ type, selectOptionsProviderAction }) {
|
||||||
return inputTypes.DYNAMIC_SELECT;
|
return inputTypes.DYNAMIC_SELECT;
|
||||||
}
|
}
|
||||||
return inputTypes.SELECT;
|
return inputTypes.SELECT;
|
||||||
|
case 'seriesTag':
|
||||||
|
return inputTypes.SERIES_TAG;
|
||||||
case 'tag':
|
case 'tag':
|
||||||
return inputTypes.TEXT_TAG;
|
return inputTypes.TEXT_TAG;
|
||||||
case 'tagSelect':
|
case 'tagSelect':
|
||||||
|
|
|
@ -0,0 +1,53 @@
|
||||||
|
import React, { useCallback } from 'react';
|
||||||
|
import TagInputConnector from './TagInputConnector';
|
||||||
|
|
||||||
|
interface SeriesTageInputProps {
|
||||||
|
name: string;
|
||||||
|
value: number | number[];
|
||||||
|
onChange: ({
|
||||||
|
name,
|
||||||
|
value,
|
||||||
|
}: {
|
||||||
|
name: string;
|
||||||
|
value: number | number[];
|
||||||
|
}) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export default function SeriesTagInput(props: SeriesTageInputProps) {
|
||||||
|
const { value, onChange, ...otherProps } = props;
|
||||||
|
const isArray = Array.isArray(value);
|
||||||
|
|
||||||
|
const handleChange = useCallback(
|
||||||
|
({ name, value: newValue }: { name: string; value: number[] }) => {
|
||||||
|
if (isArray) {
|
||||||
|
onChange({ name, value: newValue });
|
||||||
|
} else {
|
||||||
|
onChange({
|
||||||
|
name,
|
||||||
|
value: newValue.length ? newValue[newValue.length - 1] : 0,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[isArray, onChange]
|
||||||
|
);
|
||||||
|
|
||||||
|
let finalValue: number[] = [];
|
||||||
|
|
||||||
|
if (isArray) {
|
||||||
|
finalValue = value;
|
||||||
|
} else if (value === 0) {
|
||||||
|
finalValue = [];
|
||||||
|
} else {
|
||||||
|
finalValue = [value];
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
|
||||||
|
// @ts-ignore 2786 'TagInputConnector' isn't typed yet
|
||||||
|
<TagInputConnector
|
||||||
|
{...otherProps}
|
||||||
|
value={finalValue}
|
||||||
|
onChange={handleChange}
|
||||||
|
/>
|
||||||
|
);
|
||||||
|
}
|
|
@ -17,6 +17,7 @@ export const LANGUAGE_SELECT = 'languageSelect';
|
||||||
export const DOWNLOAD_CLIENT_SELECT = 'downloadClientSelect';
|
export const DOWNLOAD_CLIENT_SELECT = 'downloadClientSelect';
|
||||||
export const ROOT_FOLDER_SELECT = 'rootFolderSelect';
|
export const ROOT_FOLDER_SELECT = 'rootFolderSelect';
|
||||||
export const SELECT = 'select';
|
export const SELECT = 'select';
|
||||||
|
export const SERIES_TAG = 'seriesTag';
|
||||||
export const DYNAMIC_SELECT = 'dynamicSelect';
|
export const DYNAMIC_SELECT = 'dynamicSelect';
|
||||||
export const SERIES_TYPE_SELECT = 'seriesTypeSelect';
|
export const SERIES_TYPE_SELECT = 'seriesTypeSelect';
|
||||||
export const TAG = 'tag';
|
export const TAG = 'tag';
|
||||||
|
@ -45,6 +46,7 @@ export const all = [
|
||||||
ROOT_FOLDER_SELECT,
|
ROOT_FOLDER_SELECT,
|
||||||
LANGUAGE_SELECT,
|
LANGUAGE_SELECT,
|
||||||
SELECT,
|
SELECT,
|
||||||
|
SERIES_TAG,
|
||||||
DYNAMIC_SELECT,
|
DYNAMIC_SELECT,
|
||||||
SERIES_TYPE_SELECT,
|
SERIES_TYPE_SELECT,
|
||||||
TAG,
|
TAG,
|
||||||
|
|
|
@ -12,7 +12,7 @@ export default function TagInUse(props) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (count > 1 && labelPlural ) {
|
if (count > 1 && labelPlural) {
|
||||||
return (
|
return (
|
||||||
<div>
|
<div>
|
||||||
{count} {labelPlural.toLowerCase()}
|
{count} {labelPlural.toLowerCase()}
|
||||||
|
|
|
@ -1,6 +1,9 @@
|
||||||
|
using System.Collections.Generic;
|
||||||
using FizzWare.NBuilder;
|
using FizzWare.NBuilder;
|
||||||
using FluentAssertions;
|
using FluentAssertions;
|
||||||
using NUnit.Framework;
|
using NUnit.Framework;
|
||||||
|
using NzbDrone.Core.AutoTagging;
|
||||||
|
using NzbDrone.Core.AutoTagging.Specifications;
|
||||||
using NzbDrone.Core.Housekeeping.Housekeepers;
|
using NzbDrone.Core.Housekeeping.Housekeepers;
|
||||||
using NzbDrone.Core.Profiles.Releases;
|
using NzbDrone.Core.Profiles.Releases;
|
||||||
using NzbDrone.Core.Tags;
|
using NzbDrone.Core.Tags;
|
||||||
|
@ -45,5 +48,35 @@ namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
|
||||||
Subject.Clean();
|
Subject.Clean();
|
||||||
AllStoredModels.Should().HaveCount(1);
|
AllStoredModels.Should().HaveCount(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_not_delete_used_auto_tagging_tag_specification_tags()
|
||||||
|
{
|
||||||
|
var tags = Builder<Tag>
|
||||||
|
.CreateListOfSize(2)
|
||||||
|
.All()
|
||||||
|
.With(x => x.Id = 0)
|
||||||
|
.BuildList();
|
||||||
|
Db.InsertMany(tags);
|
||||||
|
|
||||||
|
var autoTags = Builder<AutoTag>.CreateListOfSize(1)
|
||||||
|
.All()
|
||||||
|
.With(x => x.Id = 0)
|
||||||
|
.With(x => x.Specifications = new List<IAutoTaggingSpecification>
|
||||||
|
{
|
||||||
|
new TagSpecification
|
||||||
|
{
|
||||||
|
Name = "Test",
|
||||||
|
Value = tags[0].Id
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.BuildList();
|
||||||
|
|
||||||
|
Mocker.GetMock<IAutoTaggingRepository>().Setup(s => s.All())
|
||||||
|
.Returns(autoTags);
|
||||||
|
|
||||||
|
Subject.Clean();
|
||||||
|
AllStoredModels.Should().HaveCount(1);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -85,7 +85,8 @@ namespace NzbDrone.Core.Annotations
|
||||||
Device,
|
Device,
|
||||||
TagSelect,
|
TagSelect,
|
||||||
RootFolder,
|
RootFolder,
|
||||||
QualityProfile
|
QualityProfile,
|
||||||
|
SeriesTag
|
||||||
}
|
}
|
||||||
|
|
||||||
public enum HiddenType
|
public enum HiddenType
|
||||||
|
|
|
@ -0,0 +1,36 @@
|
||||||
|
using FluentValidation;
|
||||||
|
using NzbDrone.Core.Annotations;
|
||||||
|
using NzbDrone.Core.Tv;
|
||||||
|
using NzbDrone.Core.Validation;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.AutoTagging.Specifications
|
||||||
|
{
|
||||||
|
public class TagSpecificationValidator : AbstractValidator<TagSpecification>
|
||||||
|
{
|
||||||
|
public TagSpecificationValidator()
|
||||||
|
{
|
||||||
|
RuleFor(c => c.Value).GreaterThan(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class TagSpecification : AutoTaggingSpecificationBase
|
||||||
|
{
|
||||||
|
private static readonly TagSpecificationValidator Validator = new ();
|
||||||
|
|
||||||
|
public override int Order => 1;
|
||||||
|
public override string ImplementationName => "Tag";
|
||||||
|
|
||||||
|
[FieldDefinition(1, Label = "AutoTaggingSpecificationTag", Type = FieldType.SeriesTag)]
|
||||||
|
public int Value { get; set; }
|
||||||
|
|
||||||
|
protected override bool IsSatisfiedByWithoutNegate(Series series)
|
||||||
|
{
|
||||||
|
return series.Tags.Contains(Value);
|
||||||
|
}
|
||||||
|
|
||||||
|
public override NzbDroneValidationResult Validate()
|
||||||
|
{
|
||||||
|
return new NzbDroneValidationResult(Validator.Validate(this));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -2,6 +2,8 @@ using System.Collections.Generic;
|
||||||
using System.Data;
|
using System.Data;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using Dapper;
|
using Dapper;
|
||||||
|
using NzbDrone.Core.AutoTagging;
|
||||||
|
using NzbDrone.Core.AutoTagging.Specifications;
|
||||||
using NzbDrone.Core.Datastore;
|
using NzbDrone.Core.Datastore;
|
||||||
|
|
||||||
namespace NzbDrone.Core.Housekeeping.Housekeepers
|
namespace NzbDrone.Core.Housekeeping.Housekeepers
|
||||||
|
@ -9,17 +11,24 @@ namespace NzbDrone.Core.Housekeeping.Housekeepers
|
||||||
public class CleanupUnusedTags : IHousekeepingTask
|
public class CleanupUnusedTags : IHousekeepingTask
|
||||||
{
|
{
|
||||||
private readonly IMainDatabase _database;
|
private readonly IMainDatabase _database;
|
||||||
|
private readonly IAutoTaggingRepository _autoTaggingRepository;
|
||||||
|
|
||||||
public CleanupUnusedTags(IMainDatabase database)
|
public CleanupUnusedTags(IMainDatabase database, IAutoTaggingRepository autoTaggingRepository)
|
||||||
{
|
{
|
||||||
_database = database;
|
_database = database;
|
||||||
|
_autoTaggingRepository = autoTaggingRepository;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Clean()
|
public void Clean()
|
||||||
{
|
{
|
||||||
using var mapper = _database.OpenConnection();
|
using var mapper = _database.OpenConnection();
|
||||||
var usedTags = new[] { "Series", "Notifications", "DelayProfiles", "ReleaseProfiles", "ImportLists", "Indexers", "AutoTagging", "DownloadClients" }
|
var usedTags = new[]
|
||||||
|
{
|
||||||
|
"Series", "Notifications", "DelayProfiles", "ReleaseProfiles", "ImportLists", "Indexers",
|
||||||
|
"AutoTagging", "DownloadClients"
|
||||||
|
}
|
||||||
.SelectMany(v => GetUsedTags(v, mapper))
|
.SelectMany(v => GetUsedTags(v, mapper))
|
||||||
|
.Concat(GetAutoTaggingTagSpecificationTags(mapper))
|
||||||
.Distinct()
|
.Distinct()
|
||||||
.ToArray();
|
.ToArray();
|
||||||
|
|
||||||
|
@ -37,10 +46,31 @@ namespace NzbDrone.Core.Housekeeping.Housekeepers
|
||||||
|
|
||||||
private int[] GetUsedTags(string table, IDbConnection mapper)
|
private int[] GetUsedTags(string table, IDbConnection mapper)
|
||||||
{
|
{
|
||||||
return mapper.Query<List<int>>($"SELECT DISTINCT \"Tags\" FROM \"{table}\" WHERE NOT \"Tags\" = '[]' AND NOT \"Tags\" IS NULL")
|
return mapper
|
||||||
|
.Query<List<int>>(
|
||||||
|
$"SELECT DISTINCT \"Tags\" FROM \"{table}\" WHERE NOT \"Tags\" = '[]' AND NOT \"Tags\" IS NULL")
|
||||||
.SelectMany(x => x)
|
.SelectMany(x => x)
|
||||||
.Distinct()
|
.Distinct()
|
||||||
.ToArray();
|
.ToArray();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private List<int> GetAutoTaggingTagSpecificationTags(IDbConnection mapper)
|
||||||
|
{
|
||||||
|
var tags = new List<int>();
|
||||||
|
var autoTags = _autoTaggingRepository.All();
|
||||||
|
|
||||||
|
foreach (var autoTag in autoTags)
|
||||||
|
{
|
||||||
|
foreach (var specification in autoTag.Specifications)
|
||||||
|
{
|
||||||
|
if (specification is TagSpecification tagSpec)
|
||||||
|
{
|
||||||
|
tags.Add(tagSpec.Value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return tags;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -136,6 +136,7 @@
|
||||||
"AutoTaggingSpecificationRootFolder": "Root Folder",
|
"AutoTaggingSpecificationRootFolder": "Root Folder",
|
||||||
"AutoTaggingSpecificationSeriesType": "Series Type",
|
"AutoTaggingSpecificationSeriesType": "Series Type",
|
||||||
"AutoTaggingSpecificationStatus": "Status",
|
"AutoTaggingSpecificationStatus": "Status",
|
||||||
|
"AutoTaggingSpecificationTag": "Tag",
|
||||||
"Automatic": "Automatic",
|
"Automatic": "Automatic",
|
||||||
"AutomaticAdd": "Automatic Add",
|
"AutomaticAdd": "Automatic Add",
|
||||||
"AutomaticSearch": "Automatic Search",
|
"AutomaticSearch": "Automatic Search",
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
using NzbDrone.Core.AutoTagging;
|
using NzbDrone.Core.AutoTagging;
|
||||||
|
using NzbDrone.Core.AutoTagging.Specifications;
|
||||||
using NzbDrone.Core.Datastore;
|
using NzbDrone.Core.Datastore;
|
||||||
using NzbDrone.Core.Download;
|
using NzbDrone.Core.Download;
|
||||||
using NzbDrone.Core.ImportLists;
|
using NzbDrone.Core.ImportLists;
|
||||||
|
@ -120,7 +121,7 @@ namespace NzbDrone.Core.Tags
|
||||||
var restrictions = _releaseProfileService.All();
|
var restrictions = _releaseProfileService.All();
|
||||||
var series = _seriesService.GetAllSeriesTags();
|
var series = _seriesService.GetAllSeriesTags();
|
||||||
var indexers = _indexerService.All();
|
var indexers = _indexerService.All();
|
||||||
var autotags = _autoTaggingService.All();
|
var autoTags = _autoTaggingService.All();
|
||||||
var downloadClients = _downloadClientFactory.All();
|
var downloadClients = _downloadClientFactory.All();
|
||||||
|
|
||||||
var details = new List<TagDetails>();
|
var details = new List<TagDetails>();
|
||||||
|
@ -137,7 +138,7 @@ namespace NzbDrone.Core.Tags
|
||||||
RestrictionIds = restrictions.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
RestrictionIds = restrictions.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
||||||
SeriesIds = series.Where(c => c.Value.Contains(tag.Id)).Select(c => c.Key).ToList(),
|
SeriesIds = series.Where(c => c.Value.Contains(tag.Id)).Select(c => c.Key).ToList(),
|
||||||
IndexerIds = indexers.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
IndexerIds = indexers.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
||||||
AutoTagIds = autotags.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
AutoTagIds = GetAutoTagIds(tag, autoTags),
|
||||||
DownloadClientIds = downloadClients.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
DownloadClientIds = downloadClients.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList(),
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
@ -188,5 +189,23 @@ namespace NzbDrone.Core.Tags
|
||||||
_repo.Delete(tagId);
|
_repo.Delete(tagId);
|
||||||
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
|
_eventAggregator.PublishEvent(new TagsUpdatedEvent());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private List<int> GetAutoTagIds(Tag tag, List<AutoTag> autoTags)
|
||||||
|
{
|
||||||
|
var autoTagIds = autoTags.Where(c => c.Tags.Contains(tag.Id)).Select(c => c.Id).ToList();
|
||||||
|
|
||||||
|
foreach (var autoTag in autoTags)
|
||||||
|
{
|
||||||
|
foreach (var specification in autoTag.Specifications)
|
||||||
|
{
|
||||||
|
if (specification is TagSpecification)
|
||||||
|
{
|
||||||
|
autoTagIds.Add(autoTag.Id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return autoTagIds.Distinct().ToList();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue