INNER CODE UNIT · Python
input_fasta_from_filepath
BortonWrightonLabs/DRAM · bin/combine_annotations.py:34
def input_fasta_from_filepath(file_path: Path, splitter="___") -> str:
return file_path.stem.split(splitter)[0]
def bit_score_expr(column_name: str) -> pl.Expr:
return pl.col(column_name).fill_null(0)
def assign_rank_expr(columns: list[str]) -> pl.Expr:
kegg_score = (
bit_score_expr("kegg_bitScore") if "kegg_bitScore" in columns else pl.lit(0)
)
uniref_score = (
bit_score_expr("uniref_bitScore") if "uniref_bitScore" in columns else pl.lit(0)
)
motif_checks = [
bit_score_expr(f"{db}_bitScore") > 60
for db in ["pfam", "dbcan", "merops"]