99 lines
4.6 KiB
Python
99 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" Attachment upload methods for different kind of posts in VK. This should become the framework for posting attachment files to the social network."""
|
|
import os
|
|
import logging
|
|
import widgetUtils
|
|
from wxUI.dialogs import attach as gui
|
|
from wxUI.dialogs import selector
|
|
log = logging.getLogger("controller.attach")
|
|
|
|
class attachFromLocal(object):
|
|
""" Controller used in some sections of the application, mainly for uploading photos from the local computer to VK.
|
|
This controller will not upload the contents by itself, but will generate the data structures for being send over the VK API.
|
|
At the current time, only photo uploading is supported."""
|
|
|
|
def __init__(self, voice_messages=False):
|
|
""" Constructor.
|
|
@voice_messages bool: If True, will add a button for sending voice messages. Functionality for this button has not been added yet.
|
|
"""
|
|
# Self.attachments will hold a reference to all attachments added to the dialog.
|
|
self.attachments = list()
|
|
# Type is just a reference, as there will be local and online based attachments.
|
|
self.type = "local"
|
|
self.dialog = gui.attachDialog(voice_messages)
|
|
widgetUtils.connect_event(self.dialog.photo, widgetUtils.BUTTON_PRESSED, self.upload_image)
|
|
widgetUtils.connect_event(self.dialog.remove, widgetUtils.BUTTON_PRESSED, self.remove_attachment)
|
|
self.dialog.get_response()
|
|
log.debug("Attachments controller started.")
|
|
|
|
def upload_image(self, *args, **kwargs):
|
|
""" allows uploading an image from the computer. In theory
|
|
"""
|
|
image, description = self.dialog.get_image()
|
|
if image != None:
|
|
# Define data structure for this attachment, as will be required by VK API later.
|
|
imageInfo = {"type": "photo", "file": image, "description": description}
|
|
log.debug("Image data to upload: %r" % (imageInfo,))
|
|
self.attachments.append(imageInfo)
|
|
# Translators: This is the text displayed in the attachments dialog, when the user adds a photo.
|
|
info = [_(u"Photo"), os.path.basename(image)]
|
|
self.dialog.attachments.insert_item(False, *info)
|
|
self.dialog.remove.Enable(True)
|
|
|
|
def remove_attachment(self, *args, **kwargs):
|
|
""" Remove the currently focused item from the attachments list."""
|
|
current_item = self.dialog.attachments.get_selected()
|
|
log.debug("Removing item %d" % (current_item,))
|
|
if current_item == -1: current_item = 0
|
|
self.attachments.pop(current_item)
|
|
self.dialog.attachments.remove_item(current_item)
|
|
self.check_remove_status()
|
|
log.debug("Removed")
|
|
|
|
def check_remove_status(self):
|
|
""" Checks whether the remove button should remain enabled."""
|
|
if len(self.attachments) == 0 and self.dialog.attachments.get_count() == 0:
|
|
self.dialog.remove.Enable(False)
|
|
|
|
class attachFromOnline(object):
|
|
""" this was the previously working audio attachment uploader. As VK has deprecated their Audio API for third party clients, this class will not work.
|
|
However, I have decided to keep this here so in future it may be modified to attach different kind of online based attachments, such as posted photos, videos, etc.
|
|
"""
|
|
|
|
def __init__(self, session):
|
|
""" Default constructor.
|
|
@session vk.session: An VK session, capable of calling methods from the VK API.
|
|
"""
|
|
self.session = session
|
|
# Self.attachments will hold a reference to all attachments added to the dialog.
|
|
self.attachments = list()
|
|
# Define type as online.
|
|
self.type = "online"
|
|
self.dialog = gui.attachDialog()
|
|
widgetUtils.connect_event(self.dialog.audio, widgetUtils.BUTTON_PRESSED, self.add_audio)
|
|
# widgetUtils.connect_event(self.dialog.remove, widgetUtils.BUTTON_PRESSED, self.remove_attachment)
|
|
self.dialog.get_response()
|
|
log.debug("Attachments controller started.")
|
|
|
|
def add_audio(self, *args, **kwargs):
|
|
""" Allow adding an audio directly from the user's audio library."""
|
|
list_of_audios = self.session.vk.client_audio.get()
|
|
# list_of_audios = list_of_audios["items"]
|
|
audios = []
|
|
for i in list_of_audios:
|
|
audios.append(u"{0}, {1}".format(i["title"], i["artist"]))
|
|
select = selector.selectAttachment(_(u"Select the audio files you want to send"), audios)
|
|
if select.get_response() == widgetUtils.OK and select.attachments.GetCount() > 0:
|
|
attachments = select.get_all_attachments()
|
|
for i in attachments:
|
|
list_of_audios[i]["type"] = "audio"
|
|
self.attachments.append(list_of_audios[i])
|
|
|
|
def parse_attachments(self):
|
|
""" Retrieve all attachments and convert them to data structures required by VK API."""
|
|
result = ""
|
|
for i in self.attachments:
|
|
# Define data structures required by VK API.
|
|
preresult = "{0}{1}_{2}".format(i["type"], i["owner_id"], i["id"])
|
|
result = preresult+","
|
|
return result |