Files

172 lines
6.8 KiB
Markdown

---
name: atomk-auto-generate
description: Automatically log into AtomK, select a claimed product, choose an e-commerce platform, and AI-generate listing data.
---
# AtomK Auto Generate Listing
Automates logging into AtomK, opening a claimed product's editor, selecting a target platform (e.g., 速卖通, 亚马逊, Temu), and triggering AI generation.
## Trigger
Use when user asks to: "一键生成", "生成速卖通资料", "生成亚马逊资料", or generate product info for a specific platform.
## Prerequisites
- Node.js with Playwright installed at: `/home/ubuntu/.hermes/hermes-agent/node_modules/playwright`
- Playwright Chromium at: `/home/ubuntu/.cache/ms-playwright/chromium-1217/`
## Critical Findings
- **Always use `https://www.atomlisting.com`** — `bt109atomk.sh3.ikuai7.com` silently fails on login (stays on /login). Both serve the same "九个网AtomK" app.
- **Direct editor URLs work best**: `/editor/{product_id}` allows bypassing the "My Products" scrolling/indexing problem entirely.
- **Generation status detection**: Check body text for "已生成" AND absence of "选择图片并点击" to confirm a listing already exists. "已生成" alone may appear in the AI generate button text.
- **DrissionPage works**: Use DrissionPage (installed in venv) rather than Playwright for consistency with other automation scripts.
- Login form uses standard inputs; press Enter on password field to submit.
- AtomK is a Next.js app (uses turbopack chunks, `_rsc` server components)
## Execution Steps
1. Kill hung browser instances: `killall -9 chrome chromium chrome-headless-shell 2>/dev/null`
2. Save the script to `/tmp/atomk_generate.py`
3. Execute via terminal:
```bash
/home/ubuntu/.hermes/hermes-agent/venv/bin/python3 /tmp/atomk_generate.py <username> <password> <platform_name> [product_index]
```
- `platform_name`: e.g., "速卖通", "亚马逊", "Temu"
- `product_index`: defaults to 0 (first product in list)
- Set timeout to ~30s for generation.
## Batch Generation (Recommended)
For generating listings for multiple products efficiently, use **direct editor URLs** instead of navigating via the "My Products" page:
```python
from DrissionPage import ChromiumPage, ChromiumOptions
import random, time
TARGET_IDS = [607, 606, 605, 604] # product IDs to generate
PLATFORM = "Temu"
def generate_for_product(page, product_id):
page.get(f'https://www.atomlisting.com/editor/{product_id}')
time.sleep(4)
# Check if already generated
body_text = page.ele('tag:body').text
if '已生成' in body_text and '选择图片并点击' not in body_text:
print(f"ID={product_id}: already generated, skipping")
return True
# Select platform
platform_btn = page.ele(f'text:{PLATFORM}')
if platform_btn:
platform_btn.click()
time.sleep(1)
# Find and click generate button
for btn in page.eles('tag:button'):
if '生成' in btn.text:
btn.click()
print(f"ID={product_id}: generating...")
break
# Poll for completion (up to 20s)
for i in range(20):
time.sleep(1)
body = page.ele('tag:body').text
if '已生成' in body or '标题' in body:
print(f"ID={product_id}: success!")
return True
return False
```
**Why direct URLs are better:**
- My Products page requires scrolling to load all products
- Edit button index mapping is unreliable (page may show 100+ products)
- Direct `/editor/{id}` is O(1) vs O(n) scrolling
## Python Script (`/tmp/atomk_generate.py`)
```python
import sys
import time
from DrissionPage import ChromiumPage, ChromiumOptions
def batch_generate(username, password, platform_name, product_ids):
"""Generate AI listings for multiple products by ID."""
co = ChromiumOptions()
co.set_browser_path('/home/ubuntu/.cache/ms-playwright/chromium-1217/chrome-linux64/chrome')
co.headless(True)
co.set_argument('--headless=new')
co.set_argument('--no-sandbox')
co.set_argument('--disable-dev-shm-usage')
co.set_argument('--disable-gpu')
co.set_argument('--window-size=1440,900')
co.set_user_agent('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36')
co.set_local_port(0)
co.set_user_data_path(f'/tmp/dpcache_{int(time.time())}')
page = ChromiumPage(addr_or_opts=co)
try:
print("Logging in...")
page.get('https://www.atomlisting.com')
time.sleep(2)
page.ele('css:input[placeholder="请输入用户名"]').input(username)
page.ele('css:input[placeholder="请输入密码"]').input(password)
page.ele('css:input[placeholder="请输入密码"]').input('\n') # press Enter to submit
time.sleep(5)
success = 0
for pid in product_ids:
print(f"\nProcessing product {pid}...")
page.get(f'https://www.atomlisting.com/editor/{pid}')
time.sleep(4)
body_text = page.ele('tag:body').text
if '已生成' in body_text and '选择图片并点击' not in body_text:
print(f" Already generated, skipping.")
success += 1
continue
# Select platform
platform_btn = page.ele(f'text:{platform_name}')
if platform_btn:
platform_btn.click()
time.sleep(1)
# Click generate button
gen_btn = None
for b in page.eles('tag:button'):
if '生成' in b.text:
gen_btn = b
break
if gen_btn:
print(f" Clicking: {gen_btn.text}")
gen_btn.click()
# Poll for completion
for i in range(20):
time.sleep(1)
body = page.ele('tag:body').text
if '已生成' in body or '标题' in body:
print(f" Generation successful!")
success += 1
break
else:
print(f" Timed out, may still be processing.")
else:
print(" Generate button not found.")
time.sleep(2)
print(f"\nDone! {success}/{len(product_ids)} products processed.")
except Exception as e:
print("Fatal Error:", e)
finally:
page.quit()
if __name__ == "__main__":
if len(sys.argv) < 5:
print("Usage: python3 atomk_generate.py <username> <password> <platform_name> <product_id1> [product_id2 ...]")
sys.exit(1)
user = sys.argv[1]
pwd = sys.argv[2]
plat = sys.argv[3]
ids = [int(x) for x in sys.argv[4:]]
batch_generate(user, pwd, plat, ids)
```