textout/textoutpc/Tags/Image.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from urllib.parse import urlparse
from html import escape as _htmlescape
from .__base__ import *

__all__ = ["TextoutImageTag", "TextoutAdminImageTag"]

class TextoutImageTag(TextoutRawBlockTag):
	""" The main tag for displaying an image.
		Example uses:
	
		[img]picture_url[/img]
		[img=center]picture_url[/img]
		[img=12x24]picture_url[/img]
		[img=center|12x24]picture_url[/img]
		[img=x24|right]picture_url[/img]
	"""

	aliases = ('[img]',)

	def prepare(self, name, value):
		self.width = None
		self.height = None
		self.align = None

		for arg in ("", value)[value != None].split('|'):
			if   not arg:
				pass
			elif arg[0] in '0123456789x':
				dim = arg.split('x')
				try:    self.width = int(dim[0])
				except: pass
				try:    self.height = int(dim[1])
				except: pass
			elif arg in ('center', 'right', 'float-left', 'float-right'):
				self.align = arg

	def preprocess(self, content):
		for prefix in ('http://', 'https://', 'ftp://', '/'):
			if content.startswith(prefix):
				break
		else:
			raise Exception("No allowed prefix!")

	def process_text(self, content):
		return '![ {} ]!'.format(content)

	def process_html(self, content):
		style = []
		if   self.width:
			style.append('width: {}px'.format(self.width))
		elif self.height:
			style.append('width: auto')
		if   self.height:
			style.append('height: {}px'.format(self.height))
		elif self.width:
			style.append('height: auto')

		return '<img src="{}"{}{}>'.format(self._geturl(_htmlescape(content)),
			' class="{}"'.format(self.align) if self.align else '',
			' style="{}"'.format('; '.join(style)) if style else '')

class TextoutAdminImageTag(TextoutImageTag):
	""" This tag is special for Planète Casio, as it takes images from
		the `ad`ministration's image folder.
		It just adds this folder's prefix.
		Example uses:
		
		[adimg]some_picture.png[/img]
		[adimg=center]some_picture.png[/img]
		[adimg=12x24]some_picture.png[/img]
		[adimg=center|12x24]some_picture.png[/img]
		[adimg=x24|right]some_picture.png[/img]
	"""

	aliases = ('[adimg]',)

	def preprocess(self, content):
		# FIXME: check image URL!
		return 'https://www.planet-casio.com/images/ad/' + content

# End of file.