Spotload/bot.py at main · nyx47rd/Spotload · GitHub

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 telegram.ext import Updater, CommandHandler
import os, zipfile
from utils import download_spotify, can_download_more, increase_count

TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")

def start(update, context):
    update.message.reply_text("Send /get or /playlist with a Spotify link 💿")

def get(update, context):
    if not can_download_more():
        update.message.reply_text("❌ Haftalık 500 şarkı limitine ulaşıldı 😔")
        return

    url = context.args[0]
    paths = download_spotify(url, single=True)
    for path in paths:
        with open(path, 'rb') as song:
            update.message.reply_audio(song)
        os.remove(path)
        increase_count()

def playlist(update, context):
    if not can_download_more():
        update.message.reply_text("❌ Haftalık limit doldu 😭")
        return

    url = context.args[0]
    paths = download_spotify(url, single=False)
    zipname = "playlist.zip"

    with zipfile.ZipFile(zipname, 'w') as zipf:
        for song in paths:
            zipf.write(song)
            increase_count()
            os.remove(song)

    with open(zipname, 'rb') as zf:
        update.message.reply_document(zf)

    os.remove(zipname)

updater = Updater(TOKEN)
updater.dispatcher.add_handler(CommandHandler("start", start))
updater.dispatcher.add_handler(CommandHandler("get", get))
updater.dispatcher.add_handler(CommandHandler("playlist", playlist))

updater.start_polling()