mirror of
https://github.com/wah-su/stickerbridge.git
synced 2025-04-04 23:34:40 +00:00
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import asyncio
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
import yaml
|
|
import logging
|
|
|
|
from nio import AsyncClient, SyncResponse, RoomMessageText, InviteEvent, InviteMemberEvent
|
|
|
|
from callbacks import Callbacks
|
|
from chat_functions import upload_avatar
|
|
from telegram_exporter import TelegramExporter
|
|
|
|
|
|
async def main():
|
|
if not os.path.exists('config.yaml'):
|
|
shutil.copy('config.yaml.example', 'config.yaml')
|
|
logging.warning('Please fill in config.yaml file, then restart the bot')
|
|
return
|
|
with open("config.yaml", 'r') as config_file:
|
|
config = yaml.safe_load(config_file)
|
|
|
|
logging.basicConfig(level=os.environ.get("LOGLEVEL", config['log_level']))
|
|
|
|
client = AsyncClient(config['matrix_homeserver'], config['matrix_username'])
|
|
client.device_id = config['matrix_bot_name']
|
|
|
|
tg_exporter = TelegramExporter(config['telegram_api_id'], config['telegram_api_hash'], config['telegram_bot_token'],
|
|
'data/telegram_secrets')
|
|
await tg_exporter.connect()
|
|
|
|
callbacks = Callbacks(client, config['command_prefix'], config, tg_exporter)
|
|
client.add_response_callback(callbacks.sync, SyncResponse)
|
|
client.add_event_callback(callbacks.message, RoomMessageText)
|
|
client.add_event_callback(callbacks.autojoin_room, InviteMemberEvent)
|
|
|
|
login_response = await client.login(config['matrix_password'])
|
|
logging.info(login_response)
|
|
|
|
if os.path.exists('data/next_batch'):
|
|
with open("data/next_batch", "r") as next_batch_token:
|
|
client.next_batch = next_batch_token.read()
|
|
else:
|
|
await upload_avatar(client, 'avatar.png')
|
|
await client.set_displayname(config['matrix_bot_name'])
|
|
|
|
await client.sync_forever(30000)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|