Reftype property updated in models file

1 files changed, 17 insertions, 0 deletions

diff --git a/gstudio/models.py b/gstudio/models.py
index 80b57d1..d793db0 100644
--- a/gstudio/models.py
+++ b/gstudio/models.py
@@ -252,6 +252,23 @@ class NID(models.Model):
             return None
         
         return vrs.object
+    
+    @property
+    def reftype(self):
+        """ 
+        Returns the type the id belongs to.
+        """
+        try:
+            """ 
+            ALGO: simple wrapper for the __class__.__name__ so that it can be used in templates  
+            """
+            obj = self.ref
+            return obj.__class__.__name__
+        
+        except:
+            return None
+        
+
 
     @property
     def get_edit_url(self):