aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/vk
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2016-06-20 02:54:27 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2016-06-20 02:54:27 +0300
commiteeb81574135529fe0e7415d1028d0d76b085c93f (patch)
tree7d3bf0f1d1058b738e6bdf5dafec431c3057ca8b /vk
parentREADME update (diff)
downloadvk-scripts-eeb81574135529fe0e7415d1028d0d76b085c93f.tar.gz
vk-scripts-eeb81574135529fe0e7415d1028d0d76b085c93f.zip
implement more user fields
Diffstat (limited to 'vk')
-rw-r--r--vk/user.py68
1 files changed, 57 insertions, 11 deletions
diff --git a/vk/user.py b/vk/user.py
index 00a1674..08ce5bb 100644
--- a/vk/user.py
+++ b/vk/user.py
@@ -16,7 +16,7 @@ def _parse_last_seen(x):
else:
raise TypeError()
-def _parse_online_flag(x):
+def _parse_bool(x):
if isinstance(x, str):
if str(True) == x:
return True
@@ -27,17 +27,37 @@ def _parse_online_flag(x):
else:
return bool(x)
+def _parse_hidden(x):
+ return _parse_bool(x)
+
+def _parse_online_flag(x):
+ return _parse_bool(x)
+
class UserField(Enum):
UID = 'uid'
FIRST_NAME = 'first_name'
LAST_NAME = 'last_name'
- SCREEN_NAME = 'screen_name'
+ DEACTIVATED = 'deactivated'
+ HIDDEN = 'hidden'
+
+ DOMAIN = 'domain'
ONLINE = 'online'
LAST_SEEN = 'last_seen'
+ SCREEN_NAME = 'screen_name'
def __str__(self):
return self.value
+class DeactivationReason(Enum):
+ DELETED = 'deleted'
+ BANNED = 'banned'
+
+ def __str__(self):
+ return self.value
+
+def _parse_deactivated(s):
+ return DeactivationReason(s)
+
class User(Hashable, MutableMapping):
@staticmethod
def from_api_response(source):
@@ -83,6 +103,8 @@ class User(Hashable, MutableMapping):
_FIELD_PARSERS = {
UserField.UID: int,
+ UserField.DEACTIVATED: _parse_deactivated,
+ UserField.HIDDEN: _parse_hidden,
UserField.ONLINE: _parse_online_flag,
UserField.LAST_SEEN: _parse_last_seen,
}
@@ -107,17 +129,29 @@ class User(Hashable, MutableMapping):
def set_last_name(self, name):
self[UserField.LAST_NAME] = name
- def has_screen_name(self):
- return UserField.SCREEN_NAME in self
+ def is_deactivated(self):
+ return UserField.DEACTIVATED in self
- def get_screen_name(self):
- if self.has_screen_name():
- return self[UserField.SCREEN_NAME]
- else:
- return 'id' + str(self.get_uid())
+ def get_deactivation_reason(self):
+ return self[UserField.DEACTIVATED]
- def set_screen_name(self, name):
- self[UserField.SCREEN_NAME] = name
+ def set_deactivated(self, reason):
+ self[UserField.DEACTIVATED] = reason
+
+ def is_hidden(self):
+ return UserField.HIDDEN in self and self[UserField.HIDDEN]
+
+ def set_hidden(self, value=True):
+ self[UserField.HIDDEN] = value
+
+ def has_domain(self):
+ return UserField.DOMAIN in self
+
+ def get_domain(self):
+ return self[UserField.DOMAIN]
+
+ def set_domain(self, domain):
+ self[UserField.DOMAIN] = domain
def has_online_flag(self):
return UserField.ONLINE in self
@@ -148,3 +182,15 @@ class User(Hashable, MutableMapping):
def get_last_seen_platform(self):
return self[UserField.LAST_SEEN].get_platform()
+
+ def has_screen_name(self):
+ return UserField.SCREEN_NAME in self
+
+ def get_screen_name(self):
+ if self.has_screen_name():
+ return self[UserField.SCREEN_NAME]
+ else:
+ return 'id' + str(self.get_uid())
+
+ def set_screen_name(self, name):
+ self[UserField.SCREEN_NAME] = name