mirror of
https://github.com/ChronosX88/mastodon-bridge-bot.git
synced 2024-11-08 16:21:01 +00:00
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
|
import asyncio
|
||
|
|
import toml
|
||
|
|
import argparse
|
||
|
|
from mastodon import Mastodon
|
||
|
|
from telethon import TelegramClient, events
|
||
|
|
from telethon.events.common import EventBuilder
|
||
|
|
from telethon.tl.functions.channels import JoinChannelRequest, GetMessagesRequest
|
||
|
|
from telethon.tl.functions.contacts import ResolveUsernameRequest
|
||
|
|
from telethon.tl.types import InputChannel
|
||
|
|
|
||
|
|
|
||
|
|
class BridgeBot:
|
||
|
|
def __init__(self, cfg: dict):
|
||
|
|
self.config = cfg
|
||
|
|
self.mastodon_client = Mastodon(
|
||
|
|
client_id=cfg["mastodon"]["client_id"],
|
||
|
|
client_secret=cfg["mastodon"]["client_secret"],
|
||
|
|
access_token=cfg["mastodon"]["access_token"],
|
||
|
|
api_base_url=cfg["mastodon"]["api_base_url"]
|
||
|
|
)
|
||
|
|
self.tg_client = TelegramClient(cfg["telegram"]["session_file"], cfg["telegram"]["api_id"],
|
||
|
|
cfg["telegram"]["api_hash"])
|
||
|
|
|
||
|
|
async def run(self):
|
||
|
|
await self.tg_client.connect()
|
||
|
|
await self.tg_client.start()
|
||
|
|
for ch_id in self.config["telegram"]["channels"]:
|
||
|
|
result = await self.tg_client(ResolveUsernameRequest(ch_id))
|
||
|
|
channel = InputChannel(result.peer.channel_id, result.chats[0].access_hash)
|
||
|
|
await self.tg_client(JoinChannelRequest(channel))
|
||
|
|
self.tg_client.add_event_handler(self._tg_event_handler)
|
||
|
|
await self.tg_client.run_until_disconnected()
|
||
|
|
|
||
|
|
@events.register(events.NewMessage())
|
||
|
|
async def _tg_event_handler(self, event: events.NewMessage.Event):
|
||
|
|
if event.message.post:
|
||
|
|
channel = await event.get_chat()
|
||
|
|
if channel.broadcast:
|
||
|
|
if channel.username in self.config["telegram"]["channels"]:
|
||
|
|
print("dobbry vechur")
|
||
|
|
self.mastodon_client.toot(event.message.text)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('--config', type=str,
|
||
|
|
help='A path to bot configuration')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
config: dict = toml.loads(open(args.config, "r").read())
|
||
|
|
print(config)
|
||
|
|
bot = BridgeBot(config)
|
||
|
|
asyncio.get_event_loop().run_until_complete(bot.run())
|