from bs4 import BeautifulSoup
import requests

HEADERS = {"User-Agent": 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.159 Safari/537.36 Vivaldi/4.1.2369.21'}


class IKEA:
	def __init__(self, url):
		self.url = 'https://www.ikea.lv/en/products/' + url + '?&&page=1&order=PRICEASC'

	def _get_paige_amount(self):
		page_amount = 1
		page = requests.get(self.url, headers=HEADERS)
		soup = BeautifulSoup(page.content, 'html.parser')

		# getting max page amount
		for el in soup.find_all(class_='page-link'):
			cropped_name = el.get_text().strip()
			if cropped_name.isnumeric():
				cropped_name = int(cropped_name)
				if cropped_name > page_amount:
					page_amount = cropped_name

		return page_amount

	def get_data(self):
		prices = []
		names = []
		prices.clear()
		names.clear()
		# combined_list.clear()
		position = self.url.find('page=') + 5
		for i in range(1, self._get_paige_amount() + 1):
			url = self.url[:position] + str(i) + self.url[position + 1:]

			page = requests.get(url, headers=HEADERS)
			soup = BeautifulSoup(page.content, 'html.parser')

			# getting product name
			for el in soup.find_all(class_='display-7 mr-2'):
				cropped_name = el.get_text().strip()
				names.append(cropped_name)

			# getting product price
			for el in soup.find_all(class_='display-6'):
				cropped_price = el.get_text().strip()
				prices.append(cropped_price[:cropped_price.find("€") + 1])

		combined_list = [i + " - " + j for i, j in zip(names, prices)]
		if __name__ == '__main__':
			SEPARATOR = "\n"
		else:
			SEPARATOR = "<br>"

		output = SEPARATOR.join(str(elem) for elem in combined_list)
		return output


arm_chairs = IKEA('ikea-for-business/office/armchairs')
bathroom_furniture = IKEA('bathroom/vanity-units/bathroom-furniture')
bathroom_lighting = IKEA('bathroom/bathroom-lighting/bathroom-lighting')
bed_frames = IKEA('bedroom/beds-and-sofa-beds/bed-frames')
bookcases = IKEA('living-room/bookcases/bookcases')
boxes_and_baskets = IKEA('bedroom/sorting-solutions/boxes-and-baskets')
candles = IKEA('kitchen/kitchen-decoration/candles')
ceiling_lamps_and_spotlights = IKEA('decoration/lighting/ceiling-lamps-and-spotlights')
chairs_and_benches = IKEA('dining-room/dining-seating/chairs-and-benches')
chest_of_drawers = IKEA('bedroom/chest-of-drawers-other-furniture/chest-of-drawers')
children_storage_furniture = IKEA('children-s-room/children-3-7/children-s-storage-furniture')
curtains = IKEA('kitchen/curtains-blinds-and-fabrics/curtains')
day_beds = IKEA('bedroom/beds-and-sofa-beds/day-beds')
dining_tables = IKEA('dining-room/dining-tables/dining-tables')
dinnerware_and_serving = IKEA('kitchen/cookware-and-dinnerware/dinnerware-and-serving')
glasses = IKEA('kitchen/cookware-and-dinnerware/glasses')
home_desks = IKEA('home-office/desks/home-desks')
interior_organisers = IKEA('home-office/wardrobes/interior-organisers')
kitchen_interior_organisers = IKEA('kitchen/kitchen-interior-organisers/kitchen-interior-organisers')
light_bulbs = IKEA('bedroom/bedroom-lighting/light-bulbs')
mattresses = IKEA('bedroom/mattresses/mattresses')
mirrors = IKEA('kitchen/kitchen-decoration/mirrors')
office_chairs = IKEA('home-office/work-seating-range/office-chairs')
office_desks_and_tables = IKEA('home-office/desks/office-desks-and-tables')
open_shelving_units = IKEA('living-room/shelving-units-systems/open-shelving-units')
pax_wardrobes = IKEA('ikea-for-business/retail/system-wardrobes')
pendant_lamps = IKEA('ikea-for-business/retail/pendant-lamps')
pillows = IKEA('bedroom/bedding/pillows')
pots = IKEA('kitchen/cookware-and-dinnerware/pots')
quilt_covers_and_pillow_cases = IKEA('bedroom/bedding/quilt-covers-and-pillow-cases')
quilts = IKEA('bedroom/bedding/quilts')
rugs = IKEA('living-room/home-furnishing-rugs/rugs')
sheets_and_pillow_cases = IKEA('bedroom/bedding/sheets-and-pillow-cases')
sofa_beds_and_chair_beds = IKEA('bedroom/beds-and-sofa-beds/sofa-beds-chair-beds')
sofa_tables = IKEA('living-room/coffee-side-tables/sofa-tables')
solitaire_cabinets = IKEA('living-room/solitaire-cabinets/solitaire-cabinets')
solitaire_wardrobes = IKEA('bedroom/wardrobes/solitaire-wardrobes')
system_cabinets = IKEA('living-room/solitaire-cabinets/system-cabinets')
table_lamps = IKEA('bedroom/bedroom-lighting/table-lamps')
towels = IKEA('bathroom/towels/towels')
toys_for_small_children = IKEA('children-s-room/children-3-7/toys-for-small-children')
tv_benches = IKEA('living-room/tv-stands-media-units/tv-benches')


def main():
	# import cProfile
	# import pstats
	# with cProfile.Profile() as pr:
	# 	tv_benches.get_data()

	# stats = pstats.Stats(pr)
	# stats.sort_stats(pstats.SortKey.TIME)
	# # stats.print_stats()
	# stats.dump_stats(filename='stats.prof')
	print(tv_benches.get_data())


if __name__ == '__main__':
	main()