From e216b4c455f7d9517aac85a03110557c80ec9c22 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Sun, 23 Mar 2025 15:29:47 +0800
Subject: [PATCH] Refactor early access checks in recipe parsers

- Updated the early access condition checks in RecipeFormatParser, StandardMetadataParser, and A1111MetadataParser to use `get` method for improved readability and safety.
- Ensured consistent handling of early access status across different parser classes.
---
 py/utils/recipe_parsers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/py/utils/recipe_parsers.py b/py/utils/recipe_parsers.py
index c83ed587..21cd60ba 100644
--- a/py/utils/recipe_parsers.py
+++ b/py/utils/recipe_parsers.py
@@ -112,7 +112,7 @@ class RecipeFormatParser(RecipeMetadataParser):
                                 civitai_info = await civitai_client.get_model_version_info(lora['modelVersionId'])
                                 if civitai_info and civitai_info.get("error") != "Model not found":
                                     # Check if this is an early access lora
-                                    if 'earlyAccessEndsAt' in civitai_info:
+                                    if civitai_info.get('earlyAccessEndsAt'):
                                         # Convert earlyAccessEndsAt to a human-readable date
                                         early_access_date = civitai_info.get('earlyAccessEndsAt', '')
                                         lora_entry['isEarlyAccess'] = True
@@ -227,7 +227,7 @@ class StandardMetadataParser(RecipeMetadataParser):
                     # Check if this LoRA exists locally by SHA256 hash
                     if civitai_info and civitai_info.get("error") != "Model not found":
                         # Check if this is an early access lora
-                        if 'earlyAccessEndsAt' in civitai_info:
+                        if civitai_info.get('earlyAccessEndsAt'):
                             # Convert earlyAccessEndsAt to a human-readable date
                             early_access_date = civitai_info.get('earlyAccessEndsAt', '')
                             lora_entry['isEarlyAccess'] = True
@@ -442,7 +442,7 @@ class A1111MetadataParser(RecipeMetadataParser):
                         civitai_info = await civitai_client.get_model_by_hash(hash_value)
                         if civitai_info and civitai_info.get("error") != "Model not found":
                             # Check if this is an early access lora
-                            if 'earlyAccessEndsAt' in civitai_info:
+                            if civitai_info.get('earlyAccessEndsAt'):
                                 # Convert earlyAccessEndsAt to a human-readable date
                                 early_access_date = civitai_info.get('earlyAccessEndsAt', '')
                                 lora_entry['isEarlyAccess'] = True