AIServ/services/templateData.py

import math
import random
from typing import List, Dict, Optional

#首页数据
class indexData:
    def __init__(self):

        # 通栏权重设置
        self.sectors_config = [
            {"name": "headLineSector", "total": 1, "weight": 10 ,"CNname":"网站头条" , "random": False},
            {"name": "bannerSector", "total": 1, "weight": 9 ,"CNname":"焦点图" , "random": False},
            {"name": "linkSector", "total": 1, "weight": 8 ,"CNname":"外链面板" , "random": False},
            {"name": "manyPictureSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合1" , "random": True},
            {"name": "commentSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合2" , "random": True},
            {"name": "listSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合3" , "random": True},
            {"name": "onlyImgSector", "total": 2, "weight": 7 ,"CNname":"带广告新闻组合" , "random": True},
            {"name": "adSector", "total": 4, "weight": 6 ,"CNname":"通栏广告" , "random": False}
        ]

        # 通栏数据
        self.sectors_data = {
            #广告通栏
            "adSector": {
                "sectorName": "adSector",
                "sectorCanvasHeight": 12, #占据画布高度
                "componentList": [
                    {
                        "component_type": 2,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {}
                    }
                ],
                "ad": {
                    "width": 1200,
                    "height": 90,
                    "name": "",
                    "price": 0,
                    "introduce":"",
                    "website_id": "",
                    "thumb": "https://img.bjzxtw.org.cn/pre/image/png/20250530/1748588901281358.png",
                    "typeid": 2,
                    "ad_tag": ""
                }
            },
            #头条通栏
            "headLineSector": {
                "sectorName": "headLineSector",
                "sectorCanvasHeight": 17,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "level": 1,
                            "imgSize": 0,
                            "textSize": 4,
                            "child": {
                                "id": "",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    }
                ]
            },
            #轮播图通栏
            "bannerSector": {
                "sectorName": "bannerSector",
                "sectorCanvasHeight": 44,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "level": 2,
                            "imgSize": 5,
                            "textSize": 0,
                            "child": {
                                "id": "",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 2,
                        "componentData": {
                            "level": 6,
                            "imgSize": 0,
                            "textSize": 10,
                            "child": {
                                "id": "",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    }
                ]
            },
            #新闻组合通栏1
            "manyPictureSector": {
                "sectorName": "manyPictureSector",
                "sectorCanvasHeight": 47,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 3,
                            "textSize": 9,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 2,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 3,
                            "textSize": 6,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    }
                ]
            },
            #新闻组合通栏2
            "commentSector": {
                "sectorName": "commentSector",
                "sectorCanvasHeight": 47,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 2,
                            "textSize": 12,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 2,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 1,
                            "textSize": 3,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    }
                ]
            },
            #新闻组合通栏3
            "listSector": {
                "sectorName": "listSector",
                "sectorCanvasHeight": 98,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 2,
                            "textSize": 6,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 2,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 2,
                            "textSize": 14,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 3,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 7,
                            "textSize": 0,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    }
                ]
            },
            #广告新闻混合通栏
            "onlyImgSector": {
                "sectorName": "onlyImgSector",
                "sectorCanvasHeight": 51,
                "componentList": [
                    {
                        "component_type": 1,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {
                            "category_id":"",
                            "category_arr":"",
                            "name":"请选择导航..",
                            "level":"",
                            "imgSize": 3,
                            "textSize": 4,
                            "child":{
                                "id":"",
                                "imgSize": "",
                                "textSize": ""
                            },
                            "listType": [
                                "id",
                                "title",
                                "imgurl",
                                "author",
                                "updated_at",
                                "introduce",
                                "islink",
                                "linkurl",
                                "copyfrom",
                                "cat_arr_id",
                                "catid",
                                "pinyin"
                            ],
                        }
                    },
                ],
                "ad": {
                    "width": 450,
                    "height": 290,
                    "name": "",
                    "price": 0,
                    "introduce":"",
                    "website_id": "",
                    "thumb": "http://img.bjzxtw.org.cn/pre/image/png/20250609/174945725555092.png",
                    "typeid": 2,
                    "ad_tag": ""
                }
            },
            #外链通栏
            "linkSector": {
                "sectorName": "linkSector",
                "sectorCanvasHeight": 26,
                "componentList": [
                    {
                        "component_type": 3,
                        "component_style": 1,
                        "sort": 1,
                        "componentData": {}
                    }
                ]
            },
        }
        #模板数据
        self.sectors_webSiteData = {
            "base": {
                "websiteId": ""
            },
            "style": {
                "styleId": ""
            },
            "template": {
                "index": [],
                "class": [],
                "list": [],
                "article": [],
                "search": [],
                "aloneList": [],
                "aloneArticle": []
            },
            "ad": {
                "top":{
                    "width": 830,
                    "height": 110,
                    "name": "",
                    "price": 0,
                    "introduce":"",
                    "website_id": "",
                    "thumb": "http://img.bjzxtw.org.cn/pre/image/png/20250527/1748332370111555.png",
                    "typeid": 2,
                    "ad_tag": ""
                },
                "index": [],
                "class": [],
                "list": [],
                "article": [],
                "search": [],
                "aloneList": [],
                "aloneArticle": []
            },
            "isSearch": True
        }

        #画布数据
        self.canvas_data = {
            "topAd":{
                "width": 830,
                "height": 110,
                "name": "",
                "price": "",
                "introduce": "",
                "website_id": "",
                "thumb": "http://192.168.1.234:19000/pre/image/jpeg/20251226/1766715237727691.jpg",
                "typeid": 2,
                "ad_tag": "",
                "ad_url": ""
            },
            "template":{
                "index":[],
                "class": [],
                "list": [],
                "article": [],
                "search": [],
                "aloneList": [],
                "aloneArticle": []
            }
        }

    #初始化通栏权重
    def reset_sectors_config(self):
        self.sectors_config = [
            {"name": "headLineSector", "total": 1, "weight": 10 ,"CNname":"网站头条"},
            {"name": "bannerSector", "total": 1, "weight": 9 ,"CNname":"焦点图"},
            {"name": "linkSector", "total": 1, "weight": 8 ,"CNname":"外链面板"},
            {"name": "manyPictureSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合1"},
            {"name": "commentSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合2"},
            {"name": "listSector", "total": 2, "weight": 7 ,"CNname":"新闻图文组合3"},
            {"name": "onlyImgSector", "total": 2, "weight": 7 ,"CNname":"带广告新闻组合"},
            {"name": "adSector", "total": 4, "weight": 6 ,"CNname":"通栏广告"}
        ]

