Skip to content

Commit

Permalink
Fix obsolete unicode functions from py2 (thanks @aipolly, #266)
Browse files Browse the repository at this point in the history
  • Loading branch information
@tanghaibao
tanghaibao committed Jun 29, 2024
1 parent d0ba53e commit 3cf106b
Show file tree
Hide file tree
Showing 4 changed files with 173 additions and 122 deletions.
154 changes: 81 additions & 73 deletions goatools/gosubdag/plot/go_name_shorten.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,35 +4,45 @@
__author__ = "DV Klopfenstein"


class ShortenText(object):
class ShortenText:
"""Shorten text for concise display."""

greek2hex = {'alpha':0x03b1, 'beta':0x03b2, 'gamma':0x03b3, 'delta':0x03b4}
greek2tex = {'alpha':r'$\alpha$', 'beta':r'$\beta$', 'gamma':r'$\gamma$', 'delta':r'$\delta$'}
greek2uni = {
"alpha": "α",
"beta": "β",
"gamma": "γ",
"delta": "δ",
}
greek2tex = {
"alpha": r"$\alpha$",
"beta": r"$\beta$",
"gamma": r"$\gamma$",
"delta": r"$\delta$",
}
keep = [
"defense response to protozoan",
"defense response to bacterium",
"cellular response to interferon-beta",
"defense response to virus",
"response to interferon-gamma",
"innate immune response",
"inflammatory response",
"response to virus",
"immune response",
]

def __init__(self):
self.greek2uni = {g:unichr(h).encode('utf-8') for g, h in self.greek2hex.items()}
self.keep = [
"defense response to protozoan",
"defense response to bacterium",
"cellular response to interferon-beta",
"defense response to virus",
"response to interferon-gamma",
"innate immune response",
"inflammatory response",
"response to virus",
"immune response"]

def get_short_plot_name(self, goobj):
@classmethod
def get_short_plot_name(cls, goobj):
"""Shorten some GO names so plots are smaller."""
name = goobj.name
if self._keep_this(name):
return self.replace_greek(name)
name = name.replace("cellular response to chemical stimulus",
"cellular rsp. to chemical stim.")
if cls._keep_this(name):
return cls.replace_greek(name)
name = name.replace(
"cellular response to chemical stimulus", "cellular rsp. to chemical stim."
)
depth = goobj.depth
if depth > 1:
name = name.replace("regulation of ", "reg. of ")
name = name.replace("regulation of", "reg. of")
name = name.replace("positive reg", "+reg")
name = name.replace("negative reg", "-reg")
name = name.replace("involved in", "in")
Expand All @@ -41,12 +51,16 @@ def get_short_plot_name(self, goobj):
name = name.replace("MHC class I", "MHC-I")
if depth == 4:
if goobj.id == "GO:0002460":
before = " ".join([
"adaptive immune response based on somatic recombination of",
"immune receptors built from immunoglobulin superfamily domains"])
before = " ".join(
[
"adaptive immune response based on somatic recombination of",
"immune receptors built from immunoglobulin superfamily domains",
]
)
name = name.replace(
before,
"rsp. based on somatic recombination of Ig immune receptors")
"rsp. based on somatic recombination of Ig immune receptors",
)
if depth > 3:
name = name.replace("signaling pathway", "sig. pw.")
name = name.replace("response", "rsp.")
Expand All @@ -56,12 +70,13 @@ def get_short_plot_name(self, goobj):
name = name.replace("production", "prod.")
if depth == 6 or depth == 5:
name = name.replace("tumor necrosis factor", "TNF")
name = self.replace_greek(name)
name = cls.replace_greek(name)
return name

def shorten_go_name_ptbl1(self, name):
@classmethod
def shorten_go_name_ptbl1(cls, name):
"""Shorten GO name for tables in paper."""
if self._keep_this(name):
if cls._keep_this(name):
return name
name = name.replace("negative", "neg.")
name = name.replace("positive", "pos.")
Expand All @@ -70,80 +85,73 @@ def shorten_go_name_ptbl1(self, name):
name = name.replace("antigen processing and presentation", "app.")
return name

