X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/d5bd324573e322017f2269df1f9d1665135a341b..c1fd8ae3e058c68975edd4d98107d72c9bb2dc53:/Entity/Session.php

diff --git a/Entity/Session.php b/Entity/Session.php
index e2ffc7a..e746171 100644
--- 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
 	 *