summaryrefslogtreecommitdiffstatshomepage
path: root/vk/platform.py
blob: fccaa173e0d0ef019d00cd36ba4620351af7a1d8 (plain) (blame)
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
# Copyright (c) 2016 Egor Tensin <egor@tensin.name>
# This file is part of the "VK scripts" project.
# For details, see https://github.com/egor-tensin/vk-scripts.
# Distributed under the MIT License.

from enum import Enum
import re


class Platform(Enum):
    # https://dev.vk.com/en/reference/objects/user#last_seen
    MOBILE = 1
    IPHONE = 2
    IPAD = 3
    ANDROID = 4
    WINDOWS_PHONE = 5
    WINDOWS10 = 6
    WEB = 7

    @staticmethod
    def from_string(s):
        return Platform(int(s))

    def __str__(self):
        return str(self.value)

    @staticmethod
    def _capitalize_first_letter(s):
        m = re.search(r'\w', s)
        if m is None:
            return s
        return s[: m.start()] + m.group().upper() + s[m.end() :]

    def get_descr_header(self):
        return self._capitalize_first_letter(_PLATFORM_DESCRIPTIONS[self])

    def get_descr_text(self):
        s = _PLATFORM_DESCRIPTIONS[self]
        return 'the ' + s

    def get_descr_text_capitalized(self):
        return self._capitalize_first_letter(self.get_descr_text())


_PLATFORM_DESCRIPTIONS = {
    Platform.MOBILE: '"mobile" web version (or an unrecognized mobile app)',
    Platform.IPHONE: 'official iPhone app',
    Platform.IPAD: 'official iPad app',
    Platform.ANDROID: 'official Android app',
    Platform.WINDOWS_PHONE: 'official Windows Phone app',
    Platform.WINDOWS10: 'official Windows 10 app',
    Platform.WEB: 'web version (or an unrecognized app)',
}