def main(socket_path, port):
from alarmbot.views import app
from alarmbot.receptionist import dp
logger.level = logbook.DEBUG
logging.basicConfig(level=logging.INFO)
loop = asyncio.get_event_loop()
executor = Executor(dp, skip_updates=True, loop=loop)
logger.info('{}', executor)
executor.set_webhook(web_app=app)
if not socket_path:
executor.run_app(port=port)
else:
executor.run_app(path=socket_path)
class NazurinDispatcher(Dispatcher):
def __init__(self, bot: NazurinBot):
super().__init__(bot)
self.middleware.setup(AuthMiddleware())
self.filters_factory.bind(URLFilter,
event_handlers=[self.message_handlers])
self.server = NazurinServer(bot)
self.server.on_startup.append(self.on_startup)
self.executor = Executor(self)
def init(self):
self.bot.init()
self.register_message_handler(
self.update_collection,
URLFilter(),
content_types=[ContentType.TEXT, ContentType.PHOTO])
def register_message_handler(self, callback, *args, **kwargs):
return super().register_message_handler(self.async_task(callback),
*args, **kwargs)
async def on_startup(self, dp):
await self.bot.set_webhook(config.WEBHOOK_URL + config.TOKEN,
allowed_updates=AllowedUpdates.MESSAGE)
def start(self):
self.init()
if config.ENV == 'production':
logger.info('Set webhook')
self.executor.set_webhook(webhook_path='/' + config.TOKEN,
web_app=self.server)
# Tell aiohttp to use main thread event loop instead of creating a new one
# otherwise bot commands will run in a different loop
# from main thread functions and classes like Mongo and Mega.api_upload,
# resulting in RuntimeError: Task attached to different loop
self.executor.run_app(host="0.0.0.0",
port=config.PORT,
loop=asyncio.get_event_loop())
else:
# self.server.start()
executor.start_polling(self, skip_updates=True)
async def update_collection(self, message: Message, urls: List[str]):
await self.bot.updateCollection(urls, message)
await message.reply('Done!')
class NazurinDispatcher(Dispatcher):
def __init__(self, bot: NazurinBot):
super().__init__(bot)
self.middleware.setup(AuthMiddleware())
self.filters_factory.bind(URLFilter,
event_handlers=[self.message_handlers])
self.server = NazurinServer(bot)
self.server.on_startup.append(self.on_startup)
self.executor = Executor(self)
def init(self):
self.bot.init()
self.register_message_handler(
self.update_collection,
URLFilter(),
content_types=[ContentType.TEXT, ContentType.PHOTO])
def register_message_handler(self, callback, *args, **kwargs):
return super().register_message_handler(self.async_task(callback),
*args, **kwargs)
async def on_startup(self, dp):
await self.bot.set_webhook(config.WEBHOOK_URL + config.TOKEN,
allowed_updates=AllowedUpdates.MESSAGE)
def start(self):
self.init()
if config.ENV == 'production':
logger.info('Set webhook')
self.executor.set_webhook(webhook_path='/' + config.TOKEN,
web_app=self.server)
self.executor.run_app(host="0.0.0.0", port=config.PORT)
else:
# self.server.start()
executor.start_polling(self, skip_updates=True)
async def update_collection(self, message: Message, urls: List[str]):
await self.bot.updateCollection(urls, message)
await message.reply('Done!')