1: <?php
 2: /**
 3:  * Magento
 4:  *
 5:  * NOTICE OF LICENSE
 6:  *
 7:  * This source file is subject to the Open Software License (OSL 3.0)
 8:  * that is bundled with this package in the file LICENSE.txt.
 9:  * It is also available through the world-wide-web at this URL:
10:  * http://opensource.org/licenses/osl-3.0.php
11:  * If you did not receive a copy of the license and are unable to
12:  * obtain it through the world-wide-web, please send an email
13:  * to license@magentocommerce.com so we can send you a copy immediately.
14:  *
15:  * DISCLAIMER
16:  *
17:  * Do not edit or add to this file if you wish to upgrade Magento to newer
18:  * versions in the future. If you wish to customize Magento for your
19:  * needs please refer to http://www.magentocommerce.com for more information.
20:  *
21:  * @category    Mage
22:  * @package     Mage_Directory
23:  * @copyright   Copyright (c) 2012 Magento Inc. (http://www.magentocommerce.com)
24:  * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
25:  */
26: 
27: 
28: /**
29:  * Directory country format resource model
30:  *
31:  * @category    Mage
32:  * @package     Mage_Directory
33:  * @author      Magento Core Team <core@magentocommerce.com>
34:  */
35: class Mage_Directory_Model_Resource_Country_Format_Collection extends Mage_Core_Model_Resource_Db_Collection_Abstract
36: {
37:     /**
38:      * Define main table
39:      *
40:      */
41:     protected function _construct()
42:     {
43:         $this->_init('directory/country_format');
44:     }
45: 
46:     /**
47:      * Set country filter
48:      *
49:      * @param string|Mage_Directory_Model_Country $country
50:      * @return Mage_Directory_Model_Resource_Country_Format_Collection
51:      */
52:     public function setCountryFilter($country)
53:     {
54:         if ($country instanceof Mage_Directory_Model_Country) {
55:             $countryId = $country->getId();
56:         } else {
57:             $countryId = $country;
58:         }
59: 
60:         return $this->addFieldToFilter('country_id', $countryId);
61:     }
62: }
63: