User:WindBOT/LinkClass

class link:
	def __init__(self, content):
		content = u(content)
		self.setBody(content)
		self.setType(u'unknown')
		self.joined = False
		if len(content) > 2:
			if content[:2] == u'[[' and content[-2:] == u']]':
				split = content[2:-2].split(u'|')
				if len(split) in (1, 2):
					self.setType(u'internal')
					lnk = split[0]
					if lnk.find(u':') == -1:
						lnk = lnk.replace(u'_', u' ')
					self.setLink(lnk)
					if len(split) == 2:
						self.setLabel(split[1])
					else:
						self.setLabel(split[0])
						self.joined = True
			elif content[0] == u'[' and content[-1] == u']':
				split = content[1:-1].split(u' ', 1)
				self.setType(u'external')
				self.setLink(split[0])
				if len(split) == 2:
					self.setLabel(split[1])
				else:
					self.setLabel(None)
	def getType(self):
		return u(self.kind)
	def getBody(self):
		return u(self.body)
	def getLink(self):
		return u(self.link)
	def getLabel(self):
		if self.label is None:
			return None
		if self.joined:
			return self.getLink()
		return u(self.label)
	def setType(self, kind):
		self.kind = u(kind)
	def setBody(self, body):
		self.body = u(body)
	def setLink(self, link):
		self.link = u(link)
		if self.joined:
			self.label = u(link)
	def setLabel(self, label):
		if label is None:
			self.label = None
		else:
			self.label = u(label)
		if self.joined:
			self.link = u(label)
	def __str__(self):
		return self.__unicode__()
	def __repr__(self):
		return self.__unicode__().__repr__()
	def __unicode__(self):
		if self.getType() == u'internal':
			if self.getLabel() in (self.getLink().replace(u'_', u' '), self.getLink()):
				return u'[[' + self.getLabel() + u']]'
			return u'[[' + self.getLink() + u'|' + self.getLabel() + u']]'
		if self.getType() == u'external':
			if self.getLabel() is None:
				return u'[' + self.getLink() + u']'
			return u'[' + self.getLink() + u' ' + self.getLabel() + u']'
		return self.getBody()