Bogdan
2 years ago
parent
0de302ad48
commit
b2b877a8c3
@ -0,0 +1,17 @@
|
|||||||
|
export const IMDB_ID = 'imdbId';
|
||||||
|
export const TMDB_ID = 'tmdbId';
|
||||||
|
export const TVDB_ID = 'tvdbId';
|
||||||
|
export const TRAKT_ID = 'traktId';
|
||||||
|
export const R_ID = 'rId';
|
||||||
|
export const TVMAZE_ID = 'tvMazeId';
|
||||||
|
export const SEASON = 'season';
|
||||||
|
export const EPISODE = 'episode';
|
||||||
|
export const ARTIST = 'artist';
|
||||||
|
export const ALBUM = 'album';
|
||||||
|
export const LABEL = 'label';
|
||||||
|
export const TRACK = 'track';
|
||||||
|
export const YEAR = 'year';
|
||||||
|
export const GENRE = 'genre';
|
||||||
|
export const AUTHOR = 'author';
|
||||||
|
export const TITLE = 'title';
|
||||||
|
export const PUBLISHER = 'publisher';
|
||||||
@ -0,0 +1,91 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using FluentAssertions;
|
||||||
|
using NUnit.Framework;
|
||||||
|
using NzbDrone.Common.Serializer;
|
||||||
|
using NzbDrone.Core.Datastore.Migration;
|
||||||
|
using NzbDrone.Core.Test.Framework;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Test.Datastore.Migration
|
||||||
|
{
|
||||||
|
[TestFixture]
|
||||||
|
public class history_fix_data_titlesFixture : MigrationTest<history_fix_data_titles>
|
||||||
|
{
|
||||||
|
[Test]
|
||||||
|
public void should_update_data_for_book_search()
|
||||||
|
{
|
||||||
|
var db = WithMigrationTestDb(c =>
|
||||||
|
{
|
||||||
|
c.Insert.IntoTable("History").Row(new
|
||||||
|
{
|
||||||
|
IndexerId = 1,
|
||||||
|
Date = DateTime.UtcNow,
|
||||||
|
Data = new
|
||||||
|
{
|
||||||
|
Author = "Fake Author",
|
||||||
|
BookTitle = "Fake Book Title",
|
||||||
|
Publisher = "",
|
||||||
|
Year = "",
|
||||||
|
Genre = "",
|
||||||
|
Query = "",
|
||||||
|
QueryType = "book",
|
||||||
|
Source = "Prowlarr",
|
||||||
|
Host = "localhost"
|
||||||
|
}.ToJson(),
|
||||||
|
EventType = 2,
|
||||||
|
Successful = true
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
var items = db.Query<HistoryDefinition34>("SELECT * FROM \"History\"");
|
||||||
|
|
||||||
|
items.Should().HaveCount(1);
|
||||||
|
|
||||||
|
items.First().Data.Should().NotContainKey("bookTitle");
|
||||||
|
items.First().Data.Should().ContainKey("title");
|
||||||
|
items.First().Data.GetValueOrDefault("title").Should().Be("Fake Book Title");
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_update_data_for_release_grabbed()
|
||||||
|
{
|
||||||
|
var db = WithMigrationTestDb(c =>
|
||||||
|
{
|
||||||
|
c.Insert.IntoTable("History").Row(new
|
||||||
|
{
|
||||||
|
IndexerId = 1,
|
||||||
|
Date = DateTime.UtcNow,
|
||||||
|
Data = new
|
||||||
|
{
|
||||||
|
GrabMethod = "Proxy",
|
||||||
|
Title = "Fake Release Title",
|
||||||
|
Source = "Prowlarr",
|
||||||
|
Host = "localhost"
|
||||||
|
}.ToJson(),
|
||||||
|
EventType = 1,
|
||||||
|
Successful = true
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
var items = db.Query<HistoryDefinition34>("SELECT * FROM \"History\"");
|
||||||
|
|
||||||
|
items.Should().HaveCount(1);
|
||||||
|
|
||||||
|
items.First().Data.Should().NotContainKey("title");
|
||||||
|
items.First().Data.Should().ContainKey("grabTitle");
|
||||||
|
items.First().Data.GetValueOrDefault("grabTitle").Should().Be("Fake Release Title");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class HistoryDefinition34
|
||||||
|
{
|
||||||
|
public int Id { get; set; }
|
||||||
|
public int IndexerId { get; set; }
|
||||||
|
public DateTime Date { get; set; }
|
||||||
|
public Dictionary<string, string> Data { get; set; }
|
||||||
|
public int EventType { get; set; }
|
||||||
|
public string DownloadId { get; set; }
|
||||||
|
public bool Successful { get; set; }
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,66 @@
|
|||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Data;
|
||||||
|
using Dapper;
|
||||||
|
using FluentMigrator;
|
||||||
|
using Newtonsoft.Json.Linq;
|
||||||
|
using NzbDrone.Common.Serializer;
|
||||||
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Datastore.Migration
|
||||||
|
{
|
||||||
|
[Migration(034)]
|
||||||
|
public class history_fix_data_titles : NzbDroneMigrationBase
|
||||||
|
{
|
||||||
|
protected override void MainDbUpgrade()
|
||||||
|
{
|
||||||
|
Execute.WithConnection(MigrateHistoryDataTitle);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void MigrateHistoryDataTitle(IDbConnection conn, IDbTransaction tran)
|
||||||
|
{
|
||||||
|
var updatedHistory = new List<object>();
|
||||||
|
|
||||||
|
using (var selectCommand = conn.CreateCommand())
|
||||||
|
{
|
||||||
|
selectCommand.Transaction = tran;
|
||||||
|
selectCommand.CommandText = "SELECT \"Id\", \"Data\" FROM \"History\"";
|
||||||
|
|
||||||
|
using var reader = selectCommand.ExecuteReader();
|
||||||
|
|
||||||
|
while (reader.Read())
|
||||||
|
{
|
||||||
|
var id = reader.GetInt32(0);
|
||||||
|
var data = reader.GetString(1);
|
||||||
|
|
||||||
|
if (!string.IsNullOrWhiteSpace(data))
|
||||||
|
{
|
||||||
|
var jsonObject = Json.Deserialize<JObject>(data);
|
||||||
|
|
||||||
|
if (jsonObject.ContainsKey("title"))
|
||||||
|
{
|
||||||
|
jsonObject.Add("grabTitle", jsonObject.Value<string>("title"));
|
||||||
|
jsonObject.Remove("title");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (jsonObject.ContainsKey("bookTitle"))
|
||||||
|
{
|
||||||
|
jsonObject.Add("title", jsonObject.Value<string>("bookTitle"));
|
||||||
|
jsonObject.Remove("bookTitle");
|
||||||
|
}
|
||||||
|
|
||||||
|
data = jsonObject.ToJson();
|
||||||
|
|
||||||
|
updatedHistory.Add(new
|
||||||
|
{
|
||||||
|
Id = id,
|
||||||
|
Data = data
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var updateHistorySql = "UPDATE \"History\" SET \"Data\" = @Data WHERE \"Id\" = @Id";
|
||||||
|
conn.Execute(updateHistorySql, updatedHistory, transaction: tran);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue