botShell/utilities.py at master · AY-ME/botShell · 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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import demjson
from os.path import dirname, join, realpath, isfile
import aiohttp
from requests.sessions import session
from aiohttp import connector


class utilities:
    WD = dirname(realpath(__file__))
    client = None
    devs = []
    config = {}
    user_steps = {}
    plugins = []
    public_plugins = []
    crons = []

    @classmethod
    def markdown_escape(cls, text):
        text = text.replace("_", "\\_")
        text = text.replace("[", "\\{")
        text = text.replace("*", "\\*")
        text = text.replace("`", "\\`")
        return text

    @classmethod
    def prRed(cls, skk):
        print("\033[91m{}\033[00m".format(skk))

    @classmethod
    def prGreen(cls, skk):
        print("\033[92m{}\033[00m".format(skk))

    @classmethod
    def prYellow(cls, skk):
        print("\033[93m{}\033[00m".format(skk))

    @classmethod
    def prLightPurple(cls, skk):
        print("\033[94m{}\033[00m".format(skk))

    @classmethod
    def prPurple(cls, skk):
        print("\033[33m{}\033[00m".format(skk))

    @classmethod
    def prCyan(cls, skk):
        print("\033[96m{}\033[00m".format(skk))

    @classmethod
    def prLightGray(cls, skk):
        print("\033[97m{}\033[00m".format(skk))

    @classmethod
    def prBlack(cls, skk):
        print("\033[98m{}\033[00m".format(skk))

    @classmethod
    def is_group(cls, message):
        if not message.sender_id == message.chat_id:
            return True
        return False

    @classmethod
    async def get(cls, url, params=None, headers=None):
        connector = aiohttp.TCPConnector(verify_ssl=False)
        session = aiohttp.ClientSession(connector=connector)
        resp = await session.get(url, params=params, headers=headers)
        await session.close()
        await connector.close()
        return await resp.text()

    @classmethod
    def save_config(cls):
        file = open(join(cls.WD, "config.json"), "w")
        file.write(demjson.encode(cls.config))
        file.close()

    @classmethod
    def get_config(cls):

        file = open(join(cls.WD, "config.json"), "r")
        cls.config = demjson.decode(file.read())
        file.close()
        return cls.config

    @classmethod
    def load_plugins(cls):
        try:
            cls.plugins = []
            cls.public_plugins = []
            for pluginName in cls.config["plugins"]:
                plugin_dir = join(cls.WD, "plugins", pluginName + ".py")
                print("plugin " + str(pluginName))
                if isfile(plugin_dir):
                    values = {}
                    with open(plugin_dir, encoding="utf-8") as f:
                        code = compile(f.read(), plugin_dir, "exec")
                        exec(code, values)
                        f.close()
                    plugin = values["plugin"]
                    if not plugin["sudo"] and "usage" in plugin:
                        cls.public_plugins.append(plugin)
                    cls.plugins.append(plugin)
                else:
                    cls.config["plugins"].remove(pluginName)
                    cls.prRed(
                        pluginName + " : not found and will be removed from config.json"
                    )
            cls.save_config()
        except Exception as e:
            print("Error : " + str(e))

    @classmethod
    def load_plugin(cls, name):
        plugin_dir = join(cls.WD, "plugins", name + ".py")
        print("plugin " + str(name))
        values = {}
        with open(plugin_dir, encoding="utf-8") as f:
            code = compile(f.read(), plugin_dir, "exec")
            exec(code, values)
            f.close()
        plugin = values["plugin"]
        return plugin