updatefindarticles

huixiangdou/service/findarticles.py

@@ -60,7 +60,7 @@ class ArticleRetrieval:
         if not os.path.exists(self.repo_dir):
             os.makedirs(self.repo_dir)
         print(f"Saving articles to {self.repo_dir}.")
-        
+        success = 0
         for id in tqdm(self.pmc_ids, desc="Fetching full texts", unit="article"):
             base_url = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/efetch.fcgi"
             params = {
@@ -76,13 +76,14 @@ class ArticleRetrieval:
             else:
                 with open(os.path.join(self.repo_dir,f'PMC{id}.txt'), 'w') as f:
                     f.write(full_text)
+                success += 1
 
     def save_config(self):
         config = {
             'keywords': self.keywords,
             'repo_dir': self.repo_dir,
             'pmc_ids': self.pmc_ids,
-            'len': len(self.pmc_ids),
+            'len': success,
             'retmax': self.retmax
         }
         with open(os.path.join(self.repo_dir, 'config.json'), 'w') as f: