-
Notifications
You must be signed in to change notification settings - Fork 0
/
binding_energy_write.py
47 lines (30 loc) · 1.36 KB
/
binding_energy_write.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import os
import sys
if len(sys.argv) == 2:
energy = sys.argv[1]
with open("Energy.txt","w+") as energy_file:
final_energy = energy + "."
file_name = "dock_pf1.dlg"
cwd = os.getcwd()
dir_name = cwd.split("/")
for root, dirs, files in os.walk(cwd):
if file_name in files:
file_path = os.path.join(root, file_name)
# print(file_path)
with open(file_path) as file:
data = file.readlines()
for lines in data:
if final_energy in lines[10:14]:
line_1 = lines
if "1" == line_1[3]:
list = []
list.append(lines)
#print(list)
split = file_path.split("/")
result = (split[-2] + ": " + list[0][10:16] + " kcal/mol")
if "analyse" not in result:
print(result)
energy_file.write(F'{result} \n')
else:
print(''' python binding_energy.py <approximate Binding energy you want>
exp:python binding_energy.py -12''')