RNApolis 0.4.9__py3-none-any.whl → 0.4.11__py3-none-any.whl

{RNApolis-0.4.9.dist-info → RNApolis-0.4.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: RNApolis
-Version: 0.4.9
+Version: 0.4.11
 Summary: A Python library containing RNA-related bioinformatics functions and classes
 Home-page: https://github.com/tzok/rnapolis-py
 Author: Tomasz Zok

RNApolis-0.4.11.dist-info/RECORD

@@ -0,0 +1,17 @@
+rnapolis/annotator.py,sha256=_hsSX2VHFvIQ47l_EA7lwGFXLiVLbhFPEsOQzBKbjRk,22100
+rnapolis/clashfinder.py,sha256=i95kp0o6OWNqmJDBr-PbsZd7RY2iJtBDr7QqolJSuAQ,8513
+rnapolis/common.py,sha256=LY6Uz96Br8ki_gA8LpfatgtvVbt9jOTkwgagayqTgf8,31251
+rnapolis/metareader.py,sha256=I1-cXc2YNBPwa3zihAnMTjEsAo79tEKzSmWu5yvN1Pk,2071
+rnapolis/molecule_filter.py,sha256=hB6-nXgjmw7FAsQ3bj0cZ2FvuW2I1PXunEfcdwEUB1o,7389
+rnapolis/motif_extractor.py,sha256=Lfn1iEkhkP9eZD3GPEWNAfy00QO7QPCc8wM_XS1ory8,1147
+rnapolis/parser.py,sha256=lHI6LyFbEEPdHOzbged1-Ov0tl6MpSungIPacip0Py0,15838
+rnapolis/rfam_folder.py,sha256=SjiiyML_T1__saruFwSMJEoQ7Y55GIU8ktS8ZUn5-fw,11111
+rnapolis/tertiary.py,sha256=6t9ZB4w33-5n_M3sns1RoFXCOTgVAgGH4WDNG5OG9Kg,23426
+rnapolis/transformer.py,sha256=aC0nBmHHJf5TyLvBIV57Jj3tlwpvHbPo347opfAOlQA,3844
+rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
+RNApolis-0.4.11.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
+RNApolis-0.4.11.dist-info/METADATA,sha256=k9B0MCkh46cvBm3aoXEgmz_fwvNg4RRyAfAPJ08_X18,54323
+RNApolis-0.4.11.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+RNApolis-0.4.11.dist-info/entry_points.txt,sha256=foN2Pn5e-OzEz0fFmNoX6PnFSZFQntOlY8LbognP5F0,308
+RNApolis-0.4.11.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
+RNApolis-0.4.11.dist-info/RECORD,,

rnapolis/common.py

@@ -940,6 +940,27 @@ class BpSeq:
             solutions.add(self.__make_dot_bracket(regions, orders))
         return list(solutions)
 
+    def without_pseudoknots(self):
+        return BpSeq.from_dotbracket(self.dot_bracket.without_pseudoknots())
+
+    def without_isolated(self):
+        stems, _, _, _ = self.elements
+        to_unpair = []
+
+        for stem in stems:
+            if stem.strand5p.first == stem.strand5p.last:
+                to_unpair.append(stem.strand5p.first - 1)
+                to_unpair.append(stem.strand3p.first - 1)
+
+        if not to_unpair:
+            return self
+
+        entries = self.entries.copy()
+        for i in to_unpair:
+            entries[i].pair = 0
+
+        return BpSeq(entries)
+
 
 @dataclass
 class DotBracket:
@@ -990,6 +1011,10 @@ class DotBracket:
     def __hash__(self) -> int:
         return hash((self.sequence, self.structure))
 
+    def without_pseudoknots(self):
+        structure = re.sub(r"[\[\]\{\}\<\>A-Za-z]", ".", self.structure)
+        return DotBracket(self.sequence, structure)
+
 
 @dataclass
 class MultiStrandDotBracket(DotBracket):

rnapolis/motif_extractor.py

@@ -9,6 +9,12 @@ def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--dbn", help="path to DotBracket file")
     parser.add_argument("--bpseq", help="path to BpSeq file")
+    parser.add_argument(
+        "--remove-pseudoknots", action="store_true", help="remove pseudoknots"
+    )
+    parser.add_argument(
+        "--remove-isolated", action="store_true", help="remove isolated base pairs"
+    )
     args = parser.parse_args()
 
     if args.dbn:
@@ -19,6 +25,12 @@ def main():
         parser.print_help()
         return
 
+    if args.remove_isolated:
+        bpseq = bpseq.without_isolated()
+
+    if args.remove_pseudoknots:
+        bpseq = bpseq.without_pseudoknots()
+
     print(f"Full dot-bracket:\n{bpseq.dot_bracket}")
     stems, single_strands, hairpins, loops = bpseq.elements
 

rnapolis/parser.py

@@ -65,6 +65,7 @@ def parse_cif(
         atom_site = data[0].getObj("atom_site")
         mod_residue = data[0].getObj("pdbx_struct_mod_residue")
         entity_poly = data[0].getObj("entity_poly")
+        entity = data[0].getObj("entity")
 
         if atom_site:
             for row in atom_site.getRowList():
@@ -219,6 +220,23 @@ def parse_cif(
                 if entity_id and pdbx_seq_one_letter_code_can:
                     sequence_by_entity[entity_id] = pdbx_seq_one_letter_code_can
 
+        if entity:
+            for row in entity.getRowList():
+                row_dict = dict(zip(entity.getAttributeList(), row))
+
+                entity_id = row_dict.get("id", None)
+                type_ = row_dict.get("type", None)
+
+                if entity_id:
+                    sequence_by_entity[entity_id] = sequence_by_entity.get(
+                        entity_id, ""
+                    )
+
+                    if type_:
+                        is_nucleic_acid_by_entity[entity_id] = (
+                            is_nucleic_acid_by_entity.get(entity_id, type_)
+                        )
+
     atoms = filter_clashing_atoms(atoms_to_process)
     return atoms, modified, sequence_by_entity, is_nucleic_acid_by_entity
 

rnapolis/transformer.py

@@ -1,64 +1,133 @@
 #! /usr/bin/env python
 import argparse
-import sys
+import string
+import tempfile
+from typing import Dict, Tuple
 
 from mmcif.io.IoAdapterPy import IoAdapterPy
 from mmcif.io.PdbxReader import DataCategory
 
 
-def main():
-    parser = argparse.ArgumentParser()
-    parser.add_argument("input", help="path to input mmCIF file")
-    parser.add_argument("output", help="path to output mmCIF file")
-    parser.add_argument(
-        "--category", help="name of the category to work on, e.g., atom_site"
-    )
-    parser.add_argument(
-        "--copy-from", help="name of a data item to copy from, e.g., label_asym_id"
-    )
-    parser.add_argument(
-        "--copy-to", help="name of a data item to copy to, e.g., auth_asym_id"
-    )
-    args = parser.parse_args()
-
+def copy_from_to(
+    file_content: str,
+    category: str = "atom_site",
+    copy_from: str = "label_asym_id",
+    copy_to: str = "auth_asym_id",
+) -> str:
     adapter = IoAdapterPy()
-    data = adapter.readFile(args.input)
 
-    if len(data) == 0:
-        print("Empty mmCIF file", file=sys.stderr)
-        sys.exit(1)
+    with tempfile.NamedTemporaryFile(mode="wt") as f:
+        f.write(file_content)
+        f.seek(0)
+        data = adapter.readFile(f.name)
 
-    if args.category not in data[0].getObjNameList():
-        print(f"Failed to find {args.category} in the mmCIF file", file=sys.stderr)
-        sys.exit(1)
+    if len(data) == 0 or category not in data[0].getObjNameList():
+        return file_content
 
-    category = data[0].getObj(args.category)
-    attributes = category.getAttributeList()
+    category_obj = data[0].getObj(category)
+    attributes = category_obj.getAttributeList()
 
-    if args.copy_from not in attributes:
-        print(
-            f"Failed to find data item {args.copy_from} in {args.category}",
-            file=sys.stderr,
-        )
-        sys.exit(1)
+    if copy_from not in attributes:
+        return file_content
 
     transformed = []
 
-    if args.copy_to not in attributes:
-        attributes.append(args.copy_to)
+    if copy_to not in attributes:
+        attributes.append(copy_to)
 
-    for row in category.getRowList():
-        i = attributes.index(args.copy_from)
-        j = attributes.index(args.copy_to)
+    for row in category_obj.getRowList():
+        i = attributes.index(copy_from)
+        j = attributes.index(copy_to)
         if j >= len(row):
             row.append(row[i])
         else:
             row[j] = row[i]
         transformed.append(row)
 
-    data[0].replace(DataCategory(args.category, attributes, transformed))
+    data[0].replace(DataCategory(category_obj, attributes, transformed))
+
+    with tempfile.NamedTemporaryFile(mode="rt+") as f:
+        adapter.writeFile(f.name, data)
+        f.seek(0)
+        return f.read()
+
+
+def replace_value(
+    file_content: str,
+    category: str = "atom_site",
+    column: str = "auth_asym_id",
+    values: str = "".join([c for c in string.printable if c not in string.whitespace]),
+) -> Tuple[str, Dict]:
+    adapter = IoAdapterPy()
+    with tempfile.NamedTemporaryFile(mode="wt") as f:
+        f.write(file_content)
+        f.seek(0)
+        data = adapter.readFile(f.name)
+
+    if len(data) == 0 or category not in data[0].getObjNameList():
+        return file_content, {}
+
+    category_obj = data[0].getObj(category)
+    attributes = category_obj.getAttributeList()
+
+    if column not in attributes:
+        return file_content, {}
+
+    transformed = []
+    mapping = {}
+
+    for row in category_obj.getRowList():
+        i = attributes.index(column)
+
+        if row[i] not in mapping:
+            mapping[row[i]] = values[len(mapping)]
+
+        row[i] = mapping[row[i]]
+        transformed.append(row)
+
+    data[0].replace(DataCategory(category_obj, attributes, transformed))
+
+    with tempfile.NamedTemporaryFile(mode="rt+") as f:
+        adapter.writeFile(f.name, data)
+        f.seek(0)
+        return f.read(), mapping
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("input", help="path to input mmCIF file")
+    parser.add_argument("output", help="path to output mmCIF file")
+    parser.add_argument(
+        "--category", help="name of the category to work on, e.g., atom_site"
+    )
+    parser.add_argument(
+        "--copy-from",
+        help="name of a data item to copy from, e.g., label_asym_id (exclusive with --replace)",
+    )
+    parser.add_argument(
+        "--copy-to",
+        help="name of a data item to copy to, e.g., auth_asym_id (exclusive with --replace)",
+    )
+    parser.add_argument(
+        "--replace",
+        help="name of a data item to replace values, e.g., auth_asym_id (exclusive with --copy-from and --copy-to)",
+    )
+    parser.add_argument(
+        "--values",
+        help="values to replace with, e.g., ABCDEFGHIJKLMNOPQRSTUVWXYZ (exclusive with --copy-from and --copy-to)",
+    )
+    args = parser.parse_args()
+
+    if args.copy_from and args.copy_to:
+        output = copy_from_to(args.input, args.category, args.copy_from, args.copy_to)
+    elif args.replace and args.values:
+        output = replace_value(args.input, args.category, args.replace, args.values)
+    else:
+        parser.print_help()
+        return
 
-    adapter.writeFile(args.output, data)
+    with open(args.output, "w") as f:
+        f.write(output)
 
 
 if __name__ == "__main__":

RNApolis-0.4.9.dist-info/RECORD

@@ -1,17 +0,0 @@
-rnapolis/annotator.py,sha256=_hsSX2VHFvIQ47l_EA7lwGFXLiVLbhFPEsOQzBKbjRk,22100
-rnapolis/clashfinder.py,sha256=i95kp0o6OWNqmJDBr-PbsZd7RY2iJtBDr7QqolJSuAQ,8513
-rnapolis/common.py,sha256=7KSWZzqR7ntpaS6VRDYgpP2pC1dnBttOzYB06hQzWEI,30499
-rnapolis/metareader.py,sha256=I1-cXc2YNBPwa3zihAnMTjEsAo79tEKzSmWu5yvN1Pk,2071
-rnapolis/molecule_filter.py,sha256=hB6-nXgjmw7FAsQ3bj0cZ2FvuW2I1PXunEfcdwEUB1o,7389
-rnapolis/motif_extractor.py,sha256=duHvpi9Ulcny9K60E6VBpz5RpJZw-KdTB4_Ph0iP478,774
-rnapolis/parser.py,sha256=2pQYy0sh8TCpeluMmmSJ7C5dudK_bsfstTWCdpwwpNU,15193
-rnapolis/rfam_folder.py,sha256=SjiiyML_T1__saruFwSMJEoQ7Y55GIU8ktS8ZUn5-fw,11111
-rnapolis/tertiary.py,sha256=6t9ZB4w33-5n_M3sns1RoFXCOTgVAgGH4WDNG5OG9Kg,23426
-rnapolis/transformer.py,sha256=V9nOQvdq4-p7yUWo0vQg0CDQMpmyxz9t4TMSRVEKHnw,1817
-rnapolis/util.py,sha256=IdquFO3PV1_KDqodjupzm0Rqvgy0CeSzxGHaGEHYXVU,543
-RNApolis-0.4.9.dist-info/LICENSE,sha256=ZGRu12MzCgbYA-Lt8MyBlmjvPZh7xfiD5u5wBx0enq4,1066
-RNApolis-0.4.9.dist-info/METADATA,sha256=gOlfioYZ9tvECL9I5UWTNeWkXnJ1nv3se8kys2jAzjw,54322
-RNApolis-0.4.9.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-RNApolis-0.4.9.dist-info/entry_points.txt,sha256=foN2Pn5e-OzEz0fFmNoX6PnFSZFQntOlY8LbognP5F0,308
-RNApolis-0.4.9.dist-info/top_level.txt,sha256=LcO18koxZcWoJ21KDRRRo_tyIbmXL5z61dPitZpy8yc,9
-RNApolis-0.4.9.dist-info/RECORD,,