TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow nested folders for smart plugins

Browse Source
This commit is contained in:
Dan 2018-10-23 15:43:49 +02:00
parent 9e3cf9edf7
commit 99bdaae365
1 changed files with 36 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pyrogram/client/client.py
View File

@ -34,6 +34,7 @@ from configparser import ConfigParser
from datetime import datetime from datetime import datetime
from hashlib import sha256, md5 from hashlib import sha256, md5
from importlib import import_module from importlib import import_module
from pathlib import Path
from signal import signal, SIGINT, SIGTERM, SIGABRT from signal import signal, SIGINT, SIGTERM, SIGABRT
from threading import Thread from threading import Thread
@ -980,28 +981,29 @@ class Client(Methods, BaseClient):
def load_plugins(self): def load_plugins(self):
if self.plugins_dir is not None: if self.plugins_dir is not None:
try:
dirs = os.listdir(self.plugins_dir)
except FileNotFoundError:
log.warning('No plugin loaded: "{}" directory is missing'.format(self.plugins_dir))
else:
plugins_dir = self.plugins_dir.lstrip("./").replace("/", ".")
plugins_count = 0 plugins_count = 0
for i in dirs: for path in Path(self.plugins_dir).rglob("*.py"):
module = import_module("{}.{}".format(plugins_dir, i.split(".")[0])) file_path = os.path.splitext(str(path))[0]
import_path = []
for j in dir(module): while file_path:
file_path, tail = os.path.split(file_path)
import_path.insert(0, tail)
import_path = ".".join(import_path)
module = import_module(import_path)
for name in dir(module):
# noinspection PyBroadException # noinspection PyBroadException
try: try:
handler, group = getattr(module, j) handler, group = getattr(module, name)
if isinstance(handler, Handler) and isinstance(group, int): if isinstance(handler, Handler) and isinstance(group, int):
self.add_handler(handler, group) self.add_handler(handler, group)
log.info('{}("{}") from "{}/{}" loaded in group {}'.format( log.info('{}("{}") from "{}" loaded in group {}'.format(
type(handler).__name__, j, self.plugins_dir, i, group) type(handler).__name__, name, import_path, group))
)
plugins_count += 1 plugins_count += 1
except Exception: except Exception: