1 files changed, 15 insertions, 9 deletions

diff --git a/src/model/Prestations.php b/src/model/Prestations.php
index 54ad4b7..8591e83 100644
--- a/src/model/Prestations.php
+++ b/src/model/Prestations.php
@@ -28,13 +28,19 @@ class Prestations extends Model
     {
         return $this->ID_client;
     }
-    public function getIDsByIdClient() // obtenir une entrée avec son ID_client
+    public function getIDsByIdClient() // obtenir une entrée avec son ID_client, comportement différent si le type est connu
     {
-        $IDs = $this->execQuery('SELECT id FROM ' . $this->table . ' WHERE id_client = ' . $this->ID_client)->fetchAll();
-        // changer le tableau de tableaux en tableau simple
-        for($i = 0; $i < count($IDs); $i++)
+        $sql = 'SELECT id FROM ' . $this->table . ' WHERE id_client = ' . $this->ID_client;
+        if($this->type_presta != '')
         {
-            $IDs[$i] = $IDs[$i]['ID'];
+            $sql .= " AND type_presta = '" . $this->type_presta . "'";
+        }
+        $data = $this->execQuery($sql)->fetchAll(); // tableau de tableaux
+        
+        $IDs = []; // si $IDs reste vide, ne pas être de type NULL
+        for($i = 0; $i < count($data); $i++)
+        {
+            $IDs[$i] = $data[$i]['ID']; // tableau simple
         }
         return($IDs);
     }
@@ -152,7 +158,7 @@ class Prestations extends Model
     }
 }
 
-class CodePresta extends Prestations
-{
-    protected $numero_presta;
-}
+//~ class CodePresta extends Prestations
+//~ {
+    //~ protected $numero_presta;
+//~ }