Cleanup

[airbundle] / Entity / Session.php

diff --git a/Entity/Session.php b/Entity/Session.php
index e2ffc7afd72584d5cfdac4a6534dbd796bd7a72d..e7461711fdd30ff8ae3143ccb8524d5f9c1a7b5b 100644 (file)
--- a/Entity/Session.php
+++ b/Entity/Session.php
@@ -81,6 +81,11 @@ class Session {
         */
        private $temperaturemax;
 
+       /**
+        * @var \DateTime
+        */
+       private $locked;
+
        /**
         * @var \DateTime
         */
@@ -186,7 +191,8 @@ class Session {
                $this->start = clone $this->date;
 
                //Check if after slot
-               if ($this->slot->getTitle() == 'After') {
+               //XXX: id=4 <=> title=After
+               if ($this->slot->getId() == 4) {
                        //Add one day
                        $this->start->add(new \DateInterval('P1D'));
                }
@@ -433,6 +439,28 @@ class Session {
                return $this->temperaturemax;
        }
 
+       /**
+        * Set locked
+        *
+        * @param \DateTime $locked
+        *
+        * @return Session
+        */
+       public function setLocked($locked) {
+               $this->locked = $locked;
+
+               return $this;
+       }
+
+       /**
+        * Get locked
+        *
+        * @return \DateTime
+        */
+       public function getLocked() {
+               return $this->locked;
+       }
+
        /**
         * Set created
         *