import os
import requests
import bs4
from lxml import etree
import csv


class Functional:
    img = "/html/body/div/div[1]/div/main/header/img"
    word = "/html/body/div/div[1]/div/main/div[1]/div[1]/div[2]/b"
    translate = "/html/body/div/div[1]/div/main/div[1]/div[1]/div[1]/span"
    transcription = "/html/body/div/div[1]/div/main/div[1]/div[2]/div[2]/b"
    headers = {'Content-Type': 'text/html'}
    BASEDIR = os.path.abspath(os.path.dirname(__file__))

    def __init__(self, link, path_to_folder=BASEDIR) -> None:
        self.response = requests.get(link, headers=self.headers)
        self.path_to_folder = path_to_folder

    def get_data(self):
        result = []
        soup = bs4.BeautifulSoup(self.response.text, features="lxml")
        dom = etree.HTML(str(soup))

        word = dom.xpath(self.word)[0].text
        translate = dom.xpath(self.translate)[0].text
        transcription = dom.xpath(self.transcription)[0].text
        img = soup.find("img")["src"]

        result.append(
            [word, transcription, translate]
        )

        Functional.save_data(self, result)
        # Functional.save_img(self, img)

    def save_data(self, result):
        with open("test.csv", "r", encoding="utf-8") as file:
            if not file.read():
                print(False)
                with open("test.csv", "w", encoding="utf-8") as file:
                    writer = csv.writer(file)

                    writer.writerow(
                        (
                            "Слово",
                            "Транскрипция",
                            "Перевод"
                        )
                    )

                    writer.writerows(
                        result
                    )

            else:
                print(True)
                with open("test.csv", "a", encoding="utf-8") as file:
                    writer = csv.writer(file)

                    writer.writerows(
                        result
                    )


if __name__ == "__main__":
    a = Functional(link="https://www.toboapp.com/word/korean/russian/kokkiri-846/")
    a.get_data()

class Functional:
    img = "/html/body/div/div[1]/div/main/header/img"
    word = "/html/body/div/div[1]/div/main/div[1]/div[1]/div[2]/b"
    translate = "/html/body/div/div[1]/div/main/div[1]/div[1]/div[1]/span"
    transcription = "/html/body/div/div[1]/div/main/div[1]/div[2]/div[2]/b"
    headers = {'Content-Type': 'text/html'}
    BASEDIR = os.path.abspath(os.path.dirname(__file__))

    def __init__(self, link, img_name, path_to_folder=BASEDIR) -> None:
        self.response = requests.get(link, headers=self.headers)
        self.path_to_folder = path_to_folder
        self.img_name = img_name

    def get_data(self):
        result = []
        soup = bs4.BeautifulSoup(self.response.text, features="lxml")
        dom = etree.HTML(str(soup))

        word = dom.xpath(self.word)[0].text
        translate = dom.xpath(self.translate)[0].text
        transcription = dom.xpath(self.transcription)[0].text
        element = dom.xpath(self.img)[0]
        img = etree.tostring(element, encoding='utf-8').decode('utf-8')

        s = bs4.BeautifulSoup(img, 'html.parser')

        img = s.img['src']

        result.append(
            [word, transcription, translate]
        )

        Functional.save_data(self, result)
        Functional.save_img(self, img)

    def save_data(self, result):
        with open("test.csv", "r", encoding="utf-8") as file:
            if not file.read():
                print(False)
                with open("test.csv", "w", encoding="utf-8") as file:
                    writer = csv.writer(file)

                    writer.writerow(
                        (
                            "Слово",
                            "Транскрипция",
                            "Перевод"
                        )
                    )

                    writer.writerows(
                        result
                    )

            else:
                print(True)
                with open("test.csv", "a", encoding="utf-8") as file:
                    writer = csv.writer(file)

                    writer.writerows(
                        result
                    )

    def save_img(self, src_img):
        response = requests.get(src_img, stream=True)
        with open(f"{self.img_name}.svg", 'wb') as file:
            file.write(response.content)