[-] WS : Fix bug from 1.4 Groups for entity customers API

2012-02-21 17:13:56 +00:00
parent 53397b1512
commit 27833fe624
1 changed files with 21 additions and 0 deletions
@@ -143,6 +143,9 @@ class CustomerCore extends ObjectModel
 			'deleted' => array(),
 			'passwd' => array('setter' => 'setWsPasswd'),
 		),
+		'associations' => array(
+				'groups' => array('resource' => 'group'),
+		)
 	);

 	/**
@@ -747,4 +750,22 @@ class CustomerCore extends ObjectModel

 		return $total_paid - $total_rest;
 	}
+
+	public function getWsGroups()
+	{
+		return Db::getInstance()->ExecuteS('
+				SELECT cg.`id_group` as id
+				FROM '._DB_PREFIX_.'customer_group cg
+				WHERE cg.`id_customer` = '.(int)$this->id);
+	}
+	
+	public function setWsGroups($result)
+	{
+		$groups = array();
+		foreach ($result as $row)
+			$groups[] = $row['id'];
+		$this->cleanGroups();
+		$this->addGroups($groups);
+		return true;
+	}
 }