import pyperclip
import requests
from bs4 import BeautifulSoup
import sqlite3

def main():
    # 타오바오에 접속
    url = "https://www.taobao.com"
    response = requests.get(url)
    soup = BeautifulSoup(response.text, "html.parser")

    # 클립보드에 있는 이미지를 가져와서 붙여넣기
    image_url = pyperclip.paste()
    image_input = soup.find("input", {"id": "J_SiteSearch"})
    image_input.set_attribute("value", image_url)
    image_input.submit()

    # 검색 결과를 가져오기
    search_results = soup.find_all("div", {"class": "item J_MouserOnverReq"})

    # 검색 결과를 db에 저장하기
    connection = sqlite3.connect("taobao_search_results.db")
    cursor = connection.cursor()

    for result in search_results:
        title = result.find("div", {"class": "title"}).text.strip()  # 공백 제거
        price = result.find("div", {"class": "price"}).text.strip()  # 공백 제거
        link = result.find("a", {"class": "J_ItemPic ALink"})["href"]  # 상품 링크 가져오기
        cursor.execute("INSERT INTO search_results (title, price, link) VALUES (?, ?, ?)", (title, price, link))

    connection.commit()
    connection.close()

    # 검색 결과 상품들의 url을 보여주기
    for result in search_results:
        link = result.find("a", {"class": "J_ItemPic ALink"})["href"]
        print(link)

if __name__ == "__main__":
    main()