From a5b034d4a9c44f9bc1e83b01de82530f8fc63013 Mon Sep 17 00:00:00 2001
From: Unit 193 <unit193@ubuntu.com>
Date: Sat, 4 Apr 2015 14:41:04 -0400
Subject: Imported Upstream version 3.0.4

---
 Plugins/SqlExport/sqlexport.cpp | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'Plugins/SqlExport/sqlexport.cpp')

diff --git a/Plugins/SqlExport/sqlexport.cpp b/Plugins/SqlExport/sqlexport.cpp
index 0a379b1..15e47ed 100644
--- a/Plugins/SqlExport/sqlexport.cpp
+++ b/Plugins/SqlExport/sqlexport.cpp
@@ -145,6 +145,7 @@ bool SqlExport::exportTableRow(SqlResultsRowPtr data)
 bool SqlExport::afterExport()
 {
     writeCommit();
+    writeFkEnable();
     return true;
 }
 
@@ -237,12 +238,20 @@ void SqlExport::writeFkDisable()
     writeln("PRAGMA foreign_keys = off;");
 }
 
+void SqlExport::writeFkEnable()
+{
+    writeln("PRAGMA foreign_keys = on;");
+}
+
 QString SqlExport::formatQuery(const QString& sql)
 {
     if (cfg.SqlExport.UseFormatter.get())
         return FORMATTER->format("sql", sql, db);
 
-    return sql;
+    if (sql.trimmed().endsWith(";"))
+        return sql;
+
+    return sql.trimmed() + ";";
 }
 
 QString SqlExport::getNameForObject(const QString& database, const QString& name, bool wrapped, Dialect dialect)
-- 
cgit v1.2.3