32 lines
No EOL
941 B
C#
32 lines
No EOL
941 B
C#
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using RdtClient.Data.Models.Data;
|
|
|
|
namespace RdtClient.Data.Data;
|
|
|
|
#nullable disable
|
|
|
|
public class DataContext : IdentityDbContext
|
|
{
|
|
public DataContext(DbContextOptions options) : base(options)
|
|
{
|
|
}
|
|
|
|
public DbSet<Download> Downloads { get; set; }
|
|
public DbSet<Setting> Settings { get; set; }
|
|
public DbSet<Torrent> Torrents { get; set; }
|
|
|
|
protected override void OnModelCreating(ModelBuilder builder)
|
|
{
|
|
base.OnModelCreating(builder);
|
|
|
|
var cascadeFKs = builder.Model.GetEntityTypes()
|
|
.SelectMany(t => t.GetForeignKeys())
|
|
.Where(fk => !fk.IsOwnership && fk.DeleteBehavior == DeleteBehavior.Cascade);
|
|
|
|
foreach (var fk in cascadeFKs)
|
|
{
|
|
fk.DeleteBehavior = DeleteBehavior.Restrict;
|
|
}
|
|
}
|
|
} |