xhs/config.py

53 lines
1.2 KiB
Python
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#!/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