From a20996d83ab99775b23dcfa5b5e82c95492932bc Mon Sep 17 00:00:00 2001
From: Richard Chien <richardchienthebest@gmail.com>
Date: Thu, 3 Jan 2019 19:58:56 +0800
Subject: [PATCH] Add get_loaded_plugins() function

---
 nonebot/__init__.py | 73 ++++++++-------------------------------------
 nonebot/plugin.py   | 72 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 85 insertions(+), 60 deletions(-)
 create mode 100644 nonebot/plugin.py

diff --git a/nonebot/__init__.py b/nonebot/__init__.py
index c47c4023..e6aba0de 100644
--- a/nonebot/__init__.py
+++ b/nonebot/__init__.py
@@ -1,8 +1,5 @@
 import asyncio
-import importlib
 import logging
-import os
-import re
 from typing import Any, Optional
 
 import aiocqhttp.message
@@ -104,66 +101,22 @@ def run(host: Optional[str] = None, port: Optional[int] = None,
     get_bot().run(host=host, port=port, *args, **kwargs)
 
 
-_plugins = set()
-
-
-def load_plugin(module_name: str) -> bool:
-    """
-    Load a module as a plugin.
-
-    :param module_name: name of module to import
-    :return: successful or not
-    """
-    try:
-        _plugins.add(importlib.import_module(module_name))
-        logger.info(f'Succeeded to import "{module_name}"')
-        return True
-    except Exception as e:
-        logger.error(f'Failed to import "{module_name}", error: {e}')
-        logger.exception(e)
-        return False
-
-
-def load_plugins(plugin_dir: str, module_prefix: str) -> int:
-    """
-    Find all non-hidden modules or packages in a given directory,
-    and import them with the given module prefix.
-
-    :param plugin_dir: plugin directory to search
-    :param module_prefix: module prefix used while importing
-    :return: number of plugins successfully loaded
-    """
-    count = 0
-    for name in os.listdir(plugin_dir):
-        path = os.path.join(plugin_dir, name)
-        if os.path.isfile(path) and \
-                (name.startswith('_') or not name.endswith('.py')):
-            continue
-        if os.path.isdir(path) and \
-                (name.startswith('_') or not os.path.exists(
-                    os.path.join(path, '__init__.py'))):
-            continue
-
-        m = re.match(r'([_A-Z0-9a-z]+)(.py)?', name)
-        if not m:
-            continue
-
-        if load_plugin(f'{module_prefix}.{m.group(1)}'):
-            count += 1
-    return count
-
-
-def load_builtin_plugins() -> int:
-    """
-    Load built-in plugins distributed along with "nonebot" package.
-    """
-    plugin_dir = os.path.join(os.path.dirname(__file__), 'plugins')
-    return load_plugins(plugin_dir, 'nonebot.plugins')
-
-
 from .exceptions import *
+from .plugin import (load_plugin, load_plugins, load_builtin_plugins,
+                     get_loaded_plugins)
 from .message import message_preprocessor, Message, MessageSegment
 from .command import on_command, CommandSession, CommandGroup
 from .natural_language import on_natural_language, NLPSession, NLPResult
 from .notice_request import (on_notice, NoticeSession,
                              on_request, RequestSession)
+
+__all__ = [
+    'NoneBot', 'scheduler', 'init', 'get_bot', 'run',
+    'CQHttpError',
+    'load_plugin', 'load_plugins', 'load_builtin_plugins',
+    'get_loaded_plugins',
+    'message_preprocessor', 'Message', 'MessageSegment',
+    'on_command', 'CommandSession', 'CommandGroup',
+    'on_natural_language', 'NLPSession', 'NLPResult',
+    'on_notice', 'NoticeSession', 'on_request', 'RequestSession',
+]
diff --git a/nonebot/plugin.py b/nonebot/plugin.py
new file mode 100644
index 00000000..6135423e
--- /dev/null
+++ b/nonebot/plugin.py
@@ -0,0 +1,72 @@
+import importlib
+import os
+import re
+from collections import namedtuple
+from typing import Set
+
+from .log import logger
+
+Plugin = namedtuple('Plugin', ['module', 'name', 'usage'])
+
+_plugins: Set[Plugin] = set()
+
+
+def load_plugin(module_name: str) -> bool:
+    """
+    Load a module as a plugin.
+
+    :param module_name: name of module to import
+    :return: successful or not
+    """
+    try:
+        module = importlib.import_module(module_name)
+        name = getattr(module, '__plugin_name__', None)
+        usage = getattr(module, '__plugin_usage__', None)
+        _plugins.add(Plugin(module, name, usage))
+        logger.info(f'Succeeded to import "{module_name}"')
+        return True
+    except Exception as e:
+        logger.error(f'Failed to import "{module_name}", error: {e}')
+        logger.exception(e)
+        return False
+
+
+def load_plugins(plugin_dir: str, module_prefix: str) -> int:
+    """
+    Find all non-hidden modules or packages in a given directory,
+    and import them with the given module prefix.
+
+    :param plugin_dir: plugin directory to search
+    :param module_prefix: module prefix used while importing
+    :return: number of plugins successfully loaded
+    """
+    count = 0
+    for name in os.listdir(plugin_dir):
+        path = os.path.join(plugin_dir, name)
+        if os.path.isfile(path) and \
+                (name.startswith('_') or not name.endswith('.py')):
+            continue
+        if os.path.isdir(path) and \
+                (name.startswith('_') or not os.path.exists(
+                    os.path.join(path, '__init__.py'))):
+            continue
+
+        m = re.match(r'([_A-Z0-9a-z]+)(.py)?', name)
+        if not m:
+            continue
+
+        if load_plugin(f'{module_prefix}.{m.group(1)}'):
+            count += 1
+    return count
+
+
+def load_builtin_plugins() -> int:
+    """
+    Load built-in plugins distributed along with "nonebot" package.
+    """
+    plugin_dir = os.path.join(os.path.dirname(__file__), 'plugins')
+    return load_plugins(plugin_dir, 'nonebot.plugins')
+
+
+def get_loaded_plugins() -> Set[Plugin]:
+    return _plugins