92 lines
2.3 KiB
Python
92 lines
2.3 KiB
Python
import json
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
import aiohttp
|
|
from bs4 import BeautifulSoup
|
|
|
|
from surugaya_api.consts import SURU_COOKIE_STRING
|
|
|
|
import inspect
|
|
|
|
def get_or_create_aiohttp_session(func):
|
|
async def wrapper(*args, **kwargs):
|
|
sig = inspect.signature(func)
|
|
bound_args = sig.bind(*args, **kwargs)
|
|
bound_args.apply_defaults()
|
|
|
|
aiohttp_session = bound_args.arguments.get("aiohttp_session")
|
|
is_created = False
|
|
|
|
if not aiohttp_session:
|
|
is_created = True
|
|
aiohttp_session = aiohttp.ClientSession()
|
|
bound_args.arguments["aiohttp_session"] = aiohttp_session
|
|
|
|
result = await func(*bound_args.arguments.values())
|
|
|
|
if is_created:
|
|
await aiohttp_session.close()
|
|
|
|
return result
|
|
|
|
return wrapper
|
|
@dataclass
|
|
class ProductStock:
|
|
condition: str
|
|
stock: str
|
|
price: int
|
|
on_sale_price: Optional[int]
|
|
|
|
@property
|
|
def is_on_sale(self) -> bool:
|
|
return self.on_sale_price is not None
|
|
|
|
@staticmethod
|
|
def from_item_price(item_price):
|
|
data: dict = json.loads(item_price.select_one("input").attrs["data-zaiko"])
|
|
|
|
return ProductStock(
|
|
condition=item_price.select_one("label").attrs["data-label"].strip(),
|
|
stock=data["zaiko"],
|
|
price=data["baika"],
|
|
on_sale_price=data.get("price_sale")
|
|
)
|
|
|
|
@dataclass
|
|
class Product:
|
|
id: int
|
|
name: str
|
|
main_image_href: str
|
|
stock: list[ProductStock]
|
|
|
|
@property
|
|
def in_stock(self):
|
|
return not self.stock
|
|
|
|
|
|
@get_or_create_aiohttp_session
|
|
async def load_product(product_id, aiohttp_session=None):
|
|
async with aiohttp_session.get(
|
|
url="https://www.suruga-ya.jp/product/detail/" + str(product_id),
|
|
headers={
|
|
"Cookie": SURU_COOKIE_STRING,
|
|
},
|
|
) as response:
|
|
page = await response.text()
|
|
page_bs = BeautifulSoup(page, features="html.parser")
|
|
|
|
product = Product(
|
|
id=int(product_id),
|
|
name=page_bs.select_one("#item_title").text.strip(),
|
|
main_image_href=page_bs.select_one(".main-pro-img").attrs["src"],
|
|
stock=[
|
|
ProductStock.from_item_price(item_price)
|
|
for item_price in page_bs.select(".item-price")
|
|
],
|
|
)
|
|
|
|
return product
|
|
|
|
|