#创建一个templateData实例   
templateData = indexData()

# 动态权重计算类
class Sector:
    def __init__(self, name: str, total: int, weight: int, CNname: str):
        self.name = name # 名称
        self.total = total # 总数
        self.weight = weight # 权重
        self.remaining = total # 剩余数量 后面我们要对比 剩余数据/总数 来计算权重
        self.used_count = 0  # 记录已使用次数
        self.CNname = CNname # 中文名称

"""
    思路：
    1.按照权限的高低选择通栏
    2.如果权限一样，随机选择
    3.每选择两次新闻通栏后,adSector权重提升
"""
class SectorScheduler:
    def __init__(self, sectors_config: List[Dict]):
        # 过滤掉total为0的sector（即不存在的sector）
        self.sectors = [
            Sector(sector['name'], sector['total'], sector['weight'], sector['CNname']) 
            for sector in sectors_config 
            if sector['total'] > 0
        ]
        self.history = []  # 记录选择历史
        self.consecutive_count = 0  # 连续选择相同权重sector的次数
        self.last_weight = None  # 上一次选择的权重
        
        # 定义需要随机选择的sector组
        self.random_sector_group = {sector['name'] for sector in sectors_config if sector.get('random', False)}
    
    def calculate_dynamic_weight(self, sector: Sector) -> float:
        """计算动态权重 - 更复杂的权重计算"""
        if sector.remaining <= 0:
            return 0
        
        base_weight = sector.weight
        
        # 1. 基于剩余数量的权重调整
        usage_ratio = sector.remaining / sector.total
        quantity_factor = math.pow(usage_ratio, 0.4)  # 剩余数量越多，权重相对越高
        
        # 2. 基于使用频率的权重调整（避免过度使用）
        if sector.used_count > 0:
            usage_frequency = 1.0 / (1 + math.log(1 + sector.used_count))
        else:
            usage_frequency = 1.0
        
        # 3. 对于adSector的特殊处理
        if sector.name == "adSector":
            # 检查是否需要插入adSector（每两个相同权重后）
            if self.consecutive_count >= 2 and self.last_weight == 7:
                ad_boost = 2.0  # 大幅提升adSector权重
            else:
                # 根据历史中adSector的出现频率调整
                recent_ad_count = self.history[-3:].count("adSector") if len(self.history) >= 3 else 0
                if recent_ad_count == 0:
                    ad_boost = 1.5  # 一段时间没出现，适当提升
                else:
                    ad_boost = 0.8  # 最近出现较多，适当降低
        else:
            ad_boost = 1.0
        
        # 4. 对于高权重sector的保护（确保它们优先被选择）
        if base_weight >= 8:
            priority_boost = 1.5
        else:
            priority_boost = 1.0
        
        dynamic_weight = base_weight * quantity_factor * usage_frequency * ad_boost * priority_boost
        
        return dynamic_weight
    
    def get_random_sector_from_group(self, target_weight: int) -> Optional[Sector]:
        """从随机组中随机选择一个指定权重的sector"""
        available_sectors = [
            sector for sector in self.sectors 
            if sector.name in self.random_sector_group 
            and sector.weight == target_weight 
            and sector.remaining > 0
        ]
        
        if available_sectors:
            return random.choice(available_sectors)
        return None
    
    def select_next(self) -> Optional[str]:
        """选择下一个sector"""
        best_sector = None
        best_weight = -1
        
        current_weight = 7  # 随机选择组的权重
        
        # 首先检查是否需要强制插入adSector
        if (self.consecutive_count >= 2 and self.last_weight == current_weight and
            any(s.name == "adSector" and s.remaining > 0 for s in self.sectors)):
            
            ad_sector = next(s for s in self.sectors if s.name == "adSector" and s.remaining > 0)
            ad_sector.remaining -= 1
            ad_sector.used_count += 1
            
            # 更新历史记录
            self.history.append("adSector")
            self.consecutive_count = 0
            self.last_weight = ad_sector.weight
            
            return "adSector"
        
        # 正常选择逻辑
        for sector in self.sectors:
            if sector.remaining > 0:
                # 对于随机选择组的sector，使用统一的权重计算
                if sector.name in self.random_sector_group and sector.weight == current_weight:
                    # 使用组内统一的权重值进行比较
                    dynamic_weight = current_weight
                else:
                    dynamic_weight = self.calculate_dynamic_weight(sector)
                
                if dynamic_weight > best_weight:
                    best_weight = dynamic_weight
                    best_sector = sector
        
        if best_sector:
            # 如果最佳sector属于随机选择组，则随机选择该组中的一个
            if best_sector.name in self.random_sector_group and best_sector.weight == current_weight:
                random_sector = self.get_random_sector_from_group(current_weight)
                if random_sector:
                    best_sector = random_sector
            
            best_sector.remaining -= 1
            best_sector.used_count += 1
            
            # 更新连续选择计数
            if self.last_weight == best_sector.weight and best_sector.name != "adSector":
                self.consecutive_count += 1
            else:
                self.consecutive_count = 1
            
            self.last_weight = best_sector.weight
            self.history.append(best_sector.name)
            
            return best_sector.name
        
        return None
    
    def generate_sequence(self, count: int) -> List[str]:
        """生成指定长度的序列"""
        result = []
        for i in range(count):
            next_sector = self.select_next()
            if next_sector:
                result.append(next_sector)
            else:
                break  # 没有可选的sector了
        return result