Get pk from the model if possible

marcelgwerder · Mar 13, 2017 · b129178 · b129178
1 parent 08d31f4
commit b129178
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/Parser.php b/src/Parser.php
@@ -203,12 +203,18 @@ public function parse($options, $multiple = false)
         } else {
             $identification = $options;
 
-            if (is_numeric($identification)) {
-                $this->query->where('id', $identification);
-            } else if (is_array($identification)) {
+            if (is_array($identification)) {
                 foreach ($identification as $column => $value) {
                     $this->query->where($column, $value);
                 }
+            } else {
+                if($this->isEloquentBuilder) {
+                    $primaryKey = $this->builder->getModel()->getQualifiedKeyName();
+                } else {
+                    $primaryKey = $this->getQualifiedColumnName('id');
+                }
+
+                $this->query->where($primaryKey, $identification);
             }
         }