def shorten_go_name_ptbl3(self, name, dcnt):
@classmethod
def shorten_go_name_ptbl3(cls, name, dcnt):
"""Shorten GO description for Table 3 in manuscript."""
if self._keep_this(name):
if cls._keep_this(name):
return name
name = name.replace("positive regulation of immune system process",
"+ reg. of immune sys. process")
name = name.replace("positive regulation of immune response",
"+ reg. of immune response")
name = name.replace("positive regulation of cytokine production",
"+ reg. of cytokine production")
name = name.replace(
"positive regulation of immune system process",
"+ reg. of immune sys. process",
)
name = name.replace(
"positive regulation of immune response", "+ reg. of immune response"
)
name = name.replace(
"positive regulation of cytokine production",
"+ reg. of cytokine production",
)
if dcnt < 40:
name = name.replace("antigen processing and presentation", "a.p.p.")
if dcnt < 10:
name = name.replace("negative", "-")
name = name.replace("positive", "+")
#name = name.replace("tumor necrosis factor production", "tumor necrosis factor prod.")
name = name.replace("tumor necrosis factor production", "TNF production")
if dcnt < 4:
name = name.replace("regulation", "reg.")
name = name.replace("exogenous ", "")
name = name.replace(" via ", " w/")
name = name.replace("T cell mediated cytotoxicity", "cytotoxicity via T cell")
name = name.replace('involved in', 'in')
name = name.replace('-positive', '+')
name = name.replace(
"T cell mediated cytotoxicity", "cytotoxicity via T cell"
)
name = name.replace("involved in", "in")
name = name.replace("-positive", "+")
return name

def replace_greek(self, name):
@classmethod
def replace_greek(cls, name):
"""Replace text representing greek letters with greek letters."""
name = name.replace('gamma-delta', 'gammadelta')
name = name.replace('interleukin-1 beta', 'interleukin-1beta')
greek_present = False
for greek_txt, uni in self.greek2uni.items():
name = name.replace("gamma-delta", "gammadelta")
name = name.replace("interleukin-1 beta", "interleukin-1beta")
for greek_txt, uni in cls.greek2uni.items():
if greek_txt in name:
greek_present = True
name = name.replace(greek_txt, "{B}".format(B=uni))
if greek_present is True:
name = unicode(name, 'utf-8') # For writing to xlsx
name = name.replace(greek_txt, uni)
return name

def replace_greek_tex(self, name):
@classmethod
def replace_greek_tex(cls, name):
"""Replace text representing greek letters with greek letters."""
name = name.replace('gamma-delta', 'gammadelta')
name = name.replace('interleukin-1 beta', 'interleukin-1beta')
# greek_present = False
for greek_txt, tex in self.greek2tex.items():
name = name.replace("gamma-delta", "gammadelta")
name = name.replace("interleukin-1 beta", "interleukin-1beta")
for greek_txt, tex in cls.greek2tex.items():
if greek_txt in name:
# greek_present = True
name = name.replace(greek_txt, "{B}".format(B=tex))
# if greek_present is True:
# name = texcode(name, 'utf-8') # For writing to xlsx
name = name.replace(greek_txt, tex)
return name

def shorten_go_name_all(self, name):
@classmethod
def shorten_go_name_all(cls, name):
"""Shorten GO name for tables in paper, supplemental materials, and plots."""
name = self.replace_greek(name)
name = cls.replace_greek(name)
name = name.replace("MHC class I", "MHC-I")
return name

def _keep_this(self, name):
@classmethod
def _keep_this(cls, name):
"""Return True if there are to be no modifications to name."""
for keep_name in self.keep:
for keep_name in cls.keep:
if name == keep_name:
return True
return False

#@staticmethod
#def _shorten_go_name(name):
# name = name.replace(' and ', ' & ')
# if gont.dcnt < 15:
# name = name.replace('MHC class I', 'MHC-I')
# name = name.replace('antigen processing & presentation', 'a. p. p.')
# name = name.replace('positive regulation', 'pos. reg.')
# elif gont.depth > 2:
# name = name.replace(' & presentation', ' & pres.')
# return name

# Copyright (C) 2016-2017, DV Klopfenstein, H Tang, All rights reserved.
Loading

0 comments on commit 3cf106b

Please sign in to comment.