def main(argv): parameters = Parameters(argv) hostname = parameters.getHostname() port = parameters.getPort() dbname = parameters.getDBName() language_1, language_2 = parameters.getLanguage() collection = parameters.getCollection() filexml_1 = parameters.getInputFile_1() filexml_2 = parameters.getInputFile_2() type_corpus = parameters.getType() print 'Using parameters of configuration: ' print '- Host : ',hostname print '- Port : ',port print '- Coll : ',collection print '- DBase: ',dbname print '- XML1 : ',filexml_1 print '- XML2 : ',filexml_2 database = Mongo(hostname, dbname, collection) dic_content_1 = OrderedDict() parserxml_1 = XML(filexml_1, language_1) dic_content_1 = parserxml_1.getContent() size_1 = len(dic_content_1) del parserxml_1 dic_content_2 = OrderedDict() parserxml_2 = XML(filexml_2, language_2) dic_content_2 = parserxml_2.getContent() size_2 = len(dic_content_2) del parserxml_2 counter = 1 if size_1 == size_2: #As both files come from WebAligner, they must have the same number of documents for id_order in dic_content_1: id_file_1 = dic_content_1[id_order]['id_file'] language_1 = dic_content_1[id_order]['language'] content_1 = dic_content_1[id_order]['content'] id_file_2 = dic_content_2[id_order]['id_file'] language_2 = dic_content_2[id_order]['language'] content_2 = dic_content_2[id_order]['content'] if database.exists(language_1, id_file_1): if not database.exists(language_2, id_file_2): database.insertInExisting(language_1, id_file_1, language_2, id_file_2, content_2) else: if database.exists(language_2, id_file_2): database.insertInExisting(language_2, id_file_2, language_1, id_file_1, content_1) else: database.insertNewData(language_1, id_file_1, content_1, language_2, id_file_2, content_2, type_corpus, counter) counter += 1 else: #Files have different number of documents, so they are not aligned print '\nError: Files not aligned. Please align them with WebAligner.'
def main(argv): parameters = Parameters(argv) hostname = parameters.getHostname() port = parameters.getPort() dbname = parameters.getDBName() language_1, language_2 = parameters.getLanguage() collection = parameters.getCollection() input_folder = parameters.getInputFolder() type_corpus = parameters.getType() print 'Using parameters of configuration: ' print '- Host : ',hostname print '- Port : ',port print '- Coll : ',collection print '- DBase: ',dbname print '- Input: ',input_folder database = Mongo(hostname, dbname, collection) try: root, dirs, files = os.walk(input_folder).next()[:3] except IOError: print 'ERROR: It was not possible to open the '+input_folder+' folder' sys.exit(2) name_folder = (input_folder.split('/'))[-2] dic_files = {} for corpus_file in files: print 'Working on file: '+corpus_file if not re.match('~$', corpus_file): id_file = corpus_file[0:-7] language = corpus_file[-6:-4] if not dic_files.has_key(id_file): dic_files[id_file] = {'language_1': language} else: dic_files[id_file]['language_2'] = language counter = 1 for filename in dic_files: language_1 = dic_files[filename]['language_1'] language_2 = dic_files[filename]['language_2'] id_file_1 = name_folder+'_'+filename+'_'+language_1 id_file_2 = name_folder+'_'+filename+'_'+language_2 try: file_1 = codecs.open(input_folder+''+filename+'_'+language_1+'.snt', 'r', 'utf-8') except IOError: print 'ERROR: System cannot open the '+input_folder+''+filename+'_'+language_1+'.snt file' sys.exit(2) try: file_2 = codecs.open(input_folder+''+filename+'_'+language_2+'.snt', 'r', 'utf-8') except IOError: print 'ERROR: System cannot open the '+input_folder+''+filename+'_'+language_2+'.snt file' sys.exit(2) content_1 = '' for line in file_1: #if line.strip(): content_1 += line content_2 = '' for line in file_2: #if line.strip(): content_2 += line if database.exists(language_1, id_file_1): if not database.exists(language_2, id_file_2): database.insertInExisting(language_1, id_file_1, language_2, id_file_2, content_2) else: if database.exists(language_2, id_file_2): database.insertInExisting(language_2, id_file_2, language_1, id_file_1, content_1) else: database.insertNewData(language_1, id_file_1, content_1, language_2, id_file_2, content_2, type_corpus, counter) counter += 1