added option to provide directory with pdb files as input

barrOs_library.py

@@ -20,7 +20,7 @@ from random import *
 from scipy import stats
 
 ## DEFINE DEFAULT PARAMETERS
-accepted_input_types = ['pdb_file', 'pdbID', 'hhpred', 'pdbsum_fasta']
+accepted_input_types = ['pdb_file', 'pdbID', 'pdb_folder', 'hhpred', 'pdbsum_fasta']
 accepted_modes = ['membrane', 'non-membrane', 'all']
 
 ## 0. FOR LOGGING
@@ -333,6 +333,9 @@ def get_all_pdbIDs(input_files, input_types):
         if input_types[i] == 'pdb_file':
             pdb_file = input_string
             pdbIDs.append(pdb_file)
+        elif input_types[i] == 'pdb_folder':
+            pdb_files = ['{}/{}'.format(input_string, f) for f in os.listdir(input_string) if f.endswith('.pdb')]
+            pdbIDs += pdb_files
         elif input_types[i] == 'pdbID':
             pdbIDs.append(input_string)
         elif input_types[i] == 'hhpred':