cms-SiteForge/app/assembly.py

"""Declarative mapping from arbitrary tabular source data to SiteForge content.
Never eval mappings, no AI hallucinated fields, preview shows all errors.
"""
import json,re
from . import db
from .core import route_path,slug_of,save_entry

ALIASES={
 'title':('title','name','product_name','model_name','名称','标题','产品名称','车型','车型名称','商品名称'),
 'route':('route','url','path','permalink','链接','路径','旧站链接','原网址'),
 'summary':('summary','description','excerpt','简介','摘要','产品简介'),
 'body':('body','content','html','正文','内容','详细介绍'),
 'cover':('cover','image','thumbnail','featured_image','图片','封面','主图'),
 'brand':('brand','manufacturer','品牌','厂商'),
 'category':('category','分类','类目'),
 'seo_title':('seo_title','meta_title','SEO标题','seo标题'),
 'seo_description':('seo_description','meta_description','SEO描述','seo描述'),
 'canonical':('canonical','规范网址'),
 'model':('model','model_name','车型型号','型号','规格型号'),
 'energy':('energy','energy_type','fuel_type','动力形式','能源类型','燃料类型'),
 'range':('range','battery_range','纯电续航','续航','续航里程'),
 'year':('year','model_year','年款','年份'),
 'sku':('sku','product_sku','产品编号','货号'),
 'gallery':('gallery','images','image_gallery','产品图片','图库','详情图片'),
 'price':('price','msrp','售价','价格'),
 'specs':('specs','attributes','parameters','技术参数','产品参数'),

}
SYSTEM_FIELDS={'title','route','summary','body','cover','seo_title','seo_description','canonical'}


def setup():
    with db.connect() as c:
        c.executescript('''CREATE TABLE IF NOT EXISTS mapping_profiles (
          id TEXT PRIMARY KEY,site_id TEXT NOT NULL REFERENCES sites(id) ON DELETE CASCADE,
          name TEXT NOT NULL,content_type TEXT NOT NULL, rules_json TEXT NOT NULL,
          created_at TEXT NOT NULL,updated_at TEXT NOT NULL);
          CREATE TABLE IF NOT EXISTS import_batches (
          id TEXT PRIMARY KEY,site_id TEXT NOT NULL REFERENCES sites(id) ON DELETE CASCADE,
          profile_id TEXT, source TEXT NOT NULL, result_json TEXT NOT NULL,
          created_at TEXT NOT NULL);
        ''')


def suggest(columns):
    clean=lambda x:re.sub(r'[\s\-_]+','',str(x).casefold())
    src={clean(col):col for col in columns}
    mapping={}
    for target,aliases in ALIASES.items():
        for alias in aliases:
            if clean(alias) in src:
                mapping[target]=src[clean(alias)];break
    return mapping


def validate_rules(rules,columns):
    if not isinstance(rules,dict) or len(rules)>100:raise ValueError('字段映射规则不合法')
    for dest,src in rules.items():
        if not re.fullmatch(r'[\w\-]{1,60}',dest,re.UNICODE):raise ValueError('目标字段名称不合法')
        if not isinstance(src,str) or src not in columns:raise ValueError(f'来源字段不存在: {src}')
    if 'title' not in rules:raise ValueError('必须映射标题字段')


def transform(rows,rules,kind='product',prefix=None):
    if kind not in ('product','post','page','brand','vehicle','service','case','other'):raise ValueError('不支持的内容类型')
    columns=set().union(*(r.keys() for r in rows)) if rows else set()
    validate_rules(rules,columns)
    seen=set();out=[];errors=[]
    for index,row in enumerate(rows):
        vals={}
        for target,src in rules.items():
            raw=row.get(src)
            vals[target]=(json.dumps(raw,ensure_ascii=False) if isinstance(raw,(dict,list)) else str(raw if raw is not None else '')).strip()
        title=vals.get('title','')
        if not title:
            errors.append({'row':index+1,'error':'标题为空'});continue
        route=vals.get('route','')
        if not route:
            route=f'/{prefix or kind}/{slug_of(title)}/'
        try:route=route_path(route)
        except ValueError as exc:
            errors.append({'row':index+1,'error':f'路径非法：{exc}'});continue
        if route in seen:
            errors.append({'row':index+1,'error':'导入文件内 URL 重复：'+route});continue
        seen.add(route)
        fields={k:v for k,v in vals.items() if k not in SYSTEM_FIELDS}
        if fields.get('gallery'):
            try:
                values=json.loads(fields['gallery'])
                if isinstance(values,list):fields['gallery']=[str(x.get('url','') if isinstance(x,dict) else x) for x in values if x]
            except ValueError:
                fields['gallery']=[x.strip() for x in fields['gallery'].split(';') if x.strip()][:40]
        out.append({'type':kind,'title':title,'route':route,'summary':vals.get('summary',''),
                    'body':vals.get('body',''),'cover':vals.get('cover',''),
                    'seo_title':vals.get('seo_title',''),'seo_description':vals.get('seo_description',''),
                    'canonical':vals.get('canonical',''),'meta':{'fields':fields}})
    return out,errors


def dry_run(site_id,rows,rules,kind='product',prefix=None):
    if len(rows)>1000:raise ValueError('单批最多1000条内容')
    items,errors=transform(rows,rules,kind,prefix)
    existing={r['route'] for r in db.rows('SELECT route FROM entries WHERE site_id=?',(site_id,))}
    collisions=[r['route'] for r in items if r['route'] in existing]
    return {'input':len(rows),'mapped':len(items),'errors':errors,'conflicts':collisions,
            'unmapped_columns':sorted(set().union(*(x.keys() for x in rows)) - set(rules.values())) if rows else [],
            'sample':items[:10], 'items':items}


def commit(site_id,rows,rules,kind='product',on_conflict='skip',prefix=None):
    if on_conflict not in ('skip','replace'):raise ValueError('冲突策略只能为 skip 或 replace')
    report=dry_run(site_id,rows,rules,kind,prefix)
    created=0;updated=0;skipped=0;errors=list(report['errors'])
    for idx,item in enumerate(report['items'],1):
        try:
            existing=db.one('SELECT id FROM entries WHERE site_id=? AND route=?',(site_id,item['route']))
            if existing and on_conflict=='skip':skipped+=1;continue
            save_entry(site_id,item,existing_id=existing['id'] if existing else None,source='mapped-import')
            if existing:updated+=1
            else:created+=1
        except Exception as exc:
            errors.append({'row':idx,'error':str(exc)[:250]})
    summary={'created':created,'updated':updated,'skipped':skipped,'errors':errors,'mapped':report['mapped']}
    from .core import uid
    db.insert('INSERT INTO import_batches VALUES (?,?,?,?,?,?)',(uid(),site_id,'',f'content-map:{kind}',json.dumps(summary,ensure_ascii=False),db.now()))
    return summary