kadupul-git/BridgeDiseaseBackend-main/scripts/seed_medias.py

# -*- coding: utf-8 -*-
"""将 seed_assets/medias 中的真实照片导入 static/medias 并同步 media 表。"""
import os
import shutil
import sys
from pathlib import Path

from PIL import Image

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

os.environ.setdefault(
    'SQLALCHEMY_DATABASE_URI',
    'mysql+pymysql://root:bridgedisease_root@127.0.0.1:3307/bridge_disease',
)

from app import create_app
from app.models import Media, db

ASSETS_DIR = ROOT / 'seed_assets' / 'medias'
MEDIAS_DIR = ROOT / 'app' / 'static' / 'medias'
MAX_WIDTH = 1600

# 源文件（seed_assets 内） -> 目标与元数据
SEED_ITEMS = [
    {
        'source': '01_concrete_crack_bridge.jpg',
        'filename': '01_concrete_crack_bridge.jpg',
        'media_name': '混凝土桥面裂缝_01.jpg',
        'description': '桥梁混凝土底板裂缝（德国 Darmsheim 桥，实景）',
        'owner_id': 1,
    },
    {
        'source': '02_bridge_concrete_cracks.jpg',
        'filename': '02_bridge_concrete_cracks.jpg',
        'media_name': '桥梁混凝土结构裂缝_02.jpg',
        'description': '桥梁混凝土结构多处裂缝（实景）',
        'owner_id': 2,
    },
    {
        'source': '03_steel_bridge_corrosion.jpg',
        'filename': '03_steel_bridge_corrosion.jpg',
        'media_name': '钢构件锈蚀_03.jpg',
        'description': '金属构件锈蚀与氧化皮（实景）',
        'owner_id': 2,
    },
    {
        'source': '04_concrete_bending_cracks.jpg',
        'filename': '04_concrete_bending_cracks.jpg',
        'media_name': '混凝土弯曲裂缝_04.jpg',
        'description': '混凝土试件弯曲裂缝（RILEM 三点弯曲试验实景）',
        'owner_id': 2,
    },
    {
        'source': '05_bridge_substructure.jpg',
        'filename': '05_bridge_substructure.jpg',
        'media_name': '桥梁下部结构_05.jpg',
        'description': '桥梁混凝土底板与下部结构裂缝（实景）',
        'owner_id': 1,
        'fallback_source': '01_concrete_crack_bridge.jpg',
    },
    {
        'source': '06_shrinkage_cracks_concrete.jpg',
        'filename': '06_shrinkage_cracks_concrete.jpg',
        'media_name': '混凝土收缩裂缝_06.jpg',
        'description': '钢筋混凝土收缩裂缝（实景）',
        'owner_id': 3,
    },
    {
        'source': '07_asphalt_crocodile_cracking.jpg',
        'filename': '07_asphalt_crocodile_cracking.jpg',
        'media_name': '桥面铺装龟裂_07.jpg',
        'description': '沥青路面龟裂网状裂缝（实景）',
        'owner_id': 1,
    },
    {
        'source': '08_concrete_rebar_corrosion.jpg',
        'filename': '08_concrete_rebar_corrosion.jpg',
        'media_name': '混凝土钢筋锈蚀_08.jpg',
        'description': '混凝土结构钢筋锈蚀与表面劣化（实景）',
        'owner_id': 3,
    },
    {
        'source': '09_steel_beam_site.jpg',
        'filename': '09_steel_beam_site.jpg',
        'media_name': '施工现场钢梁_09.jpg',
        'description': '施工现场钢梁吊装与拼装（实景）',
        'owner_id': 1,
    },
]

