VKGroupParser/models.py at main · sergo-code/VKGroupParser · 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
import json
from requests import post


def LoadModels():
    with open('data/group.json') as file:
        models = json.load(file)
    return models


def LoadModelsKey():
    with open('data/group.json') as file:
        models = json.load(file)
    secondList = []
    for key in models.keys():
        secondList.append(key)
    return secondList


def AddModel(link, token):
    try:
        with open('data/group.json') as file:
            models = json.load(file)
            url = f"https://api.vk.com/method/utils.resolveScreenName?v=5.131&access_token={token}&screen_name={link}"

            response = post(url).text

            owner_id = json.loads(response)

            if 'response' in owner_id.keys():
                if 'object_id' in owner_id['response'].keys():
                    owner_id = owner_id['response']['object_id']
                    models.update({link: {"group_id": owner_id, "token": token, "active": "0"}})

                    with open('data/group.json', 'w') as file:
                        json.dump(models, file)
                else:
                    text = 'Ссылка введена неверно!'
                    return f'{link} не удалось добавить в список\n{text}'
            elif 'error' in owner_id.keys():
                text = 'Введен неверный токен!'
                return f'{link} не удалось добавить в список\n{text}'
    except:
        return f'{link} не удалось добавить в список'
    return f'Группа {link} успешно добавлена/обновлена.'


def DelModel(link):
    with open('data/group.json') as file:
        models = json.load(file)
    if link in models:
        del models[link]
    else:
        return f'Не удалось удалить {link}'
    with open('data/group.json', 'w') as file:
        json.dump(models, file)
    return f'Группа {link} успешно удалена.'


def toggleActiveModel(domain, toggle):
    with open('data/group.json') as file:
        models = json.load(file)
    models[domain]['active'] = toggle
    with open('data/group.json', 'w') as file:
        json.dump(models, file)


def activeModel():
    with open('data/group.json') as file:
        models = json.load(file)

    text = 'Список запущенных групп:\n'
    temp = len(text)
    for item in models.keys():
        if models[item]['active'] == '1':
            text += f'{item}\n'

    if temp != len(text):
        return text
    else:
        return 'В данный момент запущенных групп нет.'