xhs/config.py

53 lines
1.2 KiB
Python
Raw Normal View History

2024-05-11 15:46:56 +00:00
#!/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