Welcome To Our Shell

Mister Spy & Souheyl Bypass Shell

from .clean_html_text import clean_html_text
from bs4 import BeautifulSoup, Tag

class Link:
	def __init__(self, element=None, url=None, text=False, base=None):
		if element is not None:
			a = element.find('a')
			if a:
				element = a

		if url is None:
			if element is None:
				raise ValueError('either url or element should be provided!')
			self._url = self.get_url(element=element, base=base)
		else:
			self._url = url
		self._text = text or self.get_text(element=element) or self._url

	def __getstate__(self):
		return self._url, self._text

	def __setstate__(self, state):
		self._url, self._text = state

	def __hashkey__(self):
		return (repr(self.url))

	def __eq__(self, other):
		if isinstance(other, self.__class__):
			return self.url == other.url
		else:
			return self.url == other

	def __str__(self):
		return f'{self.text} {self.url} '

	def __repr__(self):
		return str(self)

	def is_anchor(self):
		return self._url.startswith('#')

	@property
	def text(self):
		return self._text

	@property
	def url(self):
		return self._url

	@staticmethod
	def get_url(element, base=None):
		try:
			href = element['href']
		except KeyError:
			raise AttributeError('element does not have a url!')

		if href.startswith('http://') or href.startswith('https://') or base is None or href.startswith('#'):
			return href
		elif href.startswith('/'):
			if base is None:
				raise TypeError('no base is provided!')
			else:
				return base + href

	@staticmethod
	def get_text(element):
		if element is None:
			return None
		try:
			return clean_html_text(html=element, replace_images=True)
		except AttributeError:
			return None

def parse_link(element, base=None):
	if element is None:
		return None
	try:
		link = Link(element=element, base=base)
		if link.is_anchor():
			return clean_html_text(html=element, replace_images=True)
		else:
			return link
	except AttributeError:
		return clean_html_text(html=element, replace_images=True)

def find_links(elements, base):
	if isinstance(elements, (BeautifulSoup, Tag)):
		if elements.name == 'a':
			links = [elements]
		else:
			links = elements.find_all(name='a')

		if links is not None:
			result = [parse_link(element=link, base=base) for link in links]
			return [x for x in result if x]
		else:
			return []

	elif isinstance(elements, str):
		return []
	else:
		return [link for element in elements for link in find_links(elements=element, base=base)]