Function Optimization #67

Former-commit-id: 4fd4bd2a95c68a4012817c47a5e131cf76833b1b

scripts/get_extra_networks.py

@@ -1,4 +1,16 @@
+# -*- coding: UTF-8 -*-
+
 from modules import script_callbacks, extra_networks, prompt_parser, shared, ui_extra_networks
+import json
+import os
+
+filters = [
+    'filename',
+    'description',
+    'search_term',
+    'local_preview',
+    'metadata',
+]
 
 def get_extra_networks():
     result = []
@@ -10,7 +22,44 @@ def get_extra_networks():
                 'items': []
             }
             for item in extra_page.list_items():
+                # 解析metadata
+                output_name = None
+                try:
+                    if 'metadata' in item and item['metadata']:
+                        metadata = json.loads(item['metadata'])
+                        if metadata and 'ss_output_name' in metadata:
+                            output_name = metadata['ss_output_name']
+                except Exception as e:
+                    pass
+                item['output_name'] = output_name
+
+                # 获取civitai.info
+                item['civitai_info'] = {}
+                try:
+                    if 'filename' in item and item['filename']:
+                        if extra_page.name == 'textual inversion':
+                            base, ext = os.path.splitext(item['filename'])
+                            info_file = base + '.civitai.info'
+                        else:
+                            info_file = item['filename'] + '.civitai.info'
+                        if os.path.isfile(info_file):
+                            with open(info_file, 'r') as f:
+                                info = json.load(f)
+                                info = {
+                                    'name': info.get('name', ''),
+                                    'model': info.get('model', {}),
+                                }
+                                item['civitai_info'] = info
+                except Exception as e:
+                    pass
+
+                # 过滤掉不需要的字段
+                for filter in filters:
+                    if filter in item:
+                        del item[filter]
+
                 resultItem['items'].append(item)
+
             result.append(resultItem)
     except Exception as e:
         pass