Initial Membership Database Work

application/classes/DB.php

@@ -0,0 +1,35 @@
+<?php defined('SYSPATH') or die('No direct access allowed.');
+
+/**
+ * This class overrides Kohana's DB
+ *
+ * @package    Membership Database
+ * @category   Modifications
+ * @author     Deon George
+ * @copyright  (c) 2014 Deon George
+ * @license    http://dev.leenooks.net/license.html
+ */
+class DB extends Kohana_DB {
+	// Add the site_id to the delete query
+	public static function delete($table = NULL)
+	{
+		$db = new Database_Query_Builder_Delete($table);
+
+		if (! in_array($table,ORM::$no_site_id_tables))
+			return $db->where($table.'.site_id','=',Company::instance()->site());
+		else
+			return $db;
+	}
+
+	// Add the site_id to the update query
+	final public static function update($table = NULL)
+	{
+		$db = new Database_Query_Builder_Update($table);
+
+		if (! in_array($table,ORM::$no_site_id_tables))
+			return $db->where($table.'.site_id','=',Company::instance()->site());
+		else
+			return $db;
+	}
+}
+?>