You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.6 KiB
47 lines
1.6 KiB
using System.Data;
|
|
using FluentMigrator;
|
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
|
|
namespace NzbDrone.Core.Datastore.Migration
|
|
{
|
|
[Migration(166)]
|
|
public class update_series_sort_title : NzbDroneMigrationBase
|
|
{
|
|
protected override void MainDbUpgrade()
|
|
{
|
|
Execute.WithConnection(UpdateSortTitles);
|
|
}
|
|
|
|
private void UpdateSortTitles(IDbConnection conn, IDbTransaction tran)
|
|
{
|
|
using (IDbCommand getSeriesCmd = conn.CreateCommand())
|
|
{
|
|
getSeriesCmd.Transaction = tran;
|
|
getSeriesCmd.CommandText = @"SELECT Id, TvdbId, Title FROM Series";
|
|
using (IDataReader seriesReader = getSeriesCmd.ExecuteReader())
|
|
{
|
|
while (seriesReader.Read())
|
|
{
|
|
var id = seriesReader.GetInt32(0);
|
|
var tvdbId = seriesReader.GetInt32(1);
|
|
var title = seriesReader.GetString(2);
|
|
|
|
var sortTitle = SeriesTitleNormalizer.Normalize(title, tvdbId);
|
|
|
|
using (IDbCommand updateCmd = conn.CreateCommand())
|
|
{
|
|
updateCmd.Transaction = tran;
|
|
updateCmd.CommandText = "UPDATE Series SET SortTitle = ? WHERE Id = ?";
|
|
updateCmd.AddParameter(sortTitle);
|
|
updateCmd.AddParameter(id);
|
|
|
|
updateCmd.ExecuteNonQuery();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|