ht80/houduan/ai_planner/management/commands/import_red_spots.py

from django.core.management.base import BaseCommand
from ai_planner.models import Attraction, City
import os
from django.core.files import File


class Command(BaseCommand):
    help = '导入所有红色旅游景点数据到Attraction表（含图片）'

    def handle(self, *args, **options):
        # 处理门票价格转换
        def parse_ticket_price(ticket_info):
            if "免费" in ticket_info:
                return 0
            try:
                # 提取数字部分
                price_str = ''.join(filter(str.isdigit, ticket_info))
                return float(price_str) if price_str else 0
            except:
                return 0

        # 更健壮的城市名称提取方法
        def get_city_name(location):
            shandong_cities = [
                '济南市', '青岛市', '淄博市', '枣庄市', '东营市',
                '烟台市', '潍坊市', '济宁市', '泰安市', '威海市',
                '日照市', '临沂市', '德州市', '聊城市', '滨州市',
                '菏泽市'
            ]
            for city in shandong_cities:
                if city in location:
                    return city
            return '济南市'

        # 图片存储基础路径（根据实际路径修改）
        IMAGE_BASE_PATH = r"G:\其他\jingdian\image+name"

        # 所有红色景点数据（保持不变）...
        red_spots_data = [
            {
                "name": "台儿庄大战纪念馆",
                "category": "memorial",
                "location": "山东省枣庄市台儿庄区沿河南路6号",
                "description": "台儿庄大战纪念馆是为纪念1938年台儿庄战役而建，展示了大量珍贵的历史照片和文物。",
                "history": "台儿庄战役是抗日战争初期中国军队在山东省南部台儿庄地区抗击日军的重大胜利，是中国军队在正面战场取得的首次重大胜利。",
                "open_time": "全年开放 8:30-17:30",
                "ticket_info": "免费开放",
                "short_desc": "纪念1938年台儿庄战役的专题纪念馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "孟良崮战役遗址",
                "category": "revolution",
                "location": "山东省临沂市蒙阴县垛庄镇",
                "description": "孟良崮战役遗址是解放战争时期著名的孟良崮战役发生地，现建有纪念馆和烈士陵园。",
                "history": "1947年5月，华东野战军在陈毅、粟裕指挥下，全歼国民党军整编第74师，击毙师长张灵甫，是解放战争中的重要转折点。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票50元",
                "short_desc": "解放战争重要战役遗址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "华东革命烈士陵园",
                "category": "martyrs",
                "location": "山东省临沂市兰山区陵园前街4号",
                "description": "华东革命烈士陵园是为纪念在抗日战争和解放战争中牺牲的华东地区革命烈士而建。",
                "history": "建于1949年，安葬着包括罗炳辉、王麓水等著名烈士在内的6万余名革命烈士。",
                "open_time": "8:00-17:00",
                "ticket_info": "免费开放",
                "short_desc": "安葬6万余名革命烈士的陵园",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "沂蒙革命纪念馆",
                "category": "museum",
                "location": "山东省临沂市兰山区银雀山路与沂蒙路交汇处",
                "description": "沂蒙革命纪念馆全面展示了沂蒙革命老区的光荣历史和沂蒙精神的形成发展过程。",
                "history": "沂蒙山区是抗日战争和解放战争时期重要的革命根据地之一，涌现出'红嫂'等众多感人事迹。",
                "open_time": "周二至周日 9:00-16:30",
                "ticket_info": "免费开放",
                "short_desc": "展示沂蒙革命老区历史的专题纪念馆",
                "tags": ["museum", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "铁道游击队纪念园",
                "category": "memorial",
                "location": "山东省枣庄市薛城区临山路",
                "description": "铁道游击队纪念园是为纪念抗日战争时期活跃在鲁南地区的铁道游击队而建。",
                "history": "铁道游击队是抗日战争时期鲁南地区著名的抗日武装，以扒火车、打游击著称，小说《铁道游击队》即以此为原型。",
                "open_time": "8:30-17:30",
                "ticket_info": "门票30元",
                "short_desc": "纪念抗日战争时期铁道游击队的专题园区",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "胶东革命纪念馆",
                "category": "museum",
                "location": "山东省烟台市芝罘区南大街61号",
                "description": "胶东革命纪念馆展示了胶东地区党组织创建、发展和领导人民进行革命斗争的历史。",
                "history": "胶东地区是山东党组织活动较早的地区之一，在抗日战争和解放战争中做出了重要贡献。",
                "open_time": "周二至周日 9:00-16:30",
                "ticket_info": "免费开放",
                "short_desc": "展示胶东地区革命历史的专题纪念馆",
                "tags": ["museum", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "沂蒙山小调诞生地",
                "category": "revolution",
                "location": "山东省临沂市费县薛庄镇",
                "description": "《沂蒙山小调》诞生地是这首著名革命歌曲的创作地，现建有纪念设施。",
                "history": "1940年，抗大一分校文工团在此创作了《沂蒙山小调》，后成为歌颂沂蒙革命精神的代表性歌曲。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票20元",
                "short_desc": "著名革命歌曲《沂蒙山小调》的创作地",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "鲁南人民抗日武装起义纪念馆",
                "category": "memorial",
                "location": "山东省枣庄市山亭区北庄镇",
                "description": "纪念馆展示了1938年鲁南人民抗日武装起义的历史和意义。",
                "history": "1938年5月，中共苏鲁豫皖边区特委领导发动鲁南人民抗日武装起义，创建了鲁南抗日根据地。",
                "open_time": "周二至周日 9:00-16:30",
                "ticket_info": "免费开放",
                "short_desc": "纪念1938年鲁南人民抗日武装起义的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "大青山突围战遗址",
                "category": "revolution",
                "location": "山东省临沂市费县薛庄镇",
                "description": "大青山突围战遗址是抗日战争时期著名的突围战发生地，现建有纪念碑和纪念馆。",
                "history": "1941年11月，抗大一分校及中共山东分局、省战工会等机关在此成功突围，粉碎了日军的'铁壁合围'",
                "open_time": "8:30-17:00",
                "ticket_info": "门票30元",
                "short_desc": "抗日战争时期著名突围战遗址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "胶东抗日烈士陵园",
                "category": "martyrs",
                "location": "山东省烟台市牟平区龙泉镇",
                "description": "胶东抗日烈士陵园是为纪念在胶东抗日斗争中牺牲的烈士而建。",
                "history": "建于1945年，安葬着包括理琪、林江等著名烈士在内的数千名抗日烈士。",
                "open_time": "8:00-17:00",
                "ticket_info": "免费开放",
                "short_desc": "纪念胶东抗日斗争中牺牲烈士的陵园",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "八路军115师司令部旧址",
                "category": "revolution",
                "location": "山东省临沂市莒南县大店镇",
                "description": "八路军115师司令部旧址是抗日战争时期115师司令部的驻地，现为全国重点文物保护单位。",
                "history": "1941年3月至1945年9月，八路军115师司令部驻扎于此，罗荣桓等曾在此办公。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票40元",
                "short_desc": "抗日战争时期八路军115师司令部驻地",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "红嫂纪念馆",
                "category": "memorial",
                "location": "山东省临沂市沂南县马牧池乡",
                "description": "红嫂纪念馆是为纪念沂蒙红嫂明德英等革命妇女而建，展示了红嫂们的感人事迹。",
                "history": "抗日战争时期，沂蒙山区涌现出许多像明德英这样的'红嫂'，她们用乳汁救治伤员，谱写了军民鱼水情的感人篇章。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票30元",
                "short_desc": "纪念沂蒙红嫂感人事迹的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "山东省政府旧址",
                "category": "revolution",
                "location": "山东省临沂市莒南县大店镇",
                "description": "山东省政府旧址是抗日战争时期山东省政府所在地，现为全国重点文物保护单位。",
                "history": "1945年8月13日，山东省政府在此成立，是中国共产党领导的第一个省级人民政府。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票30元",
                "short_desc": "中国共产党领导的第一个省级政府旧址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "徂徕山抗日武装起义旧址",
                "category": "revolution",
                "location": "山东省泰安市岱岳区徂徕镇",
                "description": "徂徕山抗日武装起义旧址是1938年1月徂徕山抗日武装起义的发生地。",
                "history": "1938年1月1日，中共山东省委在此领导发动徂徕山抗日武装起义，打响了山东省委独立领导山东抗战的第一枪。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票20元",
                "short_desc": "1938年徂徕山抗日武装起义发生地",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "黑铁山抗日武装起义旧址",
                "category": "revolution",
                "location": "山东省淄博市张店区卫固镇",
                "description": "黑铁山抗日武装起义旧址是1937年12月黑铁山抗日武装起义的发生地。",
                "history": "1937年12月26日，中共山东省委在此领导发动黑铁山抗日武装起义，是山东最早的抗日武装起义之一。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票20元",
                "short_desc": "1937年黑铁山抗日武装起义发生地",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "天福山起义纪念馆",
                "category": "memorial",
                "location": "山东省威海市文登区文登营镇",
                "description": "天福山起义纪念馆是为纪念1937年12月天福山起义而建。",
                "history": "1937年12月24日，中共胶东特委在此领导发动天福山起义，创建了山东人民抗日救国军第三军。",
                "open_time": "周二至周日 9:00-16:30",
                "ticket_info": "免费开放",
                "short_desc": "纪念1937年天福山起义的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "马石山十勇士纪念馆",
                "category": "memorial",
                "location": "山东省威海市乳山市马石山",
                "description": "马石山十勇士纪念馆是为纪念在1942年反'扫荡'中英勇牺牲的十位八路军战士而建。",
                "history": "1942年冬，十位八路军战士为掩护群众突围，与日军激战后全部壮烈牺牲，谱写了气壮山河的英雄赞歌。",
                "open_time": "8:30-17:00",
                "ticket_info": "免费开放",
                "short_desc": "纪念1942年十位英勇牺牲的八路军战士",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "沂蒙革命根据地旧址群",
                "category": "revolution",
                "location": "山东省临沂市沂南县",
                "description": "沂蒙革命根据地旧址群包括中共山东分局、八路军115师司令部、山东省战工会等多处革命旧址。",
                "history": "抗日战争和解放战争时期，沂蒙山区是山东党政军领导机关所在地，被誉为'山东的小延安'",
                "open_time": "8:30-17:00",
                "ticket_info": "联票60元",
                "short_desc": "包含多处重要革命旧址的遗址群",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "济南战役纪念馆",
                "category": "memorial",
                "location": "济南市市中区英雄山路18号",
                "description": "全国爱国主义教育示范基地，全景展示济南战役历史进程，含全景画馆、文物陈列厅等。",
                "history": "1948年9月16-24日，华东野战军经8昼夜激战攻克济南，揭开战略决战序幕。纪念馆2003年正式开放。",
                "open_time": "周二至周日 9:00-16:30（周一闭馆）",
                "ticket_info": "凭身份证免费参观",
                "short_desc": "全景展示济南战役历史的纪念馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "解放阁",
                "category": "memorial",
                "location": "济南市历下区黑虎泉北路",
                "description": "济南战役突破城垣遗址上建立的纪念性建筑，登阁可俯瞰护城河及老城区。",
                "history": "原为济南城墙东南角，1948年解放军由此突破。1965年建阁，1986年改建为战役纪念建筑。",
                "open_time": "8:30-17:00",
                "ticket_info": "免费开放",
                "short_desc": "济南战役突破城垣遗址纪念建筑",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "英雄山革命烈士陵园",
                "category": "martyrs",
                "location": "济南市市中区英雄山路18号",
                "description": "安葬着1502名革命烈士，包括山东早期中共领导人刘谦初等，建有烈士纪念塔和悼念广场。",
                "history": "始建于1949年，1968年毛泽东题词'革命烈士纪念塔'。是济南规模最大的烈士陵园。",
                "open_time": "全天开放",
                "ticket_info": "免费",
                "short_desc": "济南规模最大的革命烈士陵园",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "五三惨案纪念园",
                "category": "memorial",
                "location": "济南市趵突泉公园内",
                "description": "含'五三'纪念碑、纪念亭和蔡公时纪念馆，纪念1928年日军制造的'五三惨案'。",
                "history": "1928年5月3日日军杀害中国外交官蔡公时及军民6000余人，此遗址为惨案见证地。",
                "open_time": "8:00-18:00",
                "ticket_info": "含在趵突泉门票内（40元）",
                "short_desc": "纪念1928年'五三惨案'的遗址",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "大峰山革命遗址",
                "category": "revolution",
                "location": "济南市长清区孝里镇",
                "description": "含中共长清县委旧址、八路军山东纵队第六支队成立旧址等革命遗迹群。",
                "history": "抗战时期长清县革命活动中心，被誉为'长清的小延安'，现存石屋、地道等遗址。",
                "open_time": "8:30-17:00",
                "ticket_info": "联票50元",
                "short_desc": "抗战时期长清县革命活动中心遗址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "莱芜战役纪念馆",
                "category": "memorial",
                "location": "济南市莱芜区英雄北路",
                "description": "全景展示1947年莱芜战役的专题纪念馆，含展览馆、全景画馆和烈士纪念碑。",
                "history": "1947年2月华东野战军歼灭国民党军5.6万余人，创运动战典范。纪念馆1997年建成。",
                "open_time": "8:30-17:00",
                "ticket_info": "免费",
                "short_desc": "展示1947年莱芜战役的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "辛锐纪念馆",
                "category": "memorial",
                "location": "济南市章丘区刁镇",
                "description": "纪念革命烈士辛锐的专题馆，展示其书画作品、生平事迹及抗战文物。",
                "history": "辛锐（1918-1941）为著名革命艺术家，在沂蒙山区反扫荡中壮烈牺牲。2014年在其家乡建馆。",
                "open_time": "9:00-16:30",
                "ticket_info": "免费",
                "short_desc": "纪念革命艺术家辛锐的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "历城革命烈士陵园",
                "category": "martyrs",
                "location": "济南市历城区英雄山路",
                "description": "安葬历城战役等各时期烈士1276名，建有22米高革命烈士纪念碑。",
                "history": "始建于1955年，2014年改造升级，集中展现历城区革命斗争史。",
                "open_time": "8:00-17:00",
                "ticket_info": "免费",
                "short_desc": "安葬1276名革命烈士的陵园",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "济南市党史陈列馆",
                "category": "museum",
                "location": "济南市历下区经十路21028号",
                "description": "系统展示济南党组织1921年成立以来的发展历程，含珍贵党史档案500余件。",
                "history": "2011年建党90周年时建成开放，2021年完成展陈升级，包括'齐鲁曙光''浴血奋战'等六大展区。",
                "open_time": "周二至周六 9:00-11:30,13:30-17:00",
                "ticket_info": "凭身份证免费参观",
                "short_desc": "展示济南党组织发展历程的专题馆",
                "tags": ["museum", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "泺口九烈士纪念碑",
                "category": "martyrs",
                "location": "济南市天桥区泺口街道黄河大坝南侧",
                "description": "纪念1933年8月被国民党杀害的9名中共山东地下党领导人，建有12米高纪念碑。",
                "history": "1933年山东省委遭破坏，李春亭、段亦民等9人在此英勇就义，1989年立碑纪念。",
                "open_time": "全天开放",
                "ticket_info": "免费",
                "short_desc": "纪念1933年牺牲的9名中共地下党领导人",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "平阴县委旧址纪念馆",
                "category": "revolution",
                "location": "济南市平阴县孔村镇",
                "description": "抗日战争时期平阴县委驻地，复原了县委办公室、地道等历史场景。",
                "history": "1939-1942年间中共平阴县委在此领导抗日斗争，现存石砌民居6间及抗战时期地道遗址。",
                "open_time": "8:30-17:00（周一闭馆）",
                "ticket_info": "团体预约制",
                "short_desc": "抗战时期平阴县委驻地旧址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "历城抗日纪念馆",
                "category": "memorial",
                "location": "济南市历城区西营镇",
                "description": "依托大南营战斗遗址建设，展示历城军民1937-1945年抗日斗争史实。",
                "history": "1944年八路军在此歼灭日军一个小队，现存战场遗址和22处抗战时期石刻标语。",
                "open_time": "9:00-16:30（冬季至16:00）",
                "ticket_info": "免费",
                "short_desc": "展示历城军民抗日斗争史的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "商河县革命烈士陵园",
                "category": "martyrs",
                "location": "济南市商河县青年路东首",
                "description": "安葬抗日战争、解放战争时期烈士863名，中央矗立19.49米高纪念碑。",
                "history": "始建于1955年，2015年改建，园内'商河革命历史展馆'详载本县重大战役史料。",
                "open_time": "8:00-18:00",
                "ticket_info": "免费",
                "short_desc": "安葬863名革命烈士的陵园",
                "tags": ["martyrs", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "钢城区莱芜战役指挥所旧址",
                "category": "revolution",
                "location": "济南市钢城区辛庄镇石湾子村",
                "description": "陈毅、粟裕在此指挥莱芜战役的四合院旧址，保留作战地图等历史场景。",
                "history": "1947年2月华东野战军前线指挥所设此，现为省级文保单位，院内保留百年古梨树。",
                "open_time": "8:30-17:00",
                "ticket_info": "门票15元",
                "short_desc": "莱芜战役前线指挥所旧址",
                "tags": ["revolution", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "济南乡师党史陈列室",
                "category": "education",
                "location": "济南市天桥区明湖西路",
                "description": "山东省立乡村师范学校旧址，山东党组织重要活动地，现存'红色乡师'专题展。",
                "history": "1932-1937年间该校培养200余名党员，赵健民、姚仲明等曾在此开展地下工作。",
                "open_time": "预约开放",
                "ticket_info": "免费",
                "short_desc": "山东党组织重要活动地旧址",
                "tags": ["education", "红色旅游"],
                "is_featured": True
            },
            {
                "name": "章丘第一支抗日武装纪念馆",
                "category": "memorial",
                "location": "济南市章丘区普集街道三山峪村",
                "description": "展示1937年成立的章丘人民抗日救国军斗争史，含武器、文件等实物展品。",
                "history": "1937年12月李曼村等在此建立章丘首支抗日武装，后编入八路军山东纵队。",
                "open_time": "9:00-16:30",
                "ticket_info": "免费",
                "short_desc": "展示章丘首支抗日武装斗争史的专题馆",
                "tags": ["memorial", "红色旅游"],
                "is_featured": True
            }
        ]

        created_count = 0
        skipped_count = 0
        image_added_count = 0

        for spot_data in red_spots_data:
            try:
                # 获取或创建城市
                city_name = get_city_name(spot_data['location'])
                city, _ = City.objects.get_or_create(
                    name=city_name,
                    defaults={
                        'code': city_name[:2].lower(),
                        'description': f"{city_name}红色旅游城市",
                        'is_hot': city_name in ['济南市', '青岛市', '烟台市']
                    }
                )

                # 处理门票价格
                ticket_price = parse_ticket_price(spot_data['ticket_info'])

                # 创建或获取景点
                attraction, created = Attraction.objects.get_or_create(
                    name=spot_data['name'],
                    city=city,
                    defaults={
                        'description': spot_data['description'],
                        'short_desc': spot_data['short_desc'],
                        'address': spot_data['location'],
                        'ticket_price': ticket_price,
                        'open_hours': spot_data['open_time'],
                        'tags': spot_data['tags'],
                        'history': spot_data['history'],
                        'is_featured': spot_data['is_featured']
                    }
                )

                # 处理景点图片
                image_path = os.path.join(IMAGE_BASE_PATH, f"{spot_data['name']}.png")
                if os.path.exists(image_path):
                    if not attraction.image:  # 如果景点没有图片则添加
                        with open(image_path, 'rb') as f:
                            attraction.image.save(
                                os.path.basename(image_path),
                                File(f),
                                save=True
                            )
                        image_added_count += 1
                        self.stdout.write(f"成功添加图片: {spot_data['name']}")
                else:
                    self.stdout.write(f"警告: 未找到图片 {image_path}")

                if created:
                    created_count += 1
                    self.stdout.write(f"创建景点: {spot_data['name']}")
                else:
                    skipped_count += 1
                    self.stdout.write(f"已存在景点: {spot_data['name']}")

            except Exception as e:
                self.stdout.write(self.style.ERROR(
                    f"处理景点 {spot_data['name']} 时出错: {str(e)}. 位置信息: {spot_data['location']}"
                ))
                continue

        self.stdout.write(
            self.style.SUCCESS(
                f'导入完成！成功创建 {created_count} 个景点，跳过 {skipped_count} 个已存在景点，添加 {image_added_count} 张图片'
            )
        )