def _get_code_from_db(cls, fileloc, session=None): dag_code = session.query(cls).filter( cls.fileloc_hash == cls.dag_fileloc_hash(fileloc)).first() if not dag_code: raise DagCodeNotFound() else: code = dag_code.source_code return code
def _get_code_from_db(self, session=None): dag_code = session.query(DagCode) \ .filter(DagCode.fileloc_hash == self.fileloc_hash) \ .first() if not dag_code: raise DagCodeNotFound() else: code = dag_code.source_code return code