craft_parser_init

This commit is contained in:
2026-01-19 22:17:38 +07:00
parent 4e95139bb5
commit c083fc26ad
9 changed files with 978 additions and 0 deletions

5
.gitignore vendored Normal file
View File

@ -0,0 +1,5 @@
*.xlsx
*.csv
.mypy_cache
__pycache__
*.log

26
craft_parser/craft.py Normal file
View File

@ -0,0 +1,26 @@
CRAFT_TYPES = []
class CraftItem:
def __init__(self, name: str, img_url: str, img_shift: tuple[int, int], is_base=False, ):
self.name = name
self.is_base = is_base
self.img_url = img_url
self.img_shift = img_shift
class CraftRecipe:
def __init__(self, output_item_name: str, output_count: int, craft_type: str):
self.output_item_name = output_item_name
self.output_count = output_count
self.craft_type = craft_type
self.recipe_id = None # будет назначен при сохранении
class CraftComponent:
def __init__(self, recipe_id: int, input_item: str, count: int):
self.recipe_id = recipe_id
self.input_item = input_item
self.count = count

View File

@ -0,0 +1,71 @@
import sys
import time
from bs4 import BeautifulSoup
import requests
from craft_storage import CraftStorage
import device_extractor
import simple_extractor
from loguru import logger
class CraftFacade:
def __init__(self):
self.visited: set[str] = set()
logger.add('file_{time}.log')
logger.add(sys.stdout, colorize=True, format="<green>{time}</green> <level>{message}</level>")
# ---------- low level ----------
def _get_page_content(self, url: str) -> str | None:
time.sleep(0.5)
try:
r = requests.get(url)
r.raise_for_status()
return r.text
except requests.RequestException as e:
logger.error(f'Ошибка загрузки {url}: {e}')
return None
def _get_soup(self, html: str) -> BeautifulSoup:
return BeautifulSoup(html, "html.parser")
# ---------- public API ----------
def extract(self, url: str, storage: CraftStorage, recursive: bool = True):
if not url:
logger.error('Битая ссылка')
return
if url in self.visited:
logger.debug(f'Уже посещали: {url}')
return
html = self._get_page_content(url)
if not html:
return
soup = self._get_soup(html)
logger.info(f'Анализирую {url}')
self.visited.add(url)
all_links: set[str] = set()
try:
links = simple_extractor.extract_crafts(soup, storage)
all_links.update(links)
links = device_extractor.extract_crafts(soup, storage)
all_links.update(all_links)
except Exception as e:
logger.error(f'Ошибка в парсере: {e}')
logger.info(f'Готово: {url}')
if not recursive:
return
logger.info('Начинаю обход внешних ссылок...')
for link in all_links:
self.extract(link, storage, recursive=True)

View File

