From 66ead5554fed962437a4fc97a9ece02b5bacc974 Mon Sep 17 00:00:00 2001
From: marco <marco.nuechel@rwth-aachen.de>
Date: Wed, 21 Feb 2018 00:41:24 +0100
Subject: [PATCH] Add get_state_name get_state_glyph to class Protocol

---
 models/database.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/models/database.py b/models/database.py
index 49b0635..09a5971 100644
--- a/models/database.py
+++ b/models/database.py
@@ -236,6 +236,24 @@ class Protocol(DatabaseModel):
     def is_done(self):
         return self.done
 
+    def get_state_glyph(self):
+        if self.is_done():
+            state = "unchecked" #"Fertig"
+            if self.public:
+                state = "check" #"Veröffentlicht"
+        else:
+            state = "pencil" #"Geplant"
+        return state
+
+    def get_state_name(self):
+        if self.is_done():
+            state = "Fertig"
+            if self.public:
+                state = "Veröffentlicht"
+        else:
+            state = "Geplant"
+        return state
+
     def get_identifier(self):
         if self.pad_identifier is not None:
             return self.pad_identifier
-- 
GitLab