This repository has been archived on 2024-04-08. You can view files and clone it, but cannot push or open issues or pull requests.
khosb/application/classes/ORM.php

106 lines
3.1 KiB
PHP
Raw Normal View History

2011-05-14 07:35:33 +00:00
<?php defined('SYSPATH') or die('No direct access allowed.');
/**
* This class overrides Kohana's ORM
*
* @package OSB/Modifications
* @category Classes
* @category Helpers
* @author Deon George
* @copyright (c) 2010 Deon George
* @license http://dev.leenooks.net/license.html
*/
abstract class ORM extends Kohana_ORM {
protected $_table_names_plural = FALSE;
protected $_model_names_plural = FALSE;
2011-05-14 07:35:33 +00:00
private $_object_formated = array();
private $_formated = FALSE;
// Our filters used to display values in a friendly format
protected $_display_filters = array();
// Add our OSB site_id to each SELECT query
final protected function _build($type) {
2011-09-29 07:13:32 +00:00
// Exclude tables without site ID's
2011-12-30 07:10:02 +00:00
if (! in_array($this->_table_name,Config::$no_site_id_tables))
2012-11-09 23:13:57 +00:00
$this->where($this->_object_name.'.site_id','=',Config::siteid());
2011-05-14 07:35:33 +00:00
return parent::_build($type);
}
/**
* Format fields for display purposes
*
* @param string column name
* @return mixed
*/
private function _format() {
foreach ($this->_display_filters as $column => $formats)
$this->_object_formated[$column] = $this->run_filter($column,$this->__get($column),array($column=>$formats));
$this->_formated = TRUE;
}
/**
* Return a formated columns, as per the model definition
*/
public function display($column) {
// Trigger a load of the record.
$value = $this->__get($column);
// If some of our fields need to be formated for display purposes.
2012-01-11 08:59:20 +00:00
if (! $this->_formated AND $this->_display_filters)
2011-05-14 07:35:33 +00:00
$this->_format();
if (isset($this->_object_formated[$column]))
return $this->_object_formated[$column];
else
2011-07-13 22:59:32 +00:00
return HTML::nbsp($value);
2011-05-14 07:35:33 +00:00
}
2011-09-17 10:45:08 +00:00
/**
2012-11-09 23:13:57 +00:00
* Override KH's ORM count_relations() function, to include our site_id in the query.
2011-09-17 10:45:08 +00:00
*
2012-11-09 23:13:57 +00:00
* This is a copy of KH's ORM count_relations() function, with the addition of a where
2011-09-17 10:45:08 +00:00
* clause to include the site id.
*/
2012-11-09 23:13:57 +00:00
public function count_relations($alias, $far_keys = NULL)
{
if ($far_keys === NULL)
{
return (int) DB::select(array(DB::expr('COUNT(*)'), 'records_found'))
->from($this->_has_many[$alias]['through'])
->where($this->_has_many[$alias]['foreign_key'], '=', $this->pk())
->where('site_id', '=', Config::siteid())
->execute($this->_db)->get('records_found');
}
2011-09-17 10:45:08 +00:00
$far_keys = ($far_keys instanceof ORM) ? $far_keys->pk() : $far_keys;
// We need an array to simplify the logic
$far_keys = (array) $far_keys;
// Nothing to check if the model isn't loaded or we don't have any far_keys
if ( ! $far_keys OR ! $this->_loaded)
2012-11-09 23:13:57 +00:00
return 0;
2011-09-17 10:45:08 +00:00
2012-11-09 23:13:57 +00:00
$count = (int) DB::select(array(DB::expr('COUNT(*)'), 'records_found'))
2011-09-17 10:45:08 +00:00
->from($this->_has_many[$alias]['through'])
->where($this->_has_many[$alias]['foreign_key'], '=', $this->pk())
->where($this->_has_many[$alias]['far_key'], 'IN', $far_keys)
->where('site_id', '=', Config::siteid())
->execute($this->_db)->get('records_found');
// Rows found need to match the rows searched
2012-11-09 23:13:57 +00:00
return (int) $count;
2011-09-17 10:45:08 +00:00
}
protected function _where_active() {
return $this->where('status','=',TRUE);
}
public function where_active() {
return $this->_where_active();
}
2011-05-14 07:35:33 +00:00
}
?>