@ -0,0 +1,122 @@
import os
import pandas as pd
from craft import CraftComponent, CraftItem, CraftRecipe
from loguru import logger
class CraftStorage:
def __init__(self):
# items
self.items_df = pd.DataFrame(columns=[
"name", "img_url", "is_base", "img_shift"
])
##Чтобы на конкат не орал
self.items_df["is_base"] = self.items_df["is_base"].astype('bool')
# recipes
self.recipes_df = pd.DataFrame(columns=[
"recipe_id", "output_item_name", "output_count", "craft_type"
])
# components
self.components_df = pd.DataFrame(columns=[
"recipe_id", "input_item", "count"
])
self.recipe_signatures = set()
self._recipe_id_seq = 1
def try_add_recipe_signature(self, output_item_name: str, components: list[CraftComponent], craft_type: str) -> bool:
# сортируем, чтобы порядок не влиял
parts = sorted(
(c.input_item, c.count) for c in components
)
sig = (output_item_name, tuple(parts), craft_type)
if sig in self.recipe_signatures:
logger.warning(f'Дубликат рецепта {output_item_name}. Пропускаю ')
#print(f'CRAFT: {}')
# for el in components:
# print(f'{el.input_item} : {el.count}')
# print('Пропускаю')
# print('....')
return False
self.recipe_signatures.add(sig)
return True
def add_item(self, item: CraftItem):
# проверяем, есть ли уже такой item
if item.name in self.items_df["name"].values:
logger.warning(f'{item.name} уже есть в датасете, скипаю')
return # уже есть, не дублируем
row = {
"name": item.name,
"img_url": item.img_url,
"is_base": bool(item.is_base),
"img_shift": item.img_shift
}
self.items_df = pd.concat(
[self.items_df, pd.DataFrame([row])],
ignore_index=True
)
def add_recipe(self, recipe: CraftRecipe) -> int:
recipe_id = self._recipe_id_seq
self._recipe_id_seq += 1
recipe.recipe_id = recipe_id
row = {
"recipe_id": recipe_id,
"output_item_name": recipe.output_item_name,
"output_count": recipe.output_count,
"craft_type": recipe.craft_type
}
self.recipes_df = pd.concat(
[self.recipes_df, pd.DataFrame([row])],
ignore_index=True
)
return recipe_id
def add_component(self, component: CraftComponent):
# защита от дубликатов
exists = (
(self.components_df["recipe_id"] == component.recipe_id) &
(self.components_df["input_item"] == component.input_item) &
(self.components_df['count'] == component.count)
).any()
if exists:
logger.warning(f"""
Уже существует такая запись для рецепта: {component.recipe_id} <--- {component.input_item}*{component.count}
Скипаем
""")
return
row = {
"recipe_id": component.recipe_id,
"input_item": component.input_item,
"count": component.count
}
self.components_df = pd.concat(
[self.components_df, pd.DataFrame([row])],
ignore_index=True
)
def save(self):
os.makedirs('data', exist_ok=True)
self.items_df.to_csv('data/items.csv', index=False)
self.recipes_df.to_csv('data/recipes.csv', index=False)
self.components_df.to_csv('data/recipe_components.csv', index=False)
self.items_df.to_excel('data/items.xlsx', index=False)
self.recipes_df.to_excel('data/recipes.xlsx', index=False)
self.components_df.to_excel('data/recipe_components.xlsx', index=False)

View File

