pinglist_bot/data.py at master · bxkr/pinglist_bot · 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
import json
import os
from aiogram.types import Message


def _get_data() -> dict:
    return json.load(open('data.json'))


def chat_m(message: Message) -> None:
    chat = str(message.chat.id)
    if not os.path.exists('data.json'):
        json.dump({}, open('data.json', 'w'))
    data = _get_data()
    if chat not in data.keys():
        data[chat] = {}
    json.dump(data, open('data.json', 'w'))


def get_groups(message: Message) -> dict:
    data = _get_data()
    return {i: v for i, v in zip(range(len(data[str(message.chat.id)])), list(data[str(message.chat.id)].keys()))}


def get_group_members(message: Message, index: int) -> list:
    data = _get_data()
    return data[str(message.chat.id)][list(data[str(message.chat.id)].keys())[index]]


def add_group(message: Message, name: str, members: list) -> None:
    current_data = _get_data()
    new_data: dict = current_data.copy()
    new_data[str(message.chat.id)][name] = members
    json.dump(new_data, open('data.json', 'w'))


def add_to_group(message: Message, index: int, new_members: list) -> None:
    current_data = _get_data()
    new_data: dict = current_data.copy()
    new_data[str(message.chat.id)][list(new_data[str(message.chat.id)].keys())[index]] += new_members
    json.dump(new_data, open('data.json', 'w'))


def remove_group(message: Message, index: int):
    current_data = _get_data()
    new_data: dict = current_data.copy()
    del new_data[str(message.chat.id)][list(new_data[str(message.chat.id)].keys())[index]]
    json.dump(new_data, open('data.json', 'w'))


def remove_from_group(message: Message, index: int, username: str) -> None:
    current_data = _get_data()
    new_data: dict = current_data.copy()
    del new_data[str(message.chat.id)][
        list(new_data[str(message.chat.id)].keys())[index]][
        new_data[str(message.chat.id)][list(new_data[str(message.chat.id)].keys())[index]].index(username)
    ]
    json.dump(new_data, open('data.json', 'w'))