mirror of
https://github.com/Shu-Ying/main_plugin_farm.git
synced 2026-10-31 06:49:59 +08:00
add # 新增农场详述指令
This commit is contained in:
@@ -0,0 +1,170 @@
|
||||
"""一次快照生成农场全景和分页明细;不读取或修改数据库。"""
|
||||
|
||||
from dataclasses import dataclass
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from io import BytesIO
|
||||
import logging
|
||||
|
||||
from PIL import Image, ImageDraw
|
||||
|
||||
from farm_core.domain import boosted
|
||||
from farm_core.pests import advance_plot
|
||||
|
||||
from .canvas import Canvas, font
|
||||
from .assets import Assets
|
||||
from .layout import scene_from_snapshot
|
||||
from .models import RenderOptions
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
PAGE_SIZE = 10
|
||||
BEIJING = timezone(timedelta(hours=8))
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class DetailRow:
|
||||
slot: int # 与全景标号一致的本地地块编号。
|
||||
name: str # 作物名称,空地显示空地。
|
||||
state: str # 生长、成熟、枯萎或空地状态。
|
||||
stage: str # 已验证资源中的阶段序号,不猜测开花等语义。
|
||||
season: str # 当前待收获季数;枯萎时为已完成季数。
|
||||
amount: int | None # 当前虫况预估或成熟后的确定产量。
|
||||
maturity: str # 北京时间的预计成熟时间或可操作提示。
|
||||
pest_note: str # 当前虫数及生长减速或成熟减产信息。
|
||||
|
||||
|
||||
def detail_rows(renderer, identity, snapshot, now, page):
|
||||
plots = sorted(snapshot.plots, key=lambda p: p.slot)
|
||||
pages = max(1, (len(plots)+PAGE_SIZE-1)//PAGE_SIZE)
|
||||
if type(page) is not int or not 1 <= page <= pages:
|
||||
raise ValueError('Invalid detail page')
|
||||
metadata = renderer.assets.metadata if renderer else Assets(0).metadata
|
||||
views = {v.slot: v for v in scene_from_snapshot(
|
||||
identity, snapshot, now, metadata,
|
||||
).plots}
|
||||
rows = []
|
||||
for stored in plots[(page-1)*PAGE_SIZE:page*PAGE_SIZE]:
|
||||
plot = advance_plot(stored, now)
|
||||
plant = plot.plant
|
||||
if plant is None:
|
||||
rows.append(DetailRow(plot.slot, '空地', '空地', '—', '—',
|
||||
None, '可播种', ''))
|
||||
continue
|
||||
if plot.withered:
|
||||
rows.append(DetailRow(plot.slot, plant.name, '已枯萎', '枯萎',
|
||||
f'{plant.crop}/{plant.crop}季', None,
|
||||
'可铲除', ''))
|
||||
continue
|
||||
ripe = plot.mature_at is not None and plot.mature_at <= now
|
||||
pests = plot.maturity_pests if ripe else plot.pest_count
|
||||
amount = boosted(plant.harvest, plot.yield_bonus)*(100-5*pests)//100
|
||||
info = metadata['plants'].get(plant.name)
|
||||
total = len(set(info['phase'].split(','))) if info else None
|
||||
stage = ('成熟' if ripe else
|
||||
f'阶段 {views[plot.slot].stage}/{total}' if total else '生长中')
|
||||
maturity = '可收获' if ripe else datetime.fromtimestamp(
|
||||
plot.mature_at, BEIJING,
|
||||
).strftime('%m-%d %H:%M')
|
||||
note = ''
|
||||
if ripe and (plot.pest_count or pests):
|
||||
note = (f'当前虫 {plot.pest_count}只 · 成熟时 {pests}只'
|
||||
f' · 本季减产 {5*pests}%')
|
||||
elif plot.pest_count:
|
||||
note = (f'虫 {plot.pest_count}只 · 生长减速 {5*pests}%'
|
||||
f' · 预计减产 {5*pests}%')
|
||||
rows.append(DetailRow(
|
||||
plot.slot, plant.name, '已成熟' if ripe else '生长中', stage,
|
||||
f'{plot.harvest_count+1}/{plant.crop}季', amount, maturity, note,
|
||||
))
|
||||
return tuple(rows), pages
|
||||
|
||||
|
||||
def details_text(rows, page, pages):
|
||||
lines = [f'农场详述 第{page}/{pages}页']
|
||||
for row in rows:
|
||||
amount = ('—' if row.amount is None else
|
||||
f'{"当前" if row.state == "已成熟" else "预计"}'
|
||||
f'{row.amount}个')
|
||||
lines.append(f'{row.slot}号地 {row.name}|{row.stage}|{row.season}'
|
||||
f'|{amount}|{row.maturity}')
|
||||
if row.pest_note:
|
||||
lines.append(' ' + row.pest_note)
|
||||
lines.append('时间为北京时间;生长中产量按当前虫况估算,后续变化会更新。')
|
||||
if page < pages:
|
||||
lines.append(f'下一页:农场详述 {page+1}')
|
||||
if page > 1:
|
||||
lines.append(f'上一页:农场详述 {page-1}')
|
||||
return '\n'.join(lines)
|
||||
|
||||
|
||||
def render_details(renderer, identity, snapshot, now, page,
|
||||
theme='default', image_format='AUTO', avatar=None):
|
||||
rows, pages = detail_rows(renderer, identity, snapshot, now, page)
|
||||
options = RenderOptions(format=image_format, quality=85, theme=theme)
|
||||
farm = renderer.render(identity, snapshot, now, options, avatar)
|
||||
height = 922 + max(1, len(rows))*92
|
||||
canvas = Canvas.new((1024, height), '#ac713c')
|
||||
draw = ImageDraw.Draw(canvas.image)
|
||||
draw.rounded_rectangle((12, 12, 1012, height-12), 20,
|
||||
fill='#fff1d0', outline='#663e20', width=4)
|
||||
|
||||
def text(value, x, y, size=24, color='#58341c', width=970):
|
||||
value = canvas.fit_text(value, renderer.font_path, size, width)
|
||||
box = font(renderer.font_path, size).getbbox(value)
|
||||
canvas.text((x, y-box[1]), value, renderer.font_path, size, color)
|
||||
|
||||
draw.rounded_rectangle((24, 24, 1000, 82), 12,
|
||||
fill='#d98a32', outline='#a46325', width=2)
|
||||
text(snapshot.account.name+' · 农场详述', 44, 39, 30,
|
||||
'#fff5dd', width=690)
|
||||
text(f'第 {page}/{pages} 页', 802, 45, 23, '#fff5dd', width=180)
|
||||
with Image.open(BytesIO(farm.content)) as image:
|
||||
image = image.convert('RGBA').resize((976, 595),
|
||||
Image.Resampling.LANCZOS)
|
||||
canvas.paste(image, (24, 96))
|
||||
soil_w, soil_h = renderer.assets.metadata['soil_size']
|
||||
selected = {row.slot for row in rows}
|
||||
for plot in snapshot.plots:
|
||||
pos = renderer.assets.metadata['positions'][plot.slot-1]
|
||||
x = round(24 + (pos['x']+soil_w/2)*976/3072)
|
||||
y = round(96 + (pos['y']+soil_h*.85)*595/1872)
|
||||
fill = '#f8d85b' if plot.slot in selected else '#e8e2cc'
|
||||
draw.ellipse((x-14, y-14, x+14, y+14), fill=fill,
|
||||
outline='#67431f', width=2)
|
||||
label = str(plot.slot)
|
||||
box = font(renderer.font_path, 17).getbbox(label)
|
||||
text(label, x-(box[2]-box[0])//2, y-9, 17, width=30)
|
||||
text('黄色编号为本页地块 · 生长阶段与收获季数分别展示',
|
||||
40, 711, 22, '#7c704b')
|
||||
text('生长中按当前虫况估算;已成熟显示当前可收数量。',
|
||||
40, 747, 20, '#7c704b')
|
||||
columns = ((40, 238), (292, 146), (452, 126), (592, 138), (750, 228))
|
||||
draw.rounded_rectangle((24, 785, 1000, 833), 8, fill='#e5d09a')
|
||||
for label, (x, width) in zip(
|
||||
('地块 / 作物', '生长阶段', '收获季数', '预计 / 可收', '成熟时间'),
|
||||
columns,
|
||||
):
|
||||
text(label, x, 797, 22, width=width)
|
||||
for index, row in enumerate(rows):
|
||||
y = 843 + index*92
|
||||
draw.rectangle((24, y-4, 1000, y+82),
|
||||
fill='#fff8e5' if index % 2 == 0 else '#f7eac9')
|
||||
values = (f'{row.slot} {row.name}', row.stage, row.season,
|
||||
'—' if row.amount is None else f'{row.amount}个', row.maturity)
|
||||
for value, (x, width) in zip(values, columns):
|
||||
text(value, x, y+9, 23, width=width)
|
||||
if row.pest_note:
|
||||
text(row.pest_note, 60, y+49, 19, '#9a682f', width=910)
|
||||
if not rows:
|
||||
text('暂无已解锁地块', 390, 870, 26)
|
||||
footer = 858 + max(1, len(rows))*92
|
||||
navigation = f'更新于 {datetime.fromtimestamp(now, BEIJING):%m-%d %H:%M:%S}'
|
||||
if page > 1:
|
||||
navigation += f' 上一页:农场详述 {page-1}'
|
||||
if page < pages:
|
||||
navigation += f' 下一页:农场详述 {page+1}'
|
||||
text(navigation, 40, footer, 22, width=940)
|
||||
codec = options.codec(('JPEG',))
|
||||
content = canvas.encode(codec, options.quality, options.method)
|
||||
logger.info('farm details page=%s pages=%s rows=%s bytes=%s codec=%s',
|
||||
page, pages, len(rows), len(content), codec)
|
||||
return content
|
||||
@@ -7,6 +7,7 @@ import time
|
||||
import hashlib
|
||||
|
||||
from .shop import render_shop
|
||||
from .details import render_details
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -65,11 +66,12 @@ class RenderService:
|
||||
async def _run(self, *args):
|
||||
return await self._run_function(self.renderer.render, *args)
|
||||
|
||||
async def seed_shop(self, plants, account, page, pages):
|
||||
async def seed_shop(self, plants, account, page, pages,
|
||||
image_format='AUTO'):
|
||||
if self._closed:
|
||||
raise RuntimeError('Render service closed')
|
||||
key = 'shop:' + hashlib.sha256(
|
||||
repr((plants, account, page, pages)).encode(),
|
||||
repr((plants, account, page, pages, image_format)).encode(),
|
||||
).hexdigest()
|
||||
task = self._pending.get(key)
|
||||
if task is None:
|
||||
@@ -77,6 +79,32 @@ class RenderService:
|
||||
raise RuntimeError('Render queue full')
|
||||
task = asyncio.create_task(self._run_function(
|
||||
render_shop, self.renderer, plants, account, page, pages,
|
||||
image_format,
|
||||
))
|
||||
self._pending[key] = task
|
||||
|
||||
def finished(done):
|
||||
self._pending.pop(key, None)
|
||||
if not done.cancelled():
|
||||
done.exception()
|
||||
|
||||
task.add_done_callback(finished)
|
||||
return await asyncio.shield(task)
|
||||
|
||||
async def farm_details(self, identity, snapshot, now, page, theme,
|
||||
image_format='AUTO', avatar=None):
|
||||
if self._closed:
|
||||
raise RuntimeError('Render service closed')
|
||||
key = 'details:' + hashlib.sha256(repr((
|
||||
identity, snapshot, now, page, theme, image_format, avatar,
|
||||
)).encode()).hexdigest()
|
||||
task = self._pending.get(key)
|
||||
if task is None:
|
||||
if len(self._pending) >= self._max_pending:
|
||||
raise RuntimeError('Render queue full')
|
||||
task = asyncio.create_task(self._run_function(
|
||||
render_details, self.renderer, identity, snapshot, now,
|
||||
page, theme, image_format, avatar,
|
||||
))
|
||||
self._pending[key] = task
|
||||
|
||||
|
||||
@@ -6,6 +6,7 @@ import shlex
|
||||
from PIL import Image
|
||||
|
||||
from .canvas import Canvas, font
|
||||
from .models import RenderOptions
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
PAGE_SIZE = 6
|
||||
@@ -29,7 +30,9 @@ def duration(seconds):
|
||||
) if value) or '0秒'
|
||||
|
||||
|
||||
def render_shop(renderer, plants, account, page, pages):
|
||||
def render_shop(renderer, plants, account, page, pages, image_format='AUTO'):
|
||||
options = RenderOptions(format=image_format, quality=80)
|
||||
codec = options.codec()
|
||||
canvas = Canvas(renderer.assets.image('webgl/shop-board.png', (1024, 1536)))
|
||||
|
||||
def text(value, xy, size=28, color='#5b3015', width=850):
|
||||
@@ -90,4 +93,7 @@ def render_shop(renderer, plants, account, page, pages):
|
||||
elif page > 1:
|
||||
navigation += f' 上一页:种子商店 {page-1}'
|
||||
text(navigation, (263, 1420), 23, width=510)
|
||||
return canvas.encode('PNG')
|
||||
content = canvas.encode(codec, options.quality, options.method)
|
||||
logger.info('farm shop render width=%s height=%s bytes=%s codec=%s',
|
||||
*canvas.image.size, len(content), codec)
|
||||
return content
|
||||
|
||||
Reference in New Issue
Block a user