From f4afc4af10a8d3468114e205087e45d63b4f1b0d Mon Sep 17 00:00:00 2001
From: Manuel Schmid <manuel.schmid@odt.net>
Date: Fri, 2 Feb 2024 01:19:21 +0100
Subject: [PATCH] feat: add method get_scheme to Metadata

---
 modules/metadata.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/modules/metadata.py b/modules/metadata.py
index d80bad01..5c43bef0 100644
--- a/modules/metadata.py
+++ b/modules/metadata.py
@@ -23,7 +23,12 @@ def get_sha256(filepath):
 
     return hash_cache[filepath]
 
+
 class MetadataParser(ABC):
+    @abstractmethod
+    def get_scheme(self) -> MetadataScheme:
+        raise NotImplementedError
+
     @abstractmethod
     def parse_json(self, metadata: dict) -> dict:
         raise NotImplementedError
@@ -34,6 +39,8 @@ class MetadataParser(ABC):
 
 
 class A1111MetadataParser(MetadataParser):
+    def get_scheme(self) -> MetadataScheme:
+        return MetadataScheme.A1111
 
     fooocus_to_a1111 = {
         'negative_prompt': 'Negative prompt',
@@ -184,6 +191,8 @@ class A1111MetadataParser(MetadataParser):
 
 
 class FooocusMetadataParser(MetadataParser):
+    def get_scheme(self) -> MetadataScheme:
+        return MetadataScheme.FOOOCUS
 
     def parse_json(self, metadata: dict) -> dict:
         model_filenames = modules.config.model_filenames.copy()