LEGACY_NAMES = [
    'main_span_deck_inspection.png',
    'steel_box_girder_u_rib.png',
    'expansion_joint_j03.png',
    'pier_cap_east_view.png',
    'bearing_pad_top_view.png',
    'deck_pavement_drone_01.png',
    'cable_saddle_interior.png',
    'parapet_root_seepage.png',
    'concrete_crack_deck.jpg',
    'steel_corrosion_girder.jpg',
    'expansion_joint_spalling.jpg',
    'steel_beam_construction_site.jpg',
    'bearing_pad_surface.jpg',
    'coating_peel_steel.jpg',
    'concrete_rebar_exposure.jpg',
    'pavement_distress.jpg',
    '混凝土桥面裂缝_01.jpg',
    '钢箱梁锈蚀_02.jpg',
    '伸缩缝剥落_03.jpg',
    '施工现场钢梁_04.jpg',
    '支座垫石开裂_05.jpg',
    '涂层剥落_06.jpg',
    '混凝土露筋_07.jpg',
    '桥面铺装破损_08.jpg',
]


def prepare_image(src: Path, dest: Path) -> None:
    """压缩并统一为 JPEG（PNG 源文件）。"""
    dest.parent.mkdir(parents=True, exist_ok=True)
    with Image.open(src) as im:
        im = im.convert('RGB')
        w, h = im.size
        if w > MAX_WIDTH:
            nh = int(h * MAX_WIDTH / w)
            im = im.resize((MAX_WIDTH, nh), Image.Resampling.LANCZOS)
        out = dest
        if dest.suffix.lower() == '.png':
            out = dest.with_suffix('.jpg')
        im.save(out, format='JPEG', quality=88, optimize=True)
        if out != dest and dest.exists():
            dest.unlink()
        return out


def resolve_source(item: dict) -> Path | None:
    p = ASSETS_DIR / item['source']
    if p.is_file():
        return p
    fb = item.get('fallback_source')
    if fb:
        p2 = ASSETS_DIR / fb
        if p2.is_file():
            print(f"  [fallback] {item['source']} -> {fb}")
            return p2
    return None


def main():
    if not ASSETS_DIR.is_dir():
        print(f'缺少目录 {ASSETS_DIR}，请先运行: python scripts/download_real_medias.py')
        sys.exit(1)

    MEDIAS_DIR.mkdir(parents=True, exist_ok=True)

    app = create_app()
    with app.app_context():
        Media.query.filter(Media.media_name.in_(LEGACY_NAMES)).delete(synchronize_session=False)
        for legacy in LEGACY_NAMES:
            p = MEDIAS_DIR / legacy
            if p.is_file():
                p.unlink()
        db.session.commit()

        added, updated = 0, 0
        for item in SEED_ITEMS:
            src = resolve_source(item)
            if not src:
                print(f'[skip] 缺少源图: {item["source"]}')
                continue

            dest_name = Path(item['filename'])
            if dest_name.suffix.lower() == '.png':
                dest_name = dest_name.with_suffix('.jpg')
                item = {**item, 'filename': dest_name.name, 'media_name': item['media_name'].replace('.png', '.jpg')}

            dest = MEDIAS_DIR / dest_name.name
            saved = prepare_image(src, dest)
            rel_path = f'static/medias/{saved.name}'.replace('\\', '/')
            abs_path = ROOT / 'app' / rel_path.replace('/', os.sep)

            with Image.open(abs_path) as im:
                w, h = im.size
            file_size = os.path.getsize(abs_path) / 1024
            file_type = 'jpeg'

            existing = Media.query.filter_by(media_name=item['media_name']).first()
            if existing:
                existing.media_path = rel_path
                existing.description = item['description']
                existing.file_size = round(file_size, 2)
                existing.file_type = file_type
                existing.resolution_width = w
                existing.resolution_height = h
                existing.frame_count = 1
                updated += 1
                print(f'[update] {item["media_name"]}')
            else:
                db.session.add(
                    Media(
                        media_name=item['media_name'],
                        media_path=rel_path,
                        description=item['description'],
                        file_size=round(file_size, 2),
                        file_type=file_type,
                        resolution_width=w,
                        resolution_height=h,
                        frame_count=1,
                        owner_id=item['owner_id'],
                    )
                )
                added += 1
                print(f'[insert] {item["media_name"]}')

        db.session.commit()
        print(f'完成：新增 {added}，更新 {updated}，媒体库共 {Media.query.count()} 条。')
        print('照片来源说明见 seed_assets/medias/ATTRIBUTION.md')


if __name__ == '__main__':
    main()