omic/propka
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Partial pylint de-linting of bonds.py.

Browse Source
This commit is contained in:
Nathan Baker
2020-05-16 21:11:09 -07:00
parent 22d83ae29e
commit 303aaae25a
1 changed files with 107 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
propka/bonds.py
View File

@@ -1,49 +1,51 @@
from __future__ import division
from __future__ import print_function
import pickle,sys,os,math,propka.calculations
"""PROPKA representation of bonds."""
# TODO - is pickle still used?
import pickle
# TODO - eliminate use of sys
import sys
# TODO - eliminate use of os
import os
import math
import json
import pkg_resources
import propka.calculations
# TODO - replace the info/warning imports with logging functionality
from propka.lib import info, warning
# TODO - should these constants be defined higher up in the module?
# TODO - I don't know what some of these constants mean
DISULFIDE_DISTANCE = 2.5
FLUORIDE_DISTANCE = 1.7
HYDROGEN_DISTANCE = 1.5
DEFAULT_DISTANCE = 2.0
class bondmaker:
"""Makes bonds?
TODO - the documentation for this class needs to be improved.
"""
def __init__(self):
# predefined bonding distances
self.distances = {
'S-S':2.5,
'F-F':1.7}
self.distances = {'S-S' : DISULFIDE_DISTANCE, 'F-F' : FLUORIDE_DISTANCE}
self.distances_squared = {}
for k in self.distances.keys():
self.distances_squared[k] = self.distances[k] * self.distances[k]
self.H_dist = 1.5;
self.default_dist = 2.0;
self.H_dist = HYDROGEN_DISTANCE
self.default_dist = DEFAULT_DISTANCE
self.H_dist_squared = self.H_dist * self.H_dist
self.default_dist_squared = self.default_dist * self.default_dist
self.max_sq_distance = max(list(self.distances_squared.values())+[self.default_dist_squared])
distances = list(self.distances_squared.values()) + [self.default_dist_squared]
self.max_sq_distance = max(distances)
# protein bonding data
self.data_file_name = pkg_resources.resource_filename(__name__, 'protein_bonds.json')
with open(self.data_file_name, 'rt') as json_file:
self.protein_bonds = json.load(json_file)
self.intra_residue_backbone_bonds = {'N': ['CA'],
'CA':['N','C'],
'C': ['CA','O'],
'O': ['C']}
self.number_of_pi_electrons_in_bonds_in_backbone = {'C':1,
'O':1}
self.intra_residue_backbone_bonds = {'N': ['CA'], 'CA': ['N', 'C'],
'C': ['CA', 'O'], 'O': ['C']}
self.number_of_pi_electrons_in_bonds_in_backbone = {'C': 1, 'O': 1}
self.number_of_pi_electrons_in_conjugate_bonds_in_backbone = {'N': 1}
self.number_of_pi_electrons_in_bonds_in_sidechains = {'ARG-CZ' : 1,
'ARG-NH1': 1,
'ASN-OD1': 1,
@@ -78,46 +80,33 @@ class bondmaker:
'TYR-CZ' : 1,
'TYR-CE2': 1,
'TYR-CD2': 1}
self.number_of_pi_electrons_in_conjugate_bonds_in_sidechains = {'ARG-NE': 1,
'ARG-NH2': 1,
'ASN-ND2': 1,
'GLN-NE2': 1,
'HIS-NE2': 1,
'TRP-NE1': 1}
self.number_of_pi_electrons_in_bonds_ligands = {'C.ar':1,
'N.pl3':0,
'C.2':1,
'O.2':1,
'O.co2':1,
'N.ar':1,
'C.1':2,
'N.1':2}
self.number_of_pi_electrons_in_conjugate_bonds_in_ligands = {'N.am':1,'N.pl3':1}
self.number_of_pi_electrons_in_bonds_ligands = {'C.ar': 1, 'N.pl3': 0,
'C.2': 1, 'O.2': 1,
'O.co2': 1, 'N.ar': 1,
'C.1': 2, 'N.1': 2}
self.number_of_pi_electrons_in_conjugate_bonds_in_ligands = {'N.am': 1,
'N.pl3': 1}
self.backbone_atoms = list(self.intra_residue_backbone_bonds.keys())
self.terminal_oxygen_names = ['OXT', 'O\'\'']
return
def find_bonds_for_protein(self, protein):
""" Finds bonds proteins based on the way atoms
normally bond in proteins"""
"""Finds bonds proteins based on the way atoms normally bond in proteins.
Args:
protein: the protein to search for bonds
"""
info('++++ Side chains ++++')
# side chains
for chain in protein.chains:
for residue in chain.residues:
if residue.resName.replace(' ', '') not in ['N+', 'C-']:
self.find_bonds_for_side_chain(residue.atoms)
info('++++ Backbones ++++')
# backbone
last_residues = []
@@ -127,12 +116,10 @@ class bondmaker:
if chain.residues[i].resName.replace(' ', '') not in ['N+', 'C-']:
self.connect_backbone(chain.residues[i-1], chain.residues[i])
last_residues.append(chain.residues[i])
info('++++ terminal oxygen ++++')
# terminal OXT
for last_residue in last_residues:
self.find_bonds_for_terminal_oxygen(last_residue)
info('++++ cysteines ++++')
# Cysteines
for chain in protein.chains:
@@ -142,29 +129,36 @@ class bondmaker:
if chain.residues[j].resName == 'CYS' and j != i:
self.check_for_cysteine_bonds(chain.residues[i],
chain.residues[j])
return
def check_for_cysteine_bonds(self, cys1, cys2):
"""Looks for potential bonds between two cysteines.
Args:
cys1: one of the cysteines to check
cys1: one of the cysteines to check
"""
for atom1 in cys1.atoms:
if atom1.name == 'SG':
for atom2 in cys2.atoms:
if atom2.name == 'SG':
if propka.calculations.squared_distance(atom1,atom2) < self.SS_dist_squared:
dist = propka.calculations.squared_distance(atom1, atom2)
# TODO - is SS_dist_squared an attribute of this class?
if dist < self.SS_dist_squared:
self.make_bond(atom1, atom2)
return
def find_bonds_for_terminal_oxygen(self, residue):
"""Look for bonds for terminal oxygen.
Args:
residue - test residue
"""
for atom1 in residue.atoms:
if atom1.name in self.terminal_oxygen_names:
for atom2 in residue.atoms:
if atom2.name == 'C':
self.make_bond(atom1, atom2)
return
# TODO - stopped here.
def connect_backbone(self, residue1, residue2):
""" Sets up bonds in the backbone """
# residue 1