forked from AI621/ai621
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
from asyncio import PriorityQueue
|
|
from telethon import TelegramClient
|
|
import logging
|
|
import sqlite3
|
|
import coloredlogs
|
|
from process_queue import *
|
|
|
|
coloredlogs.install(level='INFO')
|
|
|
|
# https://docs.telethon.dev/en/stable/basic/signing-in.html
|
|
api_id = <YOUR TG API ID HERE>
|
|
api_hash = <YOUR TG API HASH HERE>
|
|
|
|
# Get it from @BotFather bot.
|
|
bot_token = <YOUR BOT TOKEN HERE>
|
|
|
|
client = TelegramClient('bot', api_id, api_hash)
|
|
|
|
e621 = sqlite3.connect('e621.db')
|
|
e621.row_factory = sqlite3.Row
|
|
conn = sqlite3.connect('ai621.db', isolation_level=None)
|
|
conn.row_factory = sqlite3.Row
|
|
|
|
client.conn = conn
|
|
client.process = None
|
|
|
|
# To get channel ID: Send something in private(!) channel, tap/click on it,
|
|
# choose "Copy post link", paste it somewhere. It should look like this:
|
|
# https://t.me/c/12345/2
|
|
# 12345 part is ID - add "-100" prefix before it (like this: -10012345)
|
|
# Replace placeholder with result:
|
|
client.log_channel_id = <ID OF LOG CHANNEL>
|
|
client.main_channel_id = <ID OF RAW CHANNEL>
|
|
|
|
client.queue = PriorityQueue()
|
|
client.media_lock = asyncio.Lock()
|
|
client.process_queue = False
|
|
|
|
client.admin_id = USER ID OF THE ADMIN
|
|
|
|
workers = [
|
|
Worker('http://127.0.0.1:7860', client, 'armorlink'),
|
|
#Worker('http://127.0.0.1:9001', client, 'armorlink-low'),
|
|
#Worker('http://local.proxy:9000', client, 'g14')
|
|
]
|
|
|
|
log = logging.getLogger('bot')
|
|
|
|
|
|
|