164 lines
7.3 KiB
Python
164 lines
7.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" A profile viewer and editor for VK user objects."""
|
|
from __future__ import unicode_literals
|
|
from future import standard_library
|
|
standard_library.install_aliases()
|
|
import six
|
|
import webbrowser
|
|
import logging
|
|
import arrow
|
|
import requests
|
|
import languageHandler
|
|
import widgetUtils
|
|
import output
|
|
import wx
|
|
from wxUI.dialogs import urlList, profiles
|
|
from sessionmanager import utils
|
|
|
|
log = logging.getLogger("controller.profiles")
|
|
|
|
class userProfile(object):
|
|
""" Main controller to view an user profile. This controller will retrieve needed data from the VK website and display it appropiately."""
|
|
|
|
def __init__(self, session, user_id):
|
|
""" Default constructor:
|
|
@session vk.session: The main session object, capable of calling VK methods.
|
|
@user_id integer: User ID to retrieve information of.
|
|
At the current time, only users (and not communities) are supported.
|
|
"""
|
|
# self.person will hold a reference to the user object when retrieved from VK.
|
|
self.person = None
|
|
self.session = session
|
|
self.user_id = user_id
|
|
self.dialog = profiles.userProfile(title=_("Profile"))
|
|
self.dialog.create_controls("main_info")
|
|
self.dialog.realice()
|
|
self.get_basic_information()
|
|
if self.person != None:
|
|
self.dialog.get_response()
|
|
|
|
def get_basic_information(self):
|
|
""" Gets and inserts basic user information.
|
|
See https://vk.com/dev/users.get"""
|
|
# List of fields (information) to retrieve. For a list of fields available for user objects,
|
|
# see https://vk.com/dev/fields
|
|
fields = "first_name, last_name, bdate, city, country, home_town, photo_200_orig, online, site, status, last_seen, occupation, relation, relatives, personal, connections, activities, interests, music, movies, tv, books, games, about, quotes, can_write_private_message"
|
|
# ToDo: this method supports multiple user IDS, I'm not sure if this may be of any help for profile viewer.
|
|
person = self.session.vk.client.users.get(user_ids=self.user_id, fields=fields)
|
|
# If VK does not return anything it is very likely we have found a community.
|
|
if len(person) == 0:
|
|
return output.speak(_("Information for groups is not supported, yet."))
|
|
person = person[0]
|
|
# toDo: remove this print when I will be done with creation of profile viewer logic.
|
|
print(person)
|
|
# From this part we will format data from VK so users will see it in the GUI control.
|
|
# Format full name.
|
|
n = "{0} {1}".format(person["first_name"], person["last_name"])
|
|
# Format birthdate.
|
|
if "bdate" in person and person["bdate"] != "":
|
|
self.dialog.main_info.enable("bdate")
|
|
# VK can display dd.mm or dd.mm.yyyy birthdates. So let's compare the string lenght to handle both cases accordingly.
|
|
if len(person["bdate"]) <= 5: # dd.mm
|
|
d = arrow.get(person["bdate"], "D.M")
|
|
self.dialog.main_info.set("bdate", d.format(_("MMMM D"), locale=languageHandler.curLang[:2]))
|
|
else: # mm.dd.yyyy
|
|
d = arrow.get(person["bdate"], "D.M.YYYY")
|
|
self.dialog.main_info.set("bdate", d.format(_("MMMM D, YYYY"), locale=languageHandler.curLang[:2]))
|
|
# Format current city and home town
|
|
city = ""
|
|
if "home_town" in person and person["home_town"] != "":
|
|
home_town = person["home_town"]
|
|
self.dialog.main_info.enable("home_town")
|
|
self.dialog.main_info.set("home_town", home_town)
|
|
if "city" in person and len(person["city"]) > 0:
|
|
city = person["city"]["title"]
|
|
if "country" in person and person["country"] != "":
|
|
if city != "":
|
|
city = city+", {0}".format(person["country"]["title"])
|
|
else:
|
|
city = person["country"]["title"]
|
|
self.dialog.main_info.enable("city")
|
|
self.dialog.main_info.set("city", city)
|
|
self.dialog.main_info.set("name", n)
|
|
self.dialog.SetTitle(_("{name}'s profile").format(name=n,))
|
|
# Format website (or websites, if there are multiple of them).
|
|
if "site" in person and person["site"] != "":
|
|
self.dialog.main_info.enable("website")
|
|
self.dialog.main_info.set("website", person["site"])
|
|
self.dialog.main_info.enable("go_site")
|
|
widgetUtils.connect_event(self.dialog.main_info.go_site, widgetUtils.BUTTON_PRESSED, self.visit_website)
|
|
# Format status message.
|
|
if "status" in person and person["status"] != "":
|
|
self.dialog.main_info.enable("status")
|
|
self.dialog.main_info.set("status", person["status"])
|
|
# Format occupation.
|
|
# toDo: Research in this field is needed. Sometimes it returns university information even if users have active work places.
|
|
if "occupation" in person and person["occupation"] != None:
|
|
if person["occupation"]["type"] == "work": c1 = _("Work ")
|
|
elif person["occupation"]["type"] == "school": c1 = _("Student ")
|
|
elif person["occupation"]["type"] == "university": c1 = _("Student ")
|
|
if "name" in person["occupation"] and person["occupation"]["name"] != "":
|
|
c2 = _("In {0}").format(person["occupation"]["name"],)
|
|
else:
|
|
c2 = ""
|
|
self.dialog.main_info.enable("occupation")
|
|
self.dialog.main_info.set("occupation", c1+c2)
|
|
# format relationship status.
|
|
# ToDo: When dating someone, the button associated to the information should point to the profile of the user.
|
|
if "relation" in person and person["relation"] != 0:
|
|
if person["relation"] == 1:
|
|
r = _("Single")
|
|
elif person["relation"] == 2:
|
|
if "relation_partner" in person:
|
|
r = _("Dating with {0} {1}").format(person["relation_partner"]["first_name"], person["relation_partner"]["last_name"])
|
|
else:
|
|
r = _("Dating")
|
|
elif person["relation"] == 3:
|
|
r = _("Engaged with {0} {1}").format(person["relation_partner"]["first_name"], person["relation_partner"]["last_name"])
|
|
elif person["relation"] == 4:
|
|
r = _("Married with {0} {1}").format(person["relation_partner"]["first_name"], person["relation_partner"]["last_name"])
|
|
elif person["relation"] == 5:
|
|
r = _("It's complicated")
|
|
elif person["relation"] == 6:
|
|
r = _("Actively searching")
|
|
elif person["relation"] == 7:
|
|
r = _("In love")
|
|
self.dialog.main_info.enable("relation")
|
|
self.dialog.main_info.relation.SetLabel(_("Relationship: ")+r)
|
|
# format last seen.
|
|
if "last_seen" in person and person["last_seen"] != False:
|
|
original_date = arrow.get(person["last_seen"]["time"])
|
|
# Translators: This is the date of last seen
|
|
last_seen = _("{0}").format(original_date.humanize(locale=languageHandler.curLang[:2]),)
|
|
self.dialog.main_info.enable("last_seen")
|
|
self.dialog.main_info.set("last_seen", last_seen)
|
|
self.person = person
|
|
# Adds photo to the dialog.
|
|
# ToDo: Need to ask if this has a visible effect in the dialog.
|
|
if "photo_200_orig" in person:
|
|
img = requests.get(person["photo_200_orig"])
|
|
image = wx.Image(stream=six.BytesIO(requests.get(person["photo_200_orig"]).content))
|
|
try:
|
|
self.dialog.image.SetBitmap(wx.Bitmap(image))
|
|
except ValueError:
|
|
return
|
|
self.dialog.panel.Layout()
|
|
|
|
def visit_website(self, *args, **kwargs):
|
|
""" Allows to visit an user's website. """
|
|
text = self.person["site"]
|
|
# Let's search for URLS with a regexp, as there are users with multiple websites in their profiles.
|
|
urls = utils.find_urls_in_text(text)
|
|
if len(urls) == 0:
|
|
output.speak(_("No URL addresses were detected."))
|
|
return
|
|
elif len(urls) == 1:
|
|
selected_url = urls[0]
|
|
else:
|
|
dialog = urlList.urlList()
|
|
dialog.populate_list(urls)
|
|
if dialog.get_response() != widgetUtils.OK:
|
|
return
|
|
selected_url = urls[dialog.get_item()]
|
|
output.speak(_("Opening URL..."))
|
|
webbrowser.open_new_tab(selected_url) |