|
180 | 180 | "#Let's use oddt to make the pdbqt files that we need...\n", |
181 | 181 | "print(colored('Making pdbqt files', 'blue', attrs=['bold']))\n", |
182 | 182 | "for receptor_filename in receptor_filenames:\n", |
183 | | - " receptor_prefix = receptor_filename.split(\".\")[2]\n", |
184 | | - " receptor_name = receptor_prefix.split(\"/\")[2]\n", |
185 | | - " receptor_pdbqt_file = '..' + receptor_prefix + '.pdbqt'\n", |
186 | | - " print(receptor_name)\n", |
| 183 | + " receptor_prefix = receptor_filename.removesuffix('.pdb')\n", |
| 184 | + " receptor_name = receptor_prefix.split(\"/\")[-1]\n", |
| 185 | + " receptor_pdbqt_file = receptor_prefix + '.pdbqt'\n", |
187 | 186 | " rec = next(oddt.toolkit.readfile('pdb', receptor_filename))\n", |
188 | 187 | " rec.write('pdbqt', receptor_pdbqt_file, overwrite=True, opt={'r': None, 'c': None, 'h': None})\n", |
189 | 188 | "print(colored('Finished making pdbqt files', 'blue', attrs=['bold']))" |
|
204 | 203 | "source": [ |
205 | 204 | "def dock_ligands(filename):\n", |
206 | 205 | " #This line removes the filename suffix (.pdbqt)\n", |
207 | | - " dock_input = filename.split(\".\")[0]\n", |
| 206 | + " dock_input = filename.removesuffix('.pdbqt')\n", |
208 | 207 | " #This line removes the filename prefix \n", |
209 | 208 | " dock_name = dock_input.split(\"/\")[-1]\n", |
210 | | - " vina_receptor = '..' + PDBFile + '.pdbqt'\n", |
| 209 | + " vina_receptor = PDBFile + '.pdbqt'\n", |
211 | 210 | " vina_config = \"../receptor/conf.txt\"\n", |
212 | 211 | " vina_ligand = dock_input + '.pdbqt'\n", |
213 | 212 | " vina_out = dock_name + '_out.pdbqt'\n", |
|
232 | 231 | " top_pose = next(oddt.toolkit.readfile('pdbqt', dock_name + '_out.pdbqt'))\n", |
233 | 232 | " top_pose_out = top_pose.write('mol2')\n", |
234 | 233 | " mol = top_pose\n", |
235 | | - " protein = '..' + PDBFile + '.pdb'\n", |
| 234 | + " protein = PDBFile + '.pdb'\n", |
236 | 235 | " rec = next(oddt.toolkit.readfile('pdb', protein))\n", |
237 | 236 | " rec.protein=True\n", |
238 | 237 | " mols = list(oddt.toolkit.readfile('mol2', filename))\n", |
|
294 | 293 | { |
295 | 294 | "cell_type": "code", |
296 | 295 | "execution_count": null, |
297 | | - "metadata": {}, |
| 296 | + "metadata": { |
| 297 | + "scrolled": true |
| 298 | + }, |
298 | 299 | "outputs": [], |
299 | 300 | "source": [ |
300 | 301 | "#This runs a series of docking calculations on protein conformations found in the receptor folder\n", |
301 | 302 | "for receptor_filename in receptor_filenames:\n", |
302 | | - " PDBFile = receptor_filename.split(\".\")[2]\n", |
303 | | - " PDBFile_name = PDBFile.split(\"/\")[2]\n", |
| 303 | + " PDBFile = receptor_filename.removesuffix('.pdb')\n", |
| 304 | + " PDBFile_name = PDBFile.split(\"/\")[-1]\n", |
304 | 305 | " print(colored('Docking for protein started', 'blue', attrs=['bold']))\n", |
305 | 306 | " print(PDBFile_name)\n", |
306 | 307 | " \n", |
|
0 commit comments