From 4a7b1338f0c12fbaace66f14b397b238793daa60 Mon Sep 17 00:00:00 2001
From: Kononnable <kononnable@gmail.com>
Date: Thu, 19 Mar 2020 17:37:00 +0100
Subject: [PATCH] Allow models which don't generate properly to be saved

---
 src/ModelGeneration.ts | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/ModelGeneration.ts b/src/ModelGeneration.ts
index b94d06b..97420f7 100644
--- a/src/ModelGeneration.ts
+++ b/src/ModelGeneration.ts
@@ -80,18 +80,22 @@ function generateModels(
                   )
                 : rendered
         );
+        let formatted = "";
         try {
-            const formatted = Prettier.format(withImportStatements, {
-                parser: 'typescript',
-            });
-            fs.writeFileSync(resultFilePath, formatted, {
-                encoding: 'UTF-8',
-                flag: 'w',
+            formatted = Prettier.format(withImportStatements, {
+                parser: "typescript"
             });
         } catch (error) {
-            console.log('Trouble when generate for table: ', element.sqlName);
+            console.log(
+                "There were some problems with model generation for table: ",
+                element.sqlName
+            );
             console.log(error);
         }
+        fs.writeFileSync(resultFilePath, formatted, {
+            encoding: "UTF-8",
+            flag: "w"
+        });
     });
 }