from agent.agent import Agent

class ApiService:
    @staticmethod
    def keyword_judgement(url_data, basic_data):
        """关键词引流判断"""
        if url_data['tilte'] == None or url_data['brand_name'] == None or basic_data['brand_name'] == None:
            return '信息不全，无法判定'
        
        # 标题判定
        if basic_data['brand_name'] in url_data['title'] and url_data['brand_name'] != basic_data['brand_name']:
            # 当产品名称中包含与判定商标一致的品牌名称，且详情页中的品牌名称与判定的品牌名称不一致时判定为关键词引流
            return True
        elif basic_data['brand_name'] in url_data['title'] and url_data['brand_name'] == basic_data['brand_name']:
            return False
        else:
            # 图像判定
            pass
        
        # 图像判定
        return "无法判定"
    
    @staticmethod
    def similarity_logo_judgement(url_data, basic_data):
        """近似商标侵权判定"""
        for similarity_logo in basic_data['similarity_logos']:
            if similarity_logo in url_data['title']:
                return True
            elif similarity_logo in url_data['brand_name']:
                return True
            
        # 判断基础数据中的名称是否包含在标题或者详情品牌名称中
        if basic_data['brand_name'] in url_data['title'] or basic_data['brandname'] in url_data['brand_name']:
            return False
        else:
            return '无法判定'
        
    @staticmethod
    def logo_1to1_judgement(url_data, basic_data):
        """商标一比一侵权"""
        # 首先计算基础价格百分比
        base_price = basic_data['base_price'] * basic_data['base_percent']

        if url_data['title'].strip() == basic_data['product_name'].strip():
            # 当商品名称与提供的产品名称完全一致时，进行判定
            if url_data < base_price:
                return True
            else:
                return False
        else:
            # 进行图像判定
            pass
        
    @staticmethod
    def low_price_judgement(url_data, basic_data):
        """低价判断"""
        if url_data['title'] == basic_data['product_name']:
            if url_data['price'] < basic_data['base_price']:
                return True
            else:
                return False
        else:
            # 进行图像判定
            pass