@ -0,0 +1,261 @@
import re
from bs4 import BeautifulSoup
from craft import CraftComponent, CraftItem, CraftRecipe
from craft_storage import CraftStorage
from dto import ParsedItem, ParsedRecipeInput, SpriteData
from loguru import logger
def extract_crafts(soup: BeautifulSoup, storage: CraftStorage) -> set[str]:
craft_containers = soup.find_all("div", class_="craft-gui")
if(craft_containers.__len__() < 1):
logger.info('Машинных рецептов не найдено')
return set()
src_links : set[str] = set()
for container in craft_containers:
is_hidden = check_is_hidden(container)
if(is_hidden):
continue
output_item = parse_craft_item(container)
# Получаем список рецептов (может быть несколько для печи)
recipes_list = parse_craft_components_and_recipe(
container, output_item.item.name, output_item.amount
)
for recipe_input in recipes_list: # Обрабатываем каждый рецепт отдельно
if not storage.try_add_recipe_signature(
output_item.item.name,
recipe_input.components,
recipe_input.recipe.craft_type
):
continue # Дубликат
storage.add_item(output_item.item)
for input_item in recipe_input.items:
storage.add_item(input_item)
recipe_id = storage.add_recipe(recipe_input.recipe)
for component in recipe_input.components:
component.recipe_id = recipe_id
storage.add_component(component)
logger.info(f'Добавлен рецепт машинного крафта для {output_item.item.name} '
f'(ингредиент: {recipe_input.components[0].input_item})')
return src_links
def check_is_hidden(container : BeautifulSoup) -> bool:
if(container.find_parent('table', class_='collapsed')):
logger.warning(f'{container} Не будет исследован. Причина: помечен как скрытый')
return True
return False
def parse_craft_item(container) -> ParsedItem|None:
output_span = container.find('span', class_='gt-output')
if(output_span is None):
logger.error(f'ошибка для \n{container}\n: не найдено ячейки с результатом!')
return None
output_span_name_container = output_span.find('span', class_='invslot-item')
if(output_span_name_container is None):
logger.error(f'ошибка для \n{output_span}\n: не найден текстовый контейнер!')
return None
data_from_span = extract_data_from_sprite_span(output_span_name_container)
if(data_from_span is not None):
output_item_title = data_from_span.title
output_item_img_shift = data_from_span.shift
output_item_img_url = data_from_span.img_url
output_amount = data_from_span.amount
else:
data_from_img = extract_data_from_sprite_img(output_span_name_container)
if(data_from_img is None):
logger.error(f'ошибка для \n{output_span_name_container}\n: не удалось извлечь картинку и заголовок ни одним из способов')
return None
output_item_title = data_from_img.title
output_item_img_shift = data_from_img.shift
output_item_img_url = data_from_img.img_url
output_amount = 1
item = CraftItem(output_item_title, output_item_img_url, output_item_img_shift)
return ParsedItem(item=item, amount=output_amount)
def parse_craft_components_and_recipe(container, output_item_name: str, output_item_count: int) -> list[ParsedRecipeInput]:
input_span = container.find('span', class_='gt-input')
if not input_span:
logger.error(f'Ошибка: нет блока gt-input в {container}')
return []
recipe_info = extract_recipe_type(container)
if not recipe_info:
return []
is_furnace = 'Печь' in recipe_info
recipes = [] # Список рецептов
if is_furnace:
# Для печи каждый предмет в левом слоте — отдельный рецепт
left_cell = input_span.find('span', class_='invslot')
if left_cell:
item_spans = left_cell.find_all('span', class_='invslot-item')
for item_span in item_spans:
data = extract_data_from_sprite_span(item_span) or extract_data_from_sprite_img(item_span)
if data:
item = CraftItem(data.title, data.img_url, data.shift)
# Создаём отдельный рецепт для каждого ингредиента
recipe = CraftRecipe(output_item_name, output_item_count, recipe_info)
component = CraftComponent(-1, item.name, data.amount)
recipes.append(ParsedRecipeInput(
items=[item],
recipe=recipe,
components=[component],
source_links=[]
))
else:
# Для других машин — один рецепт со всеми ингредиентами
item_spans = input_span.find_all('span', class_='invslot-item')
items = []
components = []
for item_span in item_spans:
data = extract_data_from_sprite_span(item_span) or extract_data_from_sprite_img(item_span)
if data:
item = CraftItem(data.title, data.img_url, data.shift)
##skip ic2 stuff
if(item.name=='Энергия'):
continue
items.append(item)
components.append(CraftComponent(-1, item.name, data.amount))
recipe = CraftRecipe(output_item_name, output_item_count, recipe_info)
recipes.append(ParsedRecipeInput(
items=items,
recipe=recipe,
components=components,
source_links=[]
))
return recipes
def extract_recipe_type(container) -> str|None:
processor_container = container.find('span', class_='minetip')
if(processor_container is None):
logger.error(f'Ошибка для {container}: не найден тип машинного рецепта')
return None
return clean_recipe_type_str(processor_container.get('data-minetip-text'))
def clean_recipe_type_str(input: str):
pattern = r"\s*//&7Модификация:/.*$"
input = input.replace('&3','')
input = re.sub(pattern, "", input).strip()
return input
def extract_data_from_sprite_img(input_span_name_container) -> SpriteData|None:
input_item_img = input_span_name_container.find('img')
if(input_item_img is None):
return None
title = input_item_img.get('alt')
if(title is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найдено описание картинки!')
return None
url = input_item_img.get('src')
if(url is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найден url картинки!')
return None
return SpriteData(
title=title,
shift=(0, 0),
img_url=url,
source_link=None)
def extract_data_from_sprite_span(input_span_name_container) -> SpriteData|None:
input_item_sprite_span = input_span_name_container.find('span', class_='sprite')
#Контейнера со спрайтом может не быть!
if(input_item_sprite_span is None):
return None
input_item_title = input_item_sprite_span.get('title')
if(input_item_title is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найдено заголовка предмета')
return None
amount_container = input_span_name_container.find('span', class_='invslot-stacksize')
output_amount = 1
if(amount_container is not None):
logger.info(f'Для объекта {input_item_title} найдено количество : {amount_container.text}')
output_amount = int(amount_container.text)
item_source_link = get_item_source_link(input_span_name_container)
input_item_img_url = extract_img_classname(input_item_sprite_span)
if(input_item_img_url is None):
logger.error(f'Ошибка для \n{input_item_img_url}\n: не найден файл спрайта')
return None
input_item_img_shift = extract_img_shift(input_item_sprite_span)
if(input_item_img_shift is None):
logger.error(f'Ошибка для \n{input_item_sprite_span}\n: не найдено смещения для спрайта')
return None
return SpriteData(
title=input_item_title,
shift=input_item_img_shift,
img_url=input_item_img_url,
source_link=item_source_link,
amount=output_amount
)
def extract_img_shift(output_item_sprite_span):
if(output_item_sprite_span is None):
return None
css_value = output_item_sprite_span.get('style')
if(css_value is None):
return None
# Шаблон: ищем два числа (с возможным минусом) перед 'px'
pattern = r'(-?\d+)px\s+(-?\d+)px'
match = re.search(pattern, css_value)
if match:
x = int(match.group(1))
y = int(match.group(2))
return (x, y)
else:
return None # Если шаблон не найден
def extract_img_classname(sprite_span):
if(sprite_span is None):
return None
'sprite industrialcraft-2-inv-sprite'
classes = sprite_span.get('class', [])
for cls in classes:
if cls.endswith('-sprite'):
return cls
return None # Если не найдено
def get_item_source_link(input_span_name_container) -> str|None:
link_container = input_span_name_container.find('a')
if(link_container is None):
return None
link = link_container.get('href')
if(link is None):
return None
return f'https://ru.minecraft.wiki{link}'

23
craft_parser/dto.py Normal file
View File

@ -0,0 +1,23 @@
from dataclasses import dataclass
from typing import List, Optional, Tuple
from craft import CraftItem, CraftComponent, CraftRecipe
@dataclass
class ParsedItem:
item: CraftItem
amount: int
@dataclass
class ParsedRecipeInput:
items: List[CraftItem]
recipe: CraftRecipe
components: List[CraftComponent]
source_links: List[str]
@dataclass
class SpriteData:
amount: int
title: str
shift: Tuple[int, int]
img_url: str
source_link: Optional[str] = None

17
craft_parser/main.py Normal file
View File

@ -0,0 +1,17 @@
from craft_facade import CraftFacade
from craft_storage import CraftStorage
def main():
url = 'https://ru.minecraft.wiki/w/IndustrialCraft_2/Термальная_центрифуга'
storage = CraftStorage()
facade = CraftFacade()
facade.extract(url, storage, recursive=False)
storage.save()
if __name__ == "__main__":
main()

222
craft_parser/notebook.ipynb Normal file
View File

@ -0,0 +1,222 @@
{
"cells": [
{
"cell_type": "code",
"execution_count": 2,
"id": "d3b20ae9",
"metadata": {},
"outputs": [],
"source": [
"import requests\n",
"from bs4 import BeautifulSoup\n",
"from collections import defaultdict, deque\n",
"from urllib.parse import urljoin, urlparse\n",
"\n",
"from craft import CraftComponent, CraftItem, CraftRecipe\n",
"from craft_storage import CraftStorage\n",
"from collections import Counter\n",
"import simple_extractor\n"
]
},
{
"cell_type": "code",
"execution_count": 3,
"id": "6189c095",
"metadata": {},
"outputs": [],
"source": [
"def get_page_content(url):\n",
" \"\"\"Загружает HTML-контент страницы.\"\"\"\n",
" try:\n",
" response = requests.get(url)\n",
" response.raise_for_status()\n",
" return response.text\n",
" except requests.RequestException as e:\n",
" print(f\"Ошибка при загрузке страницы: {e}\")\n",
" return None\n",
" \n",
"def get_soup(content):\n",
" soup = BeautifulSoup(content, \"html.parser\")\n",
" return soup"
]
},
{
"cell_type": "code",
"execution_count": 4,
"id": "3ea684bc",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Анализирую https://ru.minecraft.wiki/w/IndustrialCraft_2/Квантовая_броня....\n",
"Укреплённое стекло уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Улучшенная электросхема уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовый шлем\n",
"Композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовый жилет\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Основной корпус машины уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Светопыль уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовые поножи\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Резиновые ботинки уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовые ботинки\n",
"Квантовый шлем уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Улучшенная электросхема уже есть в датасете, скипаю\n",
"Укреплённое стекло уже есть в датасете, скипаю\n",
"Улучшенная электросхема уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовый шлем\n",
"Квантовый жилет уже есть в датасете, скипаю\n",
"Композит уже есть в датасете, скипаю\n",
"Композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовый жилет\n",
"Квантовые поножи уже есть в датасете, скипаю\n",
"Основной корпус машины уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Основной корпус машины уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Светопыль уже есть в датасете, скипаю\n",
"Светопыль уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовые поножи\n",
"Квантовые ботинки уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Иридиевый композит уже есть в датасете, скипаю\n",
"Резиновые ботинки уже есть в датасете, скипаю\n",
"Лазуротроновый кристалл уже есть в датасете, скипаю\n",
"Резиновые ботинки уже есть в датасете, скипаю\n",
"Добавлен рецепт крафта для предмета Квантовые ботинки\n",
"Закончен поиск по странице https://ru.minecraft.wiki/w/IndustrialCraft_2/Квантовая_броня\n"
]
},
{
"name": "stderr",
"output_type": "stream",
"text": [
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n",
"d:\\Development\\Python\\craft_calc\\craft_storage.py:56: FutureWarning: In a future version, object-dtype columns with all-bool values will not be included in reductions with bool_only=True. Explicitly cast to bool dtype instead.\n",
" self.items_df = pd.concat(\n"
]
}
],
"source": [
"url = 'https://ru.minecraft.wiki/w/IndustrialCraft_2/Квантовая_броня'\n",
"content = get_page_content(url)\n",
"soup = get_soup(content)\n",
"storage = CraftStorage()\n",
"simple_extractor.extract_crafts(soup,url,storage)"
]
},
{
"cell_type": "code",
"execution_count": 7,
"id": "df4b0ee0",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
" recipe_id input_item count\n",
"0 1 Укреплённое стекло 2\n",
"1 1 Иридиевый композит 2\n",
"2 1 Улучшенная электросхема 2\n",
"3 1 Нановолоконный шлем 1\n",
"4 1 Лазуротроновый кристалл 1\n",
"5 1 Шлем-акваланг 1\n",
"6 2 Иридиевый композит 4\n",
"7 2 Композит 2\n",
"8 2 Нановолоконный жилет 1\n",
"9 2 Лазуротроновый кристалл 1\n",
"10 2 Электрический реактивный ранец 1\n",
"11 3 Основной корпус машины 2\n",
"12 3 Иридиевый композит 2\n",
"13 3 Светопыль 2\n",
"14 3 Лазуротроновый кристалл 1\n",
"15 3 Нановолоконные поножи 1\n",
"16 4 Иридиевый композит 2\n",
"17 4 Резиновые ботинки 2\n",
"18 4 Нановолоконные ботинки 1\n",
"19 4 Лазуротроновый кристалл 1\n"
]
}
],
"source": [
"print(storage.components_df.head(20))"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.5"
}
},
"nbformat": 4,
"nbformat_minor": 5
}

