Browse Source

fix: api / moderation extension import error (#3037)

tags/0.5.11
takatost 1 year ago
parent
commit
6817eab5f1
No account linked to committer's email address
1 changed files with 12 additions and 4 deletions
  1. 12
    4
      api/core/extension/extensible.py

+ 12
- 4
api/core/extension/extensible.py View File

@@ -1,4 +1,5 @@
import enum
import importlib
import json
import logging
import os
@@ -6,7 +7,6 @@ from typing import Any, Optional

from pydantic import BaseModel

from core.utils.module_import_helper import load_single_subclass_from_source
from core.utils.position_helper import sort_to_dict_by_position_map


@@ -73,9 +73,17 @@ class Extensible:

# Dynamic loading {subdir_name}.py file and find the subclass of Extensible
py_path = os.path.join(subdir_path, extension_name + '.py')
try:
extension_class = load_single_subclass_from_source(extension_name, py_path, cls)
except Exception:
spec = importlib.util.spec_from_file_location(extension_name, py_path)
mod = importlib.util.module_from_spec(spec)
spec.loader.exec_module(mod)

extension_class = None
for name, obj in vars(mod).items():
if isinstance(obj, type) and issubclass(obj, cls) and obj != cls:
extension_class = obj
break

if not extension_class:
logging.warning(f"Missing subclass of {cls.__name__} in {py_path}, Skip.")
continue


Loading…
Cancel
Save