mirror of
https://github.com/LC044/WeChatMsg
synced 2024-11-15 06:11:19 +08:00
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import os
|
|
import winreg
|
|
|
|
from app.person_pc import MePC
|
|
from app.util import dat2pic
|
|
|
|
os.makedirs('./data/image', exist_ok=True)
|
|
|
|
|
|
def get_abs_path(path):
|
|
# return os.path.join(os.getcwd(), 'app/data/icons/404.png')
|
|
if path:
|
|
base_path = os.getcwd() + "/data/image"
|
|
output_path = dat2pic.decode_dat(os.path.join(MePC().wx_dir, path), base_path) # './data/image')
|
|
return output_path if output_path else ':/icons/icons/404.png'
|
|
else:
|
|
return ':/icons/icons/404.png'
|
|
|
|
|
|
def mkdir(path):
|
|
if not os.path.exists(path):
|
|
os.mkdir(path)
|
|
|
|
|
|
def wx_path():
|
|
try:
|
|
## 获取当前用户名
|
|
user_home = os.environ.get("USERPROFILE")
|
|
## 找到3ebffe94.ini配置文件
|
|
f = open(user_home + '\\AppData\\Roaming\\Tencent\\WeChat\\All Users\\config\\3ebffe94.ini', encoding='utf-8')
|
|
txt = f.read()
|
|
f.close()
|
|
# 打开Windows注册表
|
|
reg_key = winreg.OpenKey(winreg.HKEY_CURRENT_USER,
|
|
"Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders")
|
|
# 获取“我的文档”路径的注册表键值
|
|
documents_path_value = winreg.QueryValueEx(reg_key, "Personal")
|
|
# 输出路径
|
|
##读取文件将路径放到wx_location变量里
|
|
if txt == 'MyDocument:':
|
|
wx_location = documents_path_value[0] + '\WeChat Files'
|
|
else:
|
|
wx_location = txt + "\WeChat Files"
|
|
return wx_location
|
|
except FileNotFoundError:
|
|
return '.'
|
|
|