fixing model attribute bug

zemit-cms · Feb 24, 2024 · 297e304 · 297e304
1 parent 03489dc
commit 297e304
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Mvc/Model/Attribute.php b/src/Mvc/Model/Attribute.php
@@ -11,13 +11,15 @@
 
 namespace Zemit\Mvc\Model;
 
-use Phalcon\Mvc\ModelInterface;
+use Phalcon\Mvc\EntityInterface;
+use Zemit\Mvc\Model\AbstractTrait\AbstractEntity;
 use Zemit\Mvc\Model\AbstractTrait\AbstractMetaData;
 use Zemit\Support\Helper;
 
 trait Attribute
 {
     use AbstractMetaData;
+    use AbstractEntity;
 
     /**
      * Method to get attribute from getters or the readAttribute
@@ -26,7 +28,7 @@ trait Attribute
      */
     public function getAttribute(string $attribute)
     {
-        assert($this instanceof ModelInterface);
+        assert($this instanceof EntityInterface);
         if ($this->getModelsMetaData()->hasAttribute($this, $attribute)) {
 
             $method = 'get' . ucfirst(Helper::camelize($attribute));
@@ -48,7 +50,7 @@ public function getAttribute(string $attribute)
      */
     public function setAttribute(string $attribute, $value): void
     {
-        assert($this instanceof ModelInterface);
+        assert($this instanceof EntityInterface);
         if ($this->getModelsMetaData()->hasAttribute($this, $attribute)) {
 
             $method = 'set' . ucfirst(Helper::camelize($attribute));