mol2分子库文件拆分成单个mol2文件

    利用Python将.mol2文件格式的分子库拆分成单个.mol2格式的文件，以便从分子库中众多的化合物进行拆分，有利于逐个分析所提取的化合物。拆分主要的思想如下：

    1.用户选择分子库文件所在的目录以及选择将拆分文件所存储的目录

    2.读取分子库文件所有内容

    3.依照一定的拆分规律，将分子库文件逐个拆分并写入新的文件

    主要依照以上思路即可实现将.mol2分子库文件进行拆分，废话不多说，直接看干货：

import os
from tkinter import filedialog
# 拆分文件
def divide_file(read_fold, write_fold):
    all_files = []
    for files in os.walk(read_fold):
        all_files = files[2]
    for file in all_files:
        file_path = read_fold + '/' + file
        with open(file_path, 'r') as mol2_file:
            contents = mol2_file.readlines()
            contents.append('@<tripos>MOLECULE')
            write_file(contents, write_fold, file)

# 写入文件
def write_file(article, write_fold, file_name):
    file_name = file_name[:-5]
    file_path = write_fold + '/' + file_name
    if not os.path.exists(file_path):
        os.makedirs(file_path)
        print(file_path, "创建成功")
    else:
        print(file_path, "目录已存在")
    temp_article = []
    molecule_count = 0
    for line in article:
        if not line.startswith('@<tripos>MOLECULE'):
            temp_article.append(line)
        elif line.startswith('@<tripos>MOLECULE'):
            molecule_count = molecule_count + 1
            if molecule_count == 1:
                temp_article.append(line)
            elif molecule_count == 2:
                path = file_path + '/' + temp_article[1].rstrip("
")+'.mol2'
                with open(path, 'w') as new_file:
                    for new_line in temp_article:
                        new_file.write(new_line)
                temp_article.clear()
                temp_article.append('@<tripos>MOLECULE')
                molecule_count = 1

if __name__ == '__main__':
    # 读取获得的文件夹路径
    read_fold = filedialog.askdirectory()
    # 读取写入文件夹的路径
    write_fold = filedialog.askdirectory()
    divide_file(read_fold, write_fold)