1 files changed, 22 insertions, 3 deletions
diff --git a/htdocs/src/Post.inc.php b/htdocs/src/Post.inc.php
index 9747d3f..d887694 100644
--- a/htdocs/src/Post.inc.php
+++ b/htdocs/src/Post.inc.php
@@ -14,6 +14,7 @@ class Post {
     private $info;
     private $indatabase = false;
     private $images;
+    private $category;
 
 
     public function __construct($info=null) {
@@ -26,7 +27,8 @@ class Post {
             $this->indatabase = false;
         }
 
-        $images = null;
+        $this->images = null;
+        $this->category = null;
     }
 
     public static function getById($id) {
@@ -205,8 +207,17 @@ class Post {
         $this->info['email'] = $value;
     }
 
-    public function setCategory($value) {
-        $this->info['category_id'] = $value;
+    public function setCategory(Category $cat) {
+        $this->info['category_id'] = $cat->getId();
+        $this->category = $cat;
+    }
+
+    public function getCategory() {
+        if ($this->category == NULL) {
+            $this->category = Categories::getById($this->info['category_id']);
+        }
+
+        return $this->category;
     }
 
     public function getAge() {
@@ -237,6 +248,14 @@ class Post {
         $this->info['location'] = $value;
     }
 
+    public function getPrice() {
+        return $this->info['price'];
+    }
+
+    public function setPrice($value) {
+        $this->info['price'] = $value;
+    }
+
     public function getImages() { 
         if ($this->images == null) {
             $this->loadImages();