@@ -89,11 +89,11 @@ def _infer_molecule_type_detail(accession: Optional[str], gene_type: Optional[in
8989 if accession :
9090 if accession .startswith (("NM_" , "XM_" )):
9191 return "mRNA"
92- elif accession .startswith (("NC_" , "NT_" )):
92+ if accession .startswith (("NC_" , "NT_" )):
9393 return "genomic DNA"
94- elif accession .startswith (("NR_" , "XR_" )):
94+ if accession .startswith (("NR_" , "XR_" )):
9595 return "RNA"
96- elif accession .startswith ("NG_" ):
96+ if accession .startswith ("NG_" ):
9797 return "genomic region"
9898 # Fallback: infer from gene type if available
9999 if gene_type is not None :
@@ -215,7 +215,7 @@ def _extract_metadata_from_genbank(result: dict, accession: str):
215215 """Extract metadata from GenBank format (title, features, organism, etc.)."""
216216 with Entrez .efetch (db = "nuccore" , id = accession , rettype = "gb" , retmode = "text" ) as handle :
217217 record = SeqIO .read (handle , "genbank" )
218-
218+
219219 result ["title" ] = record .description
220220 result ["molecule_type_detail" ] = (
221221 "mRNA" if accession .startswith (("NM_" , "XM_" )) else
@@ -238,7 +238,7 @@ def _extract_metadata_from_genbank(result: dict, accession: str):
238238
239239 if not result .get ("organism" ) and 'organism' in record .annotations :
240240 result ["organism" ] = record .annotations ['organism' ]
241-
241+
242242 return result
243243
244244 def _extract_sequence_from_fasta (result : dict , accession : str ):
@@ -249,7 +249,10 @@ def _extract_sequence_from_fasta(result: dict, accession: str):
249249 result ["sequence" ] = str (fasta_record .seq )
250250 result ["sequence_length" ] = len (fasta_record .seq )
251251 except Exception as fasta_exc :
252- logger .warning ("Failed to extract sequence from accession %s using FASTA format: %s" , accession , fasta_exc )
252+ logger .warning (
253+ "Failed to extract sequence from accession %s using FASTA format: %s" ,
254+ accession , fasta_exc
255+ )
253256 result ["sequence" ] = None
254257 result ["sequence_length" ] = None
255258 return result
0 commit comments