# -*- coding: utf-8 -*-
from typing import Optional

from pydantic import BaseModel, Field


class ZhihuContent(BaseModel):
    """
    知乎内容（回答、文章、视频）
    """
    content_id: str = Field(default="", description="内容ID")
    content_type: str = Field(default="", description="内容类型(article | answer | zvideo)")
    content_text: str = Field(default="", description="内容文本, 如果是视频类型这里为空")
    content_url: str = Field(default="", description="内容落地链接")
    question_id: str = Field(default="", description="问题ID, type为answer时有值")
    title: str = Field(default="", description="内容标题")
    desc: str = Field(default="", description="内容描述")
    created_time: int = Field(default="", description="创建时间")
    updated_time: int = Field(default="", description="更新时间")
    voteup_count: int = Field(default=0, description="赞同人数")
    comment_count: int = Field(default=0, description="评论数量")
    source_keyword: str = Field(default="", description="来源关键词")

    user_id: str = Field(default="", description="用户ID")
    user_link: str = Field(default="", description="用户主页链接")
    user_nickname: str = Field(default="", description="用户昵称")
    user_avatar: str = Field(default="", description="用户头像地址")


class ZhihuComment(BaseModel):
    """
    知乎评论
    """

    comment_id: str = Field(default="", description="评论ID")
    parent_comment_id: str = Field(default="", description="父评论ID")
    content: str = Field(default="", description="评论内容")
    publish_time: int = Field(default=0, description="发布时间")
    ip_location: Optional[str] = Field(default="", description="IP地理位置")
    sub_comment_count: int = Field(default=0, description="子评论数")
    like_count: int = Field(default=0, description="点赞数")
    dislike_count: int = Field(default=0, description="踩数")
    content_id: str = Field(default="", description="内容ID")
    content_type: str = Field(default="", description="内容类型(article | answer | zvideo)")

    user_id: str = Field(default="", description="用户ID")
    user_link: str = Field(default="", description="用户主页链接")
    user_nickname: str = Field(default="", description="用户昵称")
    user_avatar: str = Field(default="", description="用户头像地址")


class ZhihuCreator(BaseModel):
    """
    知乎创作者
    """
    user_id: str = Field(default="", description="用户ID")
    user_link: str = Field(default="", description="用户主页链接")
    user_nickname: str = Field(default="", description="用户昵称")
    user_avatar: str = Field(default="", description="用户头像地址")
    gender: str = Field(default="", description="用户性别")
    ip_location: Optional[str] = Field(default="", description="IP地理位置")
    follows: int = Field(default=0, description="关注数")
    fans: int = Field(default=0, description="粉丝数")