diff --git a/scipaperloader/blueprints/config.py b/scipaperloader/blueprints/config.py index 76d9511..49bf148 100644 --- a/scipaperloader/blueprints/config.py +++ b/scipaperloader/blueprints/config.py @@ -278,6 +278,16 @@ def generate_test_papers(): except (ValueError, TypeError): paper_count = 100 + # Get the status settings + try: + dummy_paper_status = request.form.get("dummy_paper_status") + if dummy_paper_status == "new": + dummy_paper_status = "New" + else: + dummy_paper_status = random.choice(["New","Pending", "Done", "Failed"]) + except (ValueError, TypeError): + dummy_paper_status = random.choice(["New","Pending", "Done", "Failed"]) + # Get the download path for file paths download_path = DownloadPathConfig.get_path() @@ -320,7 +330,7 @@ def generate_test_papers(): type=random.choice(paper_types), language=random.choice(languages), published_online=pub_date.date(), - status=random.choice(["Pending", "Done", "Failed"]), + status=dummy_paper_status, file_path=f"{download_path}/test_paper_{i+1}.pdf" if random.random() > 0.3 else None, error_msg="Download failed: connection timeout" if random.random() < 0.1 else None, created_at=datetime.now() - timedelta(days=random.randint(0, 30)) diff --git a/scipaperloader/templates/config/database.html.jinja b/scipaperloader/templates/config/database.html.jinja index dd7e484..43bdc9f 100644 --- a/scipaperloader/templates/config/database.html.jinja +++ b/scipaperloader/templates/config/database.html.jinja @@ -33,6 +33,17 @@ Enter a number between 1 and 1000 + +