mirror of https://github.com/Radarr/Radarr
57 lines
2.3 KiB
C#
57 lines
2.3 KiB
C#
using System.Data;
|
|
using FluentMigrator;
|
|
using NzbDrone.Core.Datastore.Migration.Framework;
|
|
|
|
namespace NzbDrone.Core.Datastore.Migration
|
|
{
|
|
[Migration(109)]
|
|
public class add_movie_formats_to_naming_config : NzbDroneMigrationBase
|
|
{
|
|
protected override void MainDbUpgrade()
|
|
{
|
|
Alter.Table("NamingConfig").AddColumn("StandardMovieFormat").AsString().Nullable();
|
|
Alter.Table("NamingConfig").AddColumn("MovieFolderFormat").AsString().Nullable();
|
|
|
|
Execute.WithConnection(ConvertConfig);
|
|
}
|
|
|
|
private void ConvertConfig(IDbConnection conn, IDbTransaction tran)
|
|
{
|
|
using (IDbCommand namingConfigCmd = conn.CreateCommand())
|
|
{
|
|
namingConfigCmd.Transaction = tran;
|
|
namingConfigCmd.CommandText = @"SELECT * FROM NamingConfig LIMIT 1";
|
|
using (IDataReader namingConfigReader = namingConfigCmd.ExecuteReader())
|
|
{
|
|
while (namingConfigReader.Read())
|
|
{
|
|
// Output Settings
|
|
var movieTitlePattern = "";
|
|
var movieYearPattern = "({Release Year})";
|
|
var qualityFormat = "[{Quality Title}]";
|
|
|
|
movieTitlePattern = "{Movie Title}";
|
|
|
|
var standardMovieFormat = string.Format("{0} {1} {2}", movieTitlePattern, movieYearPattern, qualityFormat);
|
|
|
|
var movieFolderFormat = string.Format("{0} {1}", movieTitlePattern, movieYearPattern);
|
|
|
|
using (IDbCommand updateCmd = conn.CreateCommand())
|
|
{
|
|
var text = string.Format("UPDATE NamingConfig " +
|
|
"SET StandardMovieFormat = '{0}', " +
|
|
"MovieFolderFormat = '{1}'",
|
|
standardMovieFormat,
|
|
movieFolderFormat);
|
|
|
|
updateCmd.Transaction = tran;
|
|
updateCmd.CommandText = text;
|
|
updateCmd.ExecuteNonQuery();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|