Modules/DeleteMusic.py at main · i-execute/Modules · 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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# CopyLeft 2026 github.com/i-execute
# Author: I_execute.t.me
# Licensed under AGPLv3.

__version__ = (1, 0, 2)
# meta developer: Execute_forge.t.me
# meta banner: https://raw.githubusercontent.com/i-execute/Modules/main/Storage/DeleteMusic/MetaBanner.jpeg

import struct
import asyncio
from telethon.tl import tlobject
from telethon.tl.types import InputDocument
from telethon import functions

from .. import loader, utils


class SavedMusic(tlobject.TLObject):
    def __init__(self, count, documents):
        self.count = count
        self.documents = documents

    @classmethod
    def from_reader(cls, reader):
        return cls(count=reader.read_int(), documents=reader.tgread_vector())


class GetSavedMusic(tlobject.TLRequest):
    def __init__(self, id, offset, limit, hash):
        self.id = id
        self.offset = offset
        self.limit = limit
        self.hash = hash

    def _bytes(self):
        return b''.join((
            b'\xe3\x7f\x8dx',
            self.id._bytes(),
            struct.pack('<i', self.offset),
            struct.pack('<i', self.limit),
            struct.pack('<q', self.hash),
        ))

    def read_result(self, reader):
        reader.read_int()
        return SavedMusic.from_reader(reader)


@loader.tds
class DeleteMusic(loader.Module):
    """Delete all music in your profile"""

    strings = {
        "name": "DeleteMusic",
        "processing": "Processing...",
        "no_tracks": "Error: no tracks in your profile",
        "deleted": "Deleted: {} tracks",
        "error": "Error: {}",
    }

    strings_ru = {
        "processing": "Обработка...",
        "no_tracks": "Ошибка: в вашем профиле нет треков",
        "deleted": "Удалено: {} треков",
        "error": "Ошибка: {}",
    }

    @loader.command(
        ru_doc="Удалить всю музыку из вашего профиля",
        en_doc="Delete all music from your profile",
    )
    async def musicrm(self, message):
        """Delete all saved music from your profile"""
        await utils.answer(message, self.strings["processing"])
        try:
            me = await self.client.get_input_entity("me")
            music = await self.client(GetSavedMusic(me, 0, 100, 0))
            if not music or not music.documents:
                return await utils.answer(message, self.strings["no_tracks"])
            c = 0
            while music and music.documents:
                for doc in music.documents:
                    try:
                        await self.client(functions.account.SaveMusicRequest(
                            id=InputDocument(
                                id=doc.id,
                                access_hash=doc.access_hash,
                                file_reference=doc.file_reference,
                            ),
                            unsave=True,
                        ))
                        c += 1
                        await asyncio.sleep(0.1)
                    except Exception:
                        pass
                music = await self.client(GetSavedMusic(me, 0, 100, 0))
            await utils.answer(message, self.strings["deleted"].format(c))
        except Exception as e:
            await utils.answer(message, self.strings["error"].format(e))