split_list.py

# Import libraries
import argparse
import os

def save_to_file(text,output):
    with open(output, mode='wt', encoding='utf-8') as myfile:
        myfile.write('\n'.join(text))
        myfile.write('\n')
# ------------------------------------------------------------------------------------------------------
# Parse command line input and options
parser = argparse.ArgumentParser(description="	ʕっ•ᴥ•ʔっ  * Split one large MIM list into lists of 20 or less MIMs to use in table.py (for functionality, due to API limits)  * ")
parser.add_argument('-i', '--input', type=str, help='<INPUT_FILENAME.txt>   (One MIM list larger than 20 MIMs to be divided)')
parser.add_argument('-o', '--output', type=str, help='<OUTPUT_FILENAME>    (without \".txt\":  multiple numbered .txt files will be produced automatically.)')
args = parser.parse_args()

# set argument based variables
input = args.input
output = args.output

# create and print title screen logo
opening_screen = """

   _       ,/'
  (_).  ,/'
   _  ::       ---  ---  ---  ---  ---  ---  ---  ---  ---  ---  ---  ---  
  (_)'  `\.
           `\.             [  s P y d e r M I M  ]
                                        List Splitter
"""
print()
print(opening_screen)
print()
print('Splitting list into separate OMIM API compatible lists of 20 maximum MIMs...')
print()

# read file line by line
my_file = open(input, "r")
lst = my_file.read().splitlines()

# instantiate a list and a maximum size of each list
chunked_list = []
chunk_size = 20

# build a list of lists, each of length 'chunk_size':
for i in range(0, len(lst), chunk_size):
    chunked_list.append(lst[i:i+chunk_size])

# Create a dictionary of lists and save each list as a key to this dictionary
dictionary = {}
print('Saving divided lists to directory:    '+str('./{}_MIM_directory'.format(output))+ str('/'))
print()

# create a new directory (name determined by argument for output project name)
newpath = './{}_MIM_directory'.format(output)
rm_command = 'rm -R {}_MIM_directory'.format(output)

# if the directory for split lists has not yet already been created, create one based on the project name
if not os.path.exists(newpath):
    os.makedirs(newpath)

# otherwise, remove the directory that exists and start fresh by creating a new one
else:
    os.system(rm_command)
    os.makedirs(newpath)

# Create a dictionary of chunked lists, each of 'chunk_size',
for i in range(len(chunked_list)):
    dictionary['./{}_MIM_directory/'.format(output)+output+"_{0}.txt".format(i)] = chunked_list[i]
    print(str(output) + '_' + str(i) + '.txt')
print()

# save output to file
for i in dictionary:
    save_to_file(dictionary[i], i)