Adding utility function to get to models.Variable

apache · Jun 16, 2015 · 9c36068 · 9c36068
1 parent cc44e4d
commit 9c36068
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/airflow/models.py b/airflow/models.py
@@ -3,6 +3,7 @@
 import getpass
 import imp
 import jinja2
+import json
 import logging
 import os
 import dill
@@ -1784,6 +1785,15 @@ class Variable(Base):
     def __repr__(self):
         return '{} : {}'.format(self.key, self.val)
 
+    @classmethod
+    @provide_session
+    def get(cls, key, session, deserialize_json=False):
+        obj = session.query(cls).filter(cls.key == key).first()
+        v = obj.val
+        if deserialize_json and v:
+            v = json.loads(v)
+        return v
+
 
 class Pool(Base):
     __tablename__ = "slot_pool"