Skip to content

Commit

Permalink
Explictly search human pathways in custom method
Browse files Browse the repository at this point in the history
  • Loading branch information
ddomingof committed Aug 28, 2018
1 parent 8b16915 commit cb9ac96
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions src/bio2bel_reactome/manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
import bio2bel_chebi
import bio2bel_hgnc
from compath_utils import CompathManager
from sqlalchemy import and_
from tqdm import tqdm

from .constants import MODULE_NAME
Expand Down Expand Up @@ -87,6 +88,28 @@ def count_species(self):

"""Custom query methods"""

def query_similar_pathways(self, pathway_name, top=None):
"""Filter pathways by name.
:param str pathway_name: pathway name to query
:param int top: return only X entries
:return: Optional[models.Pathway]
"""
similar_pathways = self.session.query(self.pathway_model).filter(and_(
self.pathway_model.name.contains(pathway_name),
Species.name == 'Homo sapiens')
).all()

similar_pathways = [
(pathway.resource_id, pathway.name)
for pathway in similar_pathways
]

if top:
return similar_pathways[:top]

return similar_pathways

def query_gene_set(self, gene_set):
"""Return pathway counter dictionary.
Expand Down

0 comments on commit cb9ac96

Please sign in to comment.