-
Notifications
You must be signed in to change notification settings - Fork 125
/
bot.py
48 lines (36 loc) · 1.58 KB
/
bot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from os import environ
import aiohttp
from pyrogram import Client, filters
API_ID = environ.get('API_ID')
API_HASH = environ.get('API_HASH')
BOT_TOKEN = environ.get('BOT_TOKEN')
API_KEY = environ.get('API_KEY')
if not all([API_ID, API_HASH, BOT_TOKEN, API_KEY]):
raise ValueError("Please set API_ID, API_HASH, BOT_TOKEN, and API_KEY environment variables.")
bot = Client('gplink bot',
api_id=API_ID,
api_hash=API_HASH,
bot_token=BOT_TOKEN,
workers=50,
sleep_threshold=10)
@bot.on_message(filters.command('start') & filters.private)
async def start_handler(bot, message):
await message.reply(
f"**Hi {message.chat.first_name}!**\n\n"
"I'm GPlink bot. Just send me a link, and I'll provide you with a shortened version.")
@bot.on_message(filters.regex(r'https?://[^\s]+') & filters.private)
async def link_handler(bot, message):
link = message.matches[0].group(0)
try:
short_link = await get_short_link(link)
await message.reply(f'Here is your [short link]({short_link})', quote=True)
except Exception as e:
await message.reply('Oops! Something went wrong while shortening the link. Please try again later.', quote=True)
async def get_short_link(link):
url = 'https://gplinks.in/api'
params = {'api': API_KEY, 'url': link}
async with aiohttp.ClientSession() as session:
async with session.get(url, params=params, raise_for_status=True) as response:
data = await response.json()
return data.get("shortenedUrl", "No short link found")
bot.run()