kkonganti@0: """Builds SQLite3 databases"""
kkonganti@0: 
kkonganti@0: import logging, os, pickle, re, requests, sqlite3, sys, time
kkonganti@0: import lexmapr.ontology_reasoner as ontr
kkonganti@0: from nltk.tokenize import word_tokenize
kkonganti@0: from lexmapr.pipeline_helpers import punctuation_treatment
kkonganti@0: from lexmapr.definitions import embl_ontologies, synonym_db, ontol_db
kkonganti@0: from lexmapr.definitions import owl_dir, purl_link, missing_ontol_labels
kkonganti@0: from lexmapr.pipeline_resources import get_resource_label_permutations
kkonganti@0: 
kkonganti@0: logging.getLogger('requests').setLevel(logging.WARNING)
kkonganti@0: logging.getLogger('urllib3').setLevel(logging.WARNING)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: # TODO: might replace pickle with ujson
kkonganti@0: def _pickle_save(data_to_save, file_path):
kkonganti@0:     '''Write a pickle file'''
kkonganti@0:     with open(file_path,'wb') as SAVE_file:
kkonganti@0:         pickle.dump(data_to_save, SAVE_file)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _pickle_load(file_path):
kkonganti@0:     '''Read a pickle file'''
kkonganti@0:     with open(file_path,'rb') as LOAD_file:
kkonganti@0:         return(pickle.load(LOAD_file))
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _get_ontols(ontol_interest):
kkonganti@0:     '''Obtain URLs for ontologies of interest'''
kkonganti@0:     ontol_dic = {}
kkonganti@0:     embl_resp = requests.get(embl_ontologies)
kkonganti@0:     resp_blocks = re.findall('<tr>([\s\S]+?)</tr>',embl_resp.content.decode('utf-8'))
kkonganti@0:     for resp_block in resp_blocks:
kkonganti@0:         try:
kkonganti@0:             embl_abbr = re.search('class=\"ontology-source\">([\s\S]+?)<', resp_block).group(1)
kkonganti@0:             embl_name = re.search('<a href=\"/ols/ontologies/' + embl_abbr.lower() +\
kkonganti@0:                                   '\">([\s\S]+?)</a>', resp_block).group(1)
kkonganti@0:             embl_link = re.search('href=\"(\S+)\">Download', resp_block).group(1)
kkonganti@0:             if embl_link.startswith('ontologies'):
kkonganti@0:                 embl_link = embl_link[len('ontologies'):]
kkonganti@0:         # TODO: with Python 3.9- embl_link.removeprefix('ontologies')
kkonganti@0:         except(AttributeError):
kkonganti@0:             continue
kkonganti@0:         if embl_abbr in ontol_interest:
kkonganti@0:             ontol_dic[embl_abbr] = (embl_name, embl_link)
kkonganti@0:     # Continue if not find all ontologies of interest specified in definitions.py
kkonganti@0:     not_found = set(ontol_interest).difference(set(ontol_dic.keys()))
kkonganti@0:     if not_found:
kkonganti@0:         if len(not_found) == 1:
kkonganti@0:             logging.warning(f'Did not find ontology: ' + ', '.join(not_found))
kkonganti@0:         else:
kkonganti@0:             logging.warning(f'Did not find ontologies: ' + ', '.join(not_found))
kkonganti@0:     if ontol_dic == {}:
kkonganti@0:         sys.exit('Zero ontologies found from user-given list')
kkonganti@0:     return(ontol_dic)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _check_make(db_file, remake_cache, ontol_interest):
kkonganti@0:     '''Check if database file should be remade'''
kkonganti@0:     if os.path.exists(db_file) and remake_cache == False:
kkonganti@0:         if os.path.exists(os.path.join(owl_dir, 'cached_ontologies.pickle')):
kkonganti@0:             if ontol_interest == _pickle_load(os.path.join(owl_dir, 'cached_ontologies.pickle')):
kkonganti@0:                 return(False)
kkonganti@0:     try:
kkonganti@0:         os.remove(db_file)
kkonganti@0:     except(FileNotFoundError):
kkonganti@0:         pass
kkonganti@0:     return(True)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _db_insert(db_cursor, table_name, key_term, val_term):
kkonganti@0:     '''Insert new data into a database table'''
kkonganti@0:     if key_term.strip()==val_term.strip() or key_term.strip()=='' or val_term.strip()=='':
kkonganti@0:         return
kkonganti@0:     db_cursor.execute(f"INSERT OR IGNORE INTO {table_name} VALUES (:key,:value)",
kkonganti@0:                       {'key':key_term.strip(), 'value':val_term.strip()})
kkonganti@0:     
kkonganti@0: 
kkonganti@0: def _get_imports(file_handle):
kkonganti@0:     '''Check for required imports; append any new patterns to pattern_strs'''
kkonganti@0:     pattern_strs = []
kkonganti@0:     pattern_strs.append('<owl:imports rdf:resource=\"&obo;(.*)"/>')
kkonganti@0:     pattern_strs.append('<owl:imports rdf:resource=\"(http://.*)\"/>')
kkonganti@0:     whole_file = str(file_handle.read())
kkonganti@0:     for patt_str in pattern_strs:
kkonganti@0:         import_match = re.findall(patt_str, whole_file)
kkonganti@0:         if import_match != []:
kkonganti@0:             import_match = [x if re.search('^http:',x) else purl_link+x for x in import_match]
kkonganti@0:             break
kkonganti@0:     return(import_match)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _section_file(file_handle, break_pattern,
kkonganti@0:                  stt_at='    // Classes', end_at='    // Annotations'):
kkonganti@0:     '''Break OWL files into readable sections for each ontology accession'''
kkonganti@0:     whole_file = str(file_handle.read())
kkonganti@0:     if stt_at != '':
kkonganti@0:         if re.search(stt_at, whole_file):
kkonganti@0:             whole_file = ''.join(whole_file.split(stt_at)[1:])
kkonganti@0:     if end_at != '':
kkonganti@0:         if re.search(end_at, whole_file):
kkonganti@0:             whole_file = ''.join(whole_file.split(end_at)[:-1])
kkonganti@0:     file_sections = whole_file.split(break_pattern)
kkonganti@0:     return(file_sections[1:-1])
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def _labels_synonyms(obo_list, have_label=False):
kkonganti@0:     '''Identify labels, ids and exact ontology synonyms'''
kkonganti@0:     obo_ids = []
kkonganti@0:     for obo_string in obo_list:
kkonganti@0:         id_pattern = '(\w+) -->'
kkonganti@0:         lab_pattern = '\<rdfs:label[\s\S]*?\>([\s\S]+?)\<\/rdfs:label\>'
kkonganti@0:         syn_pattern = '\<oboInOwl:hasExactSynonym[\s\S]*?\>(.*?)\</oboInOwl:hasExactSynonym'
kkonganti@0:         obo_id = re.search(id_pattern, obo_string).group(1)
kkonganti@0:         # Do not save ids that are not formatted as expected, or placeholders/obsolete/Apollo:SV
kkonganti@0:         if re.search('APOLLO:SV', obo_id) or re.search('APOLLO_SV', obo_id):
kkonganti@0:             continue
kkonganti@0:         elif obo_id in missing_ontol_labels:
kkonganti@0:             continue
kkonganti@0:         elif re.search(':', obo_id) or re.search('_', obo_id):
kkonganti@0:             try:
kkonganti@0:                 obo_label = re.findall(lab_pattern, obo_string)[-1] 
kkonganti@0:                 if re.search('^obsolete:', obo_label):
kkonganti@0:                     continue
kkonganti@0:             except(IndexError):
kkonganti@0:                 obo_label = ''
kkonganti@0:             obo_synonyms = re.findall(syn_pattern, obo_string)
kkonganti@0:             obo_ids.append([str(obo_id.lstrip('_')), str(obo_label), obo_synonyms])
kkonganti@0:     if have_label:
kkonganti@0:         obo_ids = [x for x in obo_ids if x[1] != '']
kkonganti@0:     return(obo_ids)
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def get_synonyms(remake_cache, ontol_interest):
kkonganti@0:     '''Create database of predefined synonyms'''
kkonganti@0:     os.makedirs(owl_dir, exist_ok=True)
kkonganti@0:     if _check_make(synonym_db, remake_cache, ontol_interest) == False:
kkonganti@0:         return
kkonganti@0: 
kkonganti@0:     conn = sqlite3.connect(synonym_db)
kkonganti@0:     c = conn.cursor()
kkonganti@0:     c.execute("CREATE TABLE label_synonyms (key TEXT, value TEXT, UNIQUE(key, value))")
kkonganti@0:     print('{:<40}'.format('\tRetrieving predefined synonyms'),end='\r')
kkonganti@0:     with open('lexmapr/predefined_resources/resource_synonyms.csv', 'r') as IN_file:
kkonganti@0:         for read_line in IN_file.readlines():
kkonganti@0:             split_line = read_line.split(',')
kkonganti@0:             try:
kkonganti@0:                 ontol_name = split_line[2].split('_')[0].upper()
kkonganti@0:             except(IndexError):
kkonganti@0:                 ontol_name = ontol_interest[0]
kkonganti@0:             if ontol_name in ontol_interest:
kkonganti@0:                 _db_insert(c, 'label_synonyms', punctuation_treatment(split_line[0]),
kkonganti@0:                            punctuation_treatment(split_line[1]))
kkonganti@0:     conn.commit()
kkonganti@0:     with open('lexmapr/predefined_resources/SynLex.csv', 'r') as IN_file:
kkonganti@0:         IN_file.readline()
kkonganti@0:         for read_line in IN_file.readlines():
kkonganti@0:             split_line = read_line.split(',')
kkonganti@0:             try:
kkonganti@0:                 ontol_name = split_line[2].split('_')[0].upper().strip()
kkonganti@0:             except(IndexError):
kkonganti@0:                 ontol_name = ontol_interest[0]
kkonganti@0:             if ontol_name in ontol_interest:
kkonganti@0:                 _db_insert(c, 'label_synonyms', punctuation_treatment(split_line[0]),
kkonganti@0:                            punctuation_treatment(split_line[1]))
kkonganti@0:     conn.commit()
kkonganti@0:     conn.close()
kkonganti@0:     return
kkonganti@0: 
kkonganti@0: 
kkonganti@0: def get_resource_ids(remake_cache, ontol_interest):
kkonganti@0:     '''Create database of online resources and update synonym database'''
kkonganti@0:     if _check_make(ontol_db, remake_cache, ontol_interest) == False:
kkonganti@0:         return
kkonganti@0: 
kkonganti@0:     conn = sqlite3.connect(ontol_db)
kkonganti@0:     c = conn.cursor()
kkonganti@0:     c.execute("CREATE TABLE non_standard_resource_ids (key TEXT, value TEXT, UNIQUE(key))")
kkonganti@0:     c.execute("CREATE TABLE standard_resource_labels (key TEXT, value TEXT, UNIQUE(key, value))")
kkonganti@0:     c.execute("""CREATE TABLE standard_resource_permutations (key TEXT, value TEXT,
kkonganti@0:                                                               UNIQUE(key, value))""")
kkonganti@0:     sonn = sqlite3.connect(synonym_db)
kkonganti@0:     s = sonn.cursor()
kkonganti@0: 
kkonganti@0:     ontol_dic = _get_ontols(ontol_interest)
kkonganti@0:     ontol_urls = [purl_link+x[1] for x in ontol_dic.values()]
kkonganti@0:     for ontol_name in ontol_dic:
kkonganti@0:         print('{:<40}'.format('\tRetrieving '+ontol_name[:15]+' terms'), end='\r')
kkonganti@0:         no_label = set()
kkonganti@0:         found_label = set()
kkonganti@0: 
kkonganti@0:         if not os.path.isfile(os.path.join(owl_dir,ontol_name+'.owl')):
kkonganti@0:             owl_download = requests.get(embl_ontologies+ontol_dic[ontol_name][1])
kkonganti@0:             with open(os.path.join(owl_dir,ontol_name+'.owl'), 'w', encoding='utf-8') as O_file:
kkonganti@0:                 O_file.write(owl_download.content.decode('utf-8'))
kkonganti@0: 
kkonganti@0:         with open(os.path.join(owl_dir,ontol_name+'.owl'),'r', encoding='utf-8') as OWL_file:
kkonganti@0:             owl_sections = _section_file(OWL_file, '<!-- '+purl_link)
kkonganti@0:             obo_ids = _labels_synonyms(owl_sections)
kkonganti@0:             OWL_file.seek(0)
kkonganti@0:             import_urls = [x for x in _get_imports(OWL_file) if x not in ontol_urls]            
kkonganti@0: 
kkonganti@0:         for imp_url in import_urls:
kkonganti@0:             imp_download = requests.get(imp_url)
kkonganti@0:             imp_file = os.path.join(owl_dir,imp_url.split('/')[-1])
kkonganti@0:             with open(imp_file, 'w', encoding='utf-8') as T_file:
kkonganti@0:                 T_file.write(imp_download.content.decode('utf-8'))
kkonganti@0:             # Assume possible to get duplicate names for different imports, so rewrite file
kkonganti@0:             with open(imp_file, 'r', encoding='utf-8') as T_file:
kkonganti@0:                 owl_sections = _section_file(T_file, '<!-- '+purl_link, stt_at='', end_at='')
kkonganti@0:                 obo_ids.extend(_labels_synonyms(owl_sections, have_label=True))
kkonganti@0:             os.remove(imp_file)
kkonganti@0:         # TODO: more elegant way to find which ids don't have labels and to remove duplicates?
kkonganti@0:         [no_label.add(x[0]) if x[1]=='' else found_label.add(x[0]) for x in obo_ids]
kkonganti@0:         no_label_found = list(no_label.difference(found_label))
kkonganti@0:         for ontol_term in obo_ids:
kkonganti@0:             if ontol_term[0] in no_label_found:
kkonganti@0:                 ontol_acc = ontr.Ontology_accession(':'+ontol_term[0])
kkonganti@0:                 if ontol_acc.label != 'unk':
kkonganti@0:                     if not re.search('^obsolete:', ontol_acc.label):
kkonganti@0:                         ontol_term[1] = ontol_acc.label
kkonganti@0:                 time.sleep(0.05)
kkonganti@0:             if ontol_term[1] == '':
kkonganti@0:                 continue
kkonganti@0:             ontol_label = punctuation_treatment(ontol_term[1])
kkonganti@0:             _db_insert(c, 'non_standard_resource_ids', ontol_term[0], ontol_term[1])
kkonganti@0:             _db_insert(c, 'standard_resource_labels', ontol_label, ontol_term[0])
kkonganti@0:             # Note from v 0.7: To limit performance overhead, we ignore resource labels with
kkonganti@0:             # more than 7 tokens, as permutating too many tokens can be costly.
kkonganti@0:             # Ignore NCBI taxon terms
kkonganti@0:             if len(word_tokenize(ontol_label)) < 7 and len(word_tokenize(ontol_label)) > 1:
kkonganti@0:                 if not re.search('NCBITaxon', ontol_term[0]):
kkonganti@0:                     for permutation in get_resource_label_permutations(ontol_label):
kkonganti@0:                         _db_insert(c,'standard_resource_permutations',permutation,ontol_term[0])
kkonganti@0:                 # Add abbreviated binomials from NCBITaxons; TODO: may get wrong combinations?
kkonganti@0:                 elif len(word_tokenize(ontol_label)) == 2:
kkonganti@0:                     bi_name = ontol_label.split()
kkonganti@0:                     _db_insert(c, 'standard_resource_permutations',
kkonganti@0:                                bi_name[0][0]+' '+bi_name[1], ontol_term[0])
kkonganti@0:             for syn_term in ontol_term[2]:
kkonganti@0:                 _db_insert(s,'label_synonyms',punctuation_treatment(str(syn_term)),ontol_label)
kkonganti@0:         conn.commit()
kkonganti@0:         sonn.commit()
kkonganti@0: 
kkonganti@0:     conn.close()
kkonganti@0:     sonn.close()
kkonganti@0:     _pickle_save(ontol_interest, os.path.join(owl_dir,'cached_ontologies.pickle'))
kkonganti@0:     return