#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project ：redbook 
@File    ：config.py
@IDE     ：PyCharm 
@Author  ：rengengchen
@Time    ：2024/4/17 16:42 
"""
import os
import sys


def resource_path(relative_path):
    """ 获取资源的绝对路径。用于访问打包环境中的资源文件。 """
    try:
        # PyInstaller 创建的临时文件夹
        base_path = sys._MEIPASS
    except Exception:
        # 正常运行时的路径
        base_path = os.path.abspath(".")

    return os.path.join(base_path, relative_path)


HEAD = {
    'Host': 'edith.xiaohongshu.com',
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome...',
}
INFO_COLUMNS = ['nickname', 'content', 'id', 'user_id', 'status', 'liked', 'create_time', 'note_id']
RESULT_COLUMNS = ['nickname', 'content', 'ip_location', 'level', 'user_id', 'status', 'liked', 'create_time', 'note_id',
                  'id']
KEYWORDS = []
RESULT_PATH = os.path.abspath('./')
ENCODING = 'gbk'


def modify_result_path(v):
    global RESULT_PATH
    RESULT_PATH = v


def modify_filter_words(v):
    global KEYWORDS
    for v1 in v:
        if v1:
            KEYWORDS.append(v1)


def modify_encoding(v):
    global ENCODING
    ENCODING = v