View File

@ -0,0 +1,231 @@
from collections import Counter
import re
from bs4 import BeautifulSoup
from craft import CraftComponent, CraftItem, CraftRecipe
from craft_storage import CraftStorage
from dto import ParsedItem, ParsedRecipeInput, SpriteData
from loguru import logger
def extract_crafts(soup: BeautifulSoup, storage: CraftStorage) -> set[str]:
craft_containers = soup.find_all("div", class_="mcui-Crafting_Table-container")
if(craft_containers.__len__() < 1):
logger.info('Классических рецептов не найдено')
return set()
src_links : set[str] = set()
for container in craft_containers:
is_hidden = check_is_hidden(container)
if(is_hidden):
continue
output_item = parse_craft_item(container)
if(output_item is None):
continue
input_kit = parse_craft_components_and_recipe(container, output_item.item.name, output_item.amount)
if(input_kit is None):
continue
src_links.update(input_kit.source_links)
already_exists = not storage.try_add_recipe_signature(output_item.item.name, input_kit.components, 'classic')
if(already_exists):
continue
storage.add_item(output_item.item)
for input_item in input_kit.items:
storage.add_item(input_item)
recipe_id = storage.add_recipe(input_kit.recipe)
for component in input_kit.components:
component.recipe_id = recipe_id
storage.add_component(component)
logger.info(f'Добавлен рецепт крафта для предмета {output_item.item.name}')
return src_links
def check_is_hidden(container : BeautifulSoup) -> bool:
if(container.find_parent('table', class_='collapsed')):
logger.warning(f'{container} Не будет исследован. Причина: помечен как скрытый')
return True
return False
def parse_craft_item(container) -> ParsedItem|None:
output_amount = 1
output_span = container.find('span', class_='mcui-output')
if(output_span is None):
logger.error(f'ошибка для \n{container}\n: не найдено ячейки с результатом!')
return None
output_span_name_container = output_span.find('span', class_='invslot-item')
if(output_span_name_container is None):
logger.error(f'ошибка для \n{output_span}\n: не найден текстовый контейнер!')
return None
data_from_span = extract_data_from_sprite_span(output_span_name_container)
if(data_from_span is not None):
output_item_title = data_from_span.title
output_item_img_shift = data_from_span.shift
output_item_img_url = data_from_span.img_url
else:
data_from_img = extract_data_from_sprite_img(output_span_name_container)
if(data_from_img is None):
logger.error(f'ошибка для \n{output_span_name_container}\n: не удалось извлечь картинку и заголовок ни одним из способов')
return None
output_item_title = data_from_img.title
output_item_img_shift = data_from_img.shift
output_item_img_url = data_from_img.img_url
item = CraftItem(output_item_title, output_item_img_url, output_item_img_shift)
return ParsedItem(item=item, amount=output_amount)
def parse_craft_components_and_recipe(container, output_item_name: str, output_count = 1) -> ParsedRecipeInput|None:
input_span = container.find('span', class_='mcui-input')
if(input_span is None):
logger.error(f'ошибка для \n{container}\n: не найдено рецепта крафта')
return None
input_span_name_containers = input_span.find_all('span', class_='invslot-item')
if(input_span_name_containers is None or input_span_name_containers.__len__() == 0):
logger.error(f'Ошибка для \n{input_span}\n: не найдено айтемов в рецепте')
return None
craft_items = []
recipe = None
craft_components = []
src_links = []
#existing_items = set()
for input_span_name_container in input_span_name_containers:
if(input_span_name_container is None):
logger.error('ошибка для набора контейнеров: не найден текстовый контейнер!')
return None
data_from_span = extract_data_from_sprite_span(input_span_name_container)
if(data_from_span is not None):
input_item_title = data_from_span.title
input_item_img_shift = data_from_span.shift
input_item_img_url = data_from_span.img_url
if(data_from_span.source_link is not None):
src_links.append(data_from_span.source_link)
else:
data_from_img = extract_data_from_sprite_img(input_span_name_container)
if(data_from_img is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не удалось извлечь картинку и заголовок ни одним из способов')
return None
input_item_title = data_from_img.title
input_item_img_shift = data_from_img.shift
input_item_img_url = data_from_img.img_url
craft_items.append(CraftItem(input_item_title, input_item_img_url, input_item_img_shift))
counter = Counter(c.name for c in craft_items)
recipe = CraftRecipe(output_item_name, output_count, 'Верстак')
for element in counter.most_common(50):
##WARN айдишник изменится при сохранении
craft_components.append(CraftComponent(-1, element[0], element[1]))
return ParsedRecipeInput(
items=craft_items,
recipe=recipe,
components=craft_components,
source_links=[x for x in src_links if x is not None]
)
def extract_data_from_sprite_img(input_span_name_container) -> SpriteData|None:
input_item_img = input_span_name_container.find('img')
if(input_item_img is None):
return None
title = input_item_img.get('alt')
if(title is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найдено описание картинки!')
return None
url = input_item_img.get('src')
if(url is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найден url картинки!')
return None
return SpriteData(
amount=1,
title=title,
shift=(0, 0),
img_url=url,
source_link=None)
def extract_data_from_sprite_span(input_span_name_container) -> SpriteData|None:
input_item_sprite_span = input_span_name_container.find('span', class_='sprite')
#Контейнера со спрайтом может не быть!
if(input_item_sprite_span is None):
return None
input_item_title = input_item_sprite_span.get('title')
if(input_item_title is None):
logger.error(f'ошибка для \n{input_span_name_container}\n: не найдено заголовка предмета')
return None
output_amount=1
amount_container = input_span_name_container.find('span', class_='invslot-stacksize')
if(amount_container is not None):
logger.info(f'Для объекта {input_item_title} найдено количество : {amount_container.text}')
output_amount = int(amount_container.text)
item_source_link = get_item_source_link(input_span_name_container)
input_item_img_url = extract_img_classname(input_item_sprite_span)
if(input_item_img_url is None):
logger.error(f'Ошибка для \n{input_item_img_url}\n: не найден файл спрайта')
return None
input_item_img_shift = extract_img_shift(input_item_sprite_span)
if(input_item_img_shift is None):
logger.error(f'Ошибка для \n{input_item_sprite_span}\n: не найдено смещения для спрайта')
return None
return SpriteData(
amount=output_amount,
title=input_item_title,
shift=input_item_img_shift,
img_url=input_item_img_url,
source_link=item_source_link
)
def extract_img_shift(output_item_sprite_span):
if(output_item_sprite_span is None):
return None
css_value = output_item_sprite_span.get('style')
if(css_value is None):
return None
# Шаблон: ищем два числа (с возможным минусом) перед 'px'
pattern = r'(-?\d+)px\s+(-?\d+)px'
match = re.search(pattern, css_value)
if match:
x = int(match.group(1))
y = int(match.group(2))
return (x, y)
else:
return None # Если шаблон не найден
def extract_img_classname(sprite_span):
if(sprite_span is None):
return None
'sprite industrialcraft-2-inv-sprite'
classes = sprite_span.get('class', [])
for cls in classes:
if cls.endswith('-sprite'):
return cls
return None # Если не найдено
def get_item_source_link(input_span_name_container) -> str|None:
link_container = input_span_name_container.find('a')
if(link_container is None):
return None
link = link_container.get('href')
if(link is None):
return None
return f'https://ru.minecraft.wiki{link}'