This commit is contained in:
Dan Elkouby 2018-04-07 23:20:16 +03:00
parent 8c610b95e6
commit ad5c3f2d18
6 changed files with 63 additions and 18 deletions

9
stdborg.py Normal file
View File

@ -0,0 +1,9 @@
import logging
from uniborg import Uniborg
logging.basicConfig(level=logging.INFO)
borg = Uniborg("stdborg", plugin_path="stdplugins")
borg.run()

5
stdplugins/testplugin.py Normal file
View File

@ -0,0 +1,5 @@
from telethon import events
@borg.on(events.NewMessage)
async def asdf(e):
print(e.raw_text)

13
test.py
View File

@ -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
uniborg/_core.py Normal file
View File

7
uniborg/hacks.py Normal file
View File

@ -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)

View File

@ -3,25 +3,62 @@
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import asyncio
import importlib.util
import logging
from pathlib import Path
from telethon import TelegramClient
import telethon.utils
from . import hacks
class Uniborg(TelegramClient):
def __init__(self, session, **kwargs):
self.logger = logging.getLogger(session)
def __init__(self, session, *, plugin_path="plugins", **kwargs):
# TODO: handle non-string session
self._name = session
self._logger = logging.getLogger(session)
self._plugins = {}
self._plugin_path = plugin_path
super().__init__(session,
17349, "344583e45741c457fe1862106095a5eb", # yarr
**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()
self.uid = telethon.utils.get_peer_id(await self.get_me())
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}")