API
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.

DataContext.cs 11KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227
  1. using Microsoft.EntityFrameworkCore;
  2. using Npgsql;
  3. using System.Linq;
  4. using UnivateProperties_API.Model;
  5. using UnivateProperties_API.Model.Banks;
  6. using UnivateProperties_API.Model.Campaigns;
  7. using UnivateProperties_API.Model.Communication;
  8. using UnivateProperties_API.Model.Financial;
  9. using UnivateProperties_API.Model.Logging;
  10. using UnivateProperties_API.Model.Misc;
  11. using UnivateProperties_API.Model.ProcessFlow;
  12. using UnivateProperties_API.Model.Properties;
  13. using UnivateProperties_API.Model.Region;
  14. using UnivateProperties_API.Model.Timeshare;
  15. using UnivateProperties_API.Model.Users;
  16. namespace UnivateProperties_API.Context
  17. {
  18. public class DataContext : DbContext
  19. {
  20. private string connectionString = "";
  21. public DataContext(DbContextOptions<DataContext> options) : base(options)
  22. {
  23. foreach (var extention in options.Extensions)
  24. {
  25. if (extention.GetType().ToString() == "Npgsql.EntityFrameworkCore.PostgreSQL.Infrastructure.Internal.NpgsqlOptionsExtension")
  26. {
  27. connectionString = ((Npgsql.EntityFrameworkCore.PostgreSQL.Infrastructure.Internal.NpgsqlOptionsExtension)extention).ConnectionString;
  28. }
  29. }
  30. }
  31. #region User
  32. public virtual DbSet<Agency> Agencies { get; set; }
  33. public virtual DbSet<Agent> Agents { get; set; }
  34. public virtual DbSet<User> Users { get; set; }
  35. public virtual DbSet<Individual> Individuals { get; set; }
  36. public virtual DbSet<Address> Addresses { get; set; }
  37. #endregion User
  38. #region Communication
  39. public virtual DbSet<Email> Emails { get; set; }
  40. public virtual DbSet<SMTPAccount> Accounts { get; set; }
  41. public virtual DbSet<SMTPHost> Hosts { get; set; }
  42. public virtual DbSet<Template> Templates { get; set; }
  43. public virtual DbSet<PlaceHolder> PlaceHolders { get; set; }
  44. #endregion Communication
  45. #region Property
  46. public DbSet<Property> Properties { get; set; }
  47. public DbSet<PropertyImage> PropertyImages { get; set; }
  48. public DbSet<PropertyType> PropertyTypes { get; set; }
  49. public DbSet<PropertyUserField> PropertyUserFields { get; set; }
  50. public DbSet<UserDefinedField> UserDefinedFields { get; set; }
  51. public DbSet<UserDefinedGroup> UserDefinedGroups { get; set; }
  52. #endregion
  53. #region Region
  54. public DbSet<Province> Provinces { get; set; }
  55. public DbSet<City> Cities { get; set; }
  56. public DbSet<Suburb> Suburbs { get; set; }
  57. #endregion
  58. #region Timeshare
  59. public DbSet<TimeshareWeek> Weeks { get; set; }
  60. public DbSet<Status> Status { get; set; }
  61. public DbSet<UnitConfiguration> UnitConfigurations { get; set; }
  62. public DbSet<UnitConfigurationType> UnitConfigurationTypes { get; set; }
  63. public DbSet<Season> Seasons { get; set; }
  64. public DbSet<Bank> Banks { get; set; }
  65. public DbSet<BankAccount> BankAccounts { get; set; }
  66. #endregion Timeshare
  67. #region ProcessFlow
  68. public DbSet<ProcessFlow> ProcessFlows { get; set; }
  69. public DbSet<BidItem> BidItems { get; set; }
  70. #endregion
  71. #region Logs
  72. public DbSet<SearchLog> SearchLogs { get; set; }
  73. #endregion
  74. #region Misc
  75. public DbSet<Location> Location { get; set; }
  76. public DbSet<Carousel> Carousel { get; set; }
  77. public DbSet<PlaceHolderFormat> PlaceHolderFormats { get; set; }
  78. public DbSet<Default> Defaults { get; set; }
  79. #endregion
  80. #region Payments
  81. public DbSet<Payment> Payments { get; set; }
  82. public DbSet<ListingFee> ListingFees { get; set; }
  83. #endregion
  84. #region Campaign
  85. public DbSet<Campaign> Campaigns { get; set; }
  86. public DbSet<CampaignItem> CampaignItems { get; set; }
  87. public DbSet<CampaignPlaceHolder> CampaignPlaceHolders { get; set; }
  88. public DbSet<CampaignItemPlaceHolder> CampaignItemPlaceHolders { get; set; }
  89. #endregion
  90. public override int SaveChanges()
  91. {
  92. foreach (var item in ChangeTracker
  93. .Entries()
  94. .Where(x => x.State == EntityState.Modified || x.State == EntityState.Added)
  95. .Select(x => x.Entity)
  96. .ToList())
  97. {
  98. if (item is BaseEntity)
  99. {
  100. (item as BaseEntity).UpdateModified(string.Empty);
  101. }
  102. }
  103. UpdateSoftDeleteStatuses();
  104. return base.SaveChanges();
  105. }
  106. private void UpdateSoftDeleteStatuses()
  107. {
  108. foreach (var entry in ChangeTracker.Entries())
  109. {
  110. switch (entry.State)
  111. {
  112. case EntityState.Added:
  113. entry.CurrentValues["IsDeleted"] = false;
  114. break;
  115. case EntityState.Deleted:
  116. entry.State = EntityState.Modified;
  117. entry.CurrentValues["IsDeleted"] = true;
  118. break;
  119. }
  120. }
  121. }
  122. protected override void OnModelCreating(ModelBuilder modelBuilder)
  123. {
  124. modelBuilder.Entity<Individual>()
  125. .Ignore(b => b.FullName);
  126. modelBuilder.Entity<SMTPHost>().ToTable("Hosts");
  127. modelBuilder.Entity<UnitConfiguration>()
  128. .HasIndex(u => u.Code)
  129. .IsUnique();
  130. modelBuilder.Entity<Individual>(b =>
  131. {
  132. b.HasKey(e => e.Id);
  133. b.Property(e => e.Id).ValueGeneratedOnAdd();
  134. });
  135. modelBuilder.Entity<Individual>()
  136. .HasIndex(i => new { i.Telephone, i.CellNumber, i.Email })
  137. .IsUnique(true);
  138. modelBuilder.Entity<Agent>(b =>
  139. {
  140. b.HasKey(e => e.Id);
  141. b.Property(e => e.Id).ValueGeneratedOnAdd();
  142. });
  143. modelBuilder.Entity<Agency>(b =>
  144. {
  145. b.HasKey(e => e.Id);
  146. b.Property(e => e.Id).ValueGeneratedOnAdd();
  147. });
  148. modelBuilder.Entity<Person>(b =>
  149. {
  150. b.HasKey(e => e.Id);
  151. b.Property(e => e.Id).ValueGeneratedOnAdd();
  152. });
  153. modelBuilder.Entity<User>()
  154. .HasIndex(u => u.Username)
  155. .IsUnique(true);
  156. modelBuilder.Entity<Email>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  157. modelBuilder.Entity<SMTPAccount>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  158. modelBuilder.Entity<SMTPHost>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  159. modelBuilder.Entity<Property>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  160. modelBuilder.Entity<PropertyImage>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  161. modelBuilder.Entity<PropertyType>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  162. modelBuilder.Entity<PropertyUserField>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  163. modelBuilder.Entity<UserDefinedField>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  164. modelBuilder.Entity<UserDefinedGroup>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  165. modelBuilder.Entity<City>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  166. modelBuilder.Entity<Province>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  167. modelBuilder.Entity<Suburb>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  168. modelBuilder.Entity<Season>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  169. modelBuilder.Entity<Status>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  170. modelBuilder.Entity<TimeshareWeek>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  171. modelBuilder.Entity<UnitConfiguration>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  172. modelBuilder.Entity<UnitConfigurationType>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  173. modelBuilder.Entity<Agency>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  174. modelBuilder.Entity<Person>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  175. modelBuilder.Entity<User>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  176. modelBuilder.Entity<Bank>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  177. modelBuilder.Entity<BankAccount>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  178. modelBuilder.Entity<SearchLog>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  179. modelBuilder.Entity<Address>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  180. modelBuilder.Entity<BidItem>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  181. modelBuilder.Entity<ProcessFlow>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  182. modelBuilder.Entity<Template>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  183. modelBuilder.Entity<Carousel>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  184. modelBuilder.Entity<PlaceHolder>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  185. modelBuilder.Entity<Campaign>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  186. modelBuilder.Entity<CampaignItem>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  187. modelBuilder.Entity<CampaignItemPlaceHolder>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  188. modelBuilder.Entity<CampaignPlaceHolder>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  189. modelBuilder.Entity<PlaceHolderFormat>().HasQueryFilter(m => EF.Property<bool>(m, "IsDeleted") == false);
  190. }
  191. //public int GetMaxID(string tableName)
  192. //{
  193. // NpgsqlConnection connection = new NpgsqlConnection(connectionString);
  194. // connection.Open();
  195. // NpgsqlCommand cmd = connection.CreateCommand();
  196. // cmd.CommandText = string.Format("select MAX(\"Id\") from \"{0}\"", tableName);
  197. // NpgsqlDataReader reader = cmd.ExecuteReader();
  198. // int returnValue = 0;
  199. // while(reader.Read())
  200. // {
  201. // returnValue = int.Parse(reader[0] == null ? "0" : reader[0].ToString() == "" ? "0" : reader[0].ToString());
  202. // }
  203. // connection.Close();
  204. // return returnValue;
  205. //}
  206. }
  207. }