PDO Class Updates

execute, fetch and fetchAll can now accept a single string as the
second parameter if only one is needed (seems pointless sending an
array with one item)

Also, the fetchAll function accepts a third parameter to allow you to
return an array with a key being a column from the database as opposed
to the standard indexed array.

Author: a1phanumeric

class.PDO.php

@@ -49,20 +49,46 @@ function table_exists($table_name){
 	}
 
 
-	function execute($query, $values = array()){
+	function execute($query, $values = null){
+		if($values == null){
+			$values = array();
+		}else if(!is_array($values)){
+			$values = array($values);
+		}
 		$stmt = $this->pdo->prepare($query);
 		$stmt->execute($values);
 		return $stmt;
 	}
 
-	function fetch($query, $values = array()){
+	function fetch($query, $values = null){
+		if($values == null){
+			$values = array();
+		}else if(!is_array($values)){
+			$values = array($values);
+		}
 		$stmt = $this->execute($query, $values);
 		return $stmt->fetch(PDO::FETCH_ASSOC);
 	}
 
-	function fetchAll($query, $values = array()){
+	function fetchAll($query, $values = null, $key = null){
+		if($values == null){
+			$values = array();
+		}else if(!is_array($values)){
+			$values = array($values);
+		}
 		$stmt = $this->execute($query, $values);
-		return $stmt->fetchAll(PDO::FETCH_ASSOC);
+		$results = $stmt->fetchAll(PDO::FETCH_ASSOC);
+
+		// Allows the user to retrieve results using a
+		// column from the results as a key for the array
+		if($key != null && $results[0][$key]){
+			$keyed_results = array();
+			foreach($results as $result){
+				$keyed_results[$result[$key]] = $result;
+			}
+			$results = $keyed_results;
+		}
+		return $results;
 	}
 
 	function lastInsertId(){