From bcb2d8700cc95279ea479dad079af6380c332c81 Mon Sep 17 00:00:00 2001
From: Vasil Rangelov <vasil.tech@condor-gaming.com>
Date: Tue, 12 Nov 2019 11:32:55 +0200
Subject: [PATCH] Fixed imports for relations to the same table.

---
 src/ModelCustomization.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/ModelCustomization.ts b/src/ModelCustomization.ts
index 7fa0516..c2f9bfe 100644
--- a/src/ModelCustomization.ts
+++ b/src/ModelCustomization.ts
@@ -132,7 +132,10 @@ function removeColumnDefaultProperties(
 function findFileImports(dbModel: Entity[]) {
     dbModel.forEach(entity => {
         entity.relations.forEach(relation => {
-            if (!entity.fileImports.some(v => v === relation.relatedTable)) {
+            if (
+                relation.relatedTable !== entity.tscName &&
+                !entity.fileImports.some(v => v === relation.relatedTable)
+            ) {
                 entity.fileImports.push(relation.relatedTable);
             }
         });