Skip to content

Instantly share code, notes, and snippets.

Created Jan 13, 2016
Embed
What would you like to do?
appex.py
from _appex import finish, is_running_extension, get_input
import _appex
def _path2url(path):
import urlparse
import urllib
return urlparse.urljoin('file:', urllib.pathname2url(path))
def get_attachments(uti='public.data'):
output_attachments = []
input_items = get_input()
for item in input_items:
attachments = item.get('attachments', None)
if attachments:
for attachment_dict in attachments:
for attachment_type in attachment_dict:
if _appex._uti_conforms(attachment_type, uti):
output_attachments.append(attachment_dict[attachment_type])
return output_attachments
def _image_from_attachment(image_attachment, image_type='pil', raw_data=False):
image_type = image_type.lower()
if not raw_data and image_type not in ['pil', 'ui']:
raise TypeError('Unsupported image_type')
use_ui = (image_type != 'pil')
if isinstance(image_attachment, str):
# raw image data
if raw_data:
return raw_data
try:
if use_ui:
import ui
return ui.Image.from_data(image_attachment)
else:
import Image
from io import BytesIO
buffer = BytesIO(image_attachment)
img = Image.open(buffer)
return img
except:
return None
elif isinstance(image_attachment, unicode):
# image path
if raw_data:
with open(image_attachment, 'r') as f:
return f.read()
try:
if use_ui:
import ui
return ui.Image.named(image_attachment)
else:
import Image
return Image.open(image_attachment)
except:
return None
def get_images(image_type='pil'):
image_attachments = get_attachments('public.image')
images = []
if image_attachments:
for image_attachment in image_attachments:
img = _image_from_attachment(image_attachment, image_type)
if img:
images.append(img)
return images
def get_image(image_type='pil'):
image_attachments = get_attachments('public.image')
if image_attachments:
return _image_from_attachment(image_attachments[0], image_type)
return None
def get_image_data():
image_attachments = get_attachments('public.image')
if image_attachments:
return _image_from_attachment(image_attachments[0], raw_data=True)
return None
def get_images_data():
image_attachments = get_attachments('public.image')
images = []
if image_attachments:
for image_attachment in image_attachments:
data = _image_from_attachment(image_attachment, raw_data=True)
if data:
images.append(data)
return images
def get_text():
text_attachments = get_attachments('public.text')
if text_attachments:
return '\n'.join(text_attachments)
return None
def get_urls():
url_attachments = get_attachments('public.url')
urls = []
if url_attachments:
for url_attachment in url_attachments:
if url_attachment.startswith('/'):
url = _path2url(url_attachment)
urls.append(url)
else:
urls.append(url_attachment)
return urls
def get_url():
url_attachments = get_attachments('public.url')
if url_attachments:
url = url_attachments[0]
if url.startswith('/'):
return _path2url(url)
return url
return None
def get_file_paths():
url_attachments = get_attachments('public.url')
paths = []
if url_attachments:
for url_attachment in url_attachments:
if url_attachment.startswith('/'):
paths.append(url_attachment)
return paths
def get_file_path():
url_attachments = get_attachments('public.url')
if url_attachments:
first_path = url_attachments[0]
if first_path.startswith('/'):
return first_path
text = get_text()
if text and text.startswith('/') and os.path.exists(text):
return text
return None
def get_vcards():
vcard_attachments = get_attachments('public.vcard')
if vcard_attachments:
return vcard_attachments
return None
def get_vcard():
vcard_attachments = get_attachments('public.vcard')
if vcard_attachments:
return vcard_attachments[0]
return None
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment