Merge pull request #883 from martinszemitis/2.0

Add configuration option to enable or disable foreign keys check for all 4 types of database connections

Author: josephmancuso

src/masoniteorm/connections/BaseConnection.py

@@ -76,3 +76,13 @@ def select_many(self, query, bindings, amount):
             yield result
 
             result = self.format_cursor_results(self._cursor.fetchmany(amount))
+
+    def enable_disable_foreign_keys(self):
+        foreign_keys = self.full_details.get("foreign_keys")
+        platform = self.get_default_platform()()
+
+        if foreign_keys:
+            self._connection.execute(platform.enable_foreign_key_constraints())
+        elif foreign_keys is not None:
+            self._connection.execute(platform.disable_foreign_key_constraints())
+

src/masoniteorm/connections/MSSQLConnection.py

@@ -70,6 +70,8 @@ def make_connection(self):
             autocommit=True,
         )
 
+        self.enable_disable_foreign_keys()
+
         self.open = 1
         return self
 

src/masoniteorm/connections/MySQLConnection.py

@@ -78,6 +78,9 @@ def make_connection(self):
             db=self.database,
             **self.options
         )
+
+        self.enable_disable_foreign_keys()
+
         self.open = 1
 
         return self

src/masoniteorm/connections/PostgresConnection.py

@@ -69,6 +69,8 @@ def make_connection(self):
 
         self._connection.autocommit = True
 
+        self.enable_disable_foreign_keys()
+
         self.open = 1
 
         return self

src/masoniteorm/connections/SQLiteConnection.py

@@ -63,6 +63,9 @@ def make_connection(self):
         self._connection.create_function("REGEXP", 2, regexp)
 
         self._connection.row_factory = sqlite3.Row
+
+        self.enable_disable_foreign_keys()
+
         self.open = 1
 
         return self

tests/integrations/config/database.py

@@ -38,7 +38,7 @@
         "log_queries": True,
         "propagate": False,
     },
-    "t": {"driver": "sqlite", "database": "orm.sqlite3", "log_queries": True},
+    "t": {"driver": "sqlite", "database": "orm.sqlite3", "log_queries": True, "foreign_keys": True},
     "devprod": {
         "driver": "mysql",
         "host": os.getenv("MYSQL_DATABASE_HOST"),