forked from kate/uniborg
\m/
This commit is contained in:
parent
8c610b95e6
commit
ad5c3f2d18
|
@ -0,0 +1,9 @@
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from uniborg import Uniborg
|
||||||
|
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
|
||||||
|
borg = Uniborg("stdborg", plugin_path="stdplugins")
|
||||||
|
|
||||||
|
borg.run()
|
|
@ -0,0 +1,5 @@
|
||||||
|
from telethon import events
|
||||||
|
|
||||||
|
@borg.on(events.NewMessage)
|
||||||
|
async def asdf(e):
|
||||||
|
print(e.raw_text)
|
13
test.py
13
test.py
|
@ -1,13 +0,0 @@
|
||||||
from uniborg import Uniborg
|
|
||||||
from telethon import events
|
|
||||||
|
|
||||||
borg = Uniborg("uniborg")
|
|
||||||
|
|
||||||
print(borg.uid)
|
|
||||||
print(dir(borg))
|
|
||||||
|
|
||||||
@borg.on(events.NewMessage)
|
|
||||||
async def asdf(e):
|
|
||||||
print(e.raw_text)
|
|
||||||
|
|
||||||
borg.run()
|
|
|
@ -0,0 +1,7 @@
|
||||||
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
|
class ReverseList(list):
|
||||||
|
def __iter__(self):
|
||||||
|
return reversed(self)
|
|
@ -3,25 +3,62 @@
|
||||||
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
import importlib.util
|
||||||
import logging
|
import logging
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
from telethon import TelegramClient
|
from telethon import TelegramClient
|
||||||
import telethon.utils
|
import telethon.utils
|
||||||
|
|
||||||
|
from . import hacks
|
||||||
|
|
||||||
class Uniborg(TelegramClient):
|
class Uniborg(TelegramClient):
|
||||||
def __init__(self, session, **kwargs):
|
def __init__(self, session, *, plugin_path="plugins", **kwargs):
|
||||||
self.logger = logging.getLogger(session)
|
# TODO: handle non-string session
|
||||||
|
self._name = session
|
||||||
|
self._logger = logging.getLogger(session)
|
||||||
|
self._plugins = {}
|
||||||
|
self._plugin_path = plugin_path
|
||||||
|
|
||||||
super().__init__(session,
|
super().__init__(session,
|
||||||
17349, "344583e45741c457fe1862106095a5eb", # yarr
|
17349, "344583e45741c457fe1862106095a5eb", # yarr
|
||||||
**kwargs)
|
**kwargs)
|
||||||
|
|
||||||
asyncio.get_event_loop().run_until_complete(self.async_init())
|
# This is a hack, please avert your eyes
|
||||||
|
# We want this in order for the most recently added handler to take
|
||||||
|
# precedence
|
||||||
|
self._event_builders = hacks.ReverseList()
|
||||||
|
|
||||||
async def async_init(self):
|
self._loop.run_until_complete(self._async_init())
|
||||||
|
|
||||||
|
core_plugin = Path(__file__).parent / "_core.py"
|
||||||
|
self.load_plugin_from_file(core_plugin)
|
||||||
|
|
||||||
|
for p in Path().glob(f"{self._plugin_path}/*.py"):
|
||||||
|
self.load_plugin_from_file(p)
|
||||||
|
|
||||||
|
async def _async_init(self):
|
||||||
await self.start()
|
await self.start()
|
||||||
|
|
||||||
self.uid = telethon.utils.get_peer_id(await self.get_me())
|
self.uid = telethon.utils.get_peer_id(await self.get_me())
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
asyncio.get_event_loop().run_forever()
|
self._loop.run_forever()
|
||||||
|
|
||||||
|
def load_plugin(self, shortname):
|
||||||
|
self.load_plugin_from_file(f"{self._plugin_path}/{shortname}.py")
|
||||||
|
|
||||||
|
def load_plugin_from_file(self, path):
|
||||||
|
path = Path(path)
|
||||||
|
shortname = path.stem
|
||||||
|
name = f"_UniborgPlugins.{self._name}.{shortname}"
|
||||||
|
|
||||||
|
spec = importlib.util.spec_from_file_location(name, path)
|
||||||
|
mod = importlib.util.module_from_spec(spec)
|
||||||
|
|
||||||
|
mod.borg = self
|
||||||
|
mod.logger = logging.getLogger(shortname)
|
||||||
|
|
||||||
|
spec.loader.exec_module(mod)
|
||||||
|
self._plugins[shortname] = mod
|
||||||
|
self._logger.info(f"Successfully loaded plugin {shortname}")
|
||||||
|
|
Loading…
Reference in New Issue