mirror of
https://github.com/LibraHp/GetQzonehistory.git
synced 2024-12-27 06:29:55 +00:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
import configparser
|
||
import os
|
||
|
||
config = configparser.ConfigParser()
|
||
config.read('./resource/config/config.ini')
|
||
|
||
temp_path = config.get('File', 'temp')
|
||
user_path = config.get('File', 'user')
|
||
result_path = config.get('File', 'result')
|
||
|
||
|
||
def save_user(cookies):
|
||
with open(user_path + cookies.get('uin'), 'w') as f:
|
||
f.write(str(cookies))
|
||
|
||
|
||
def init_flooder():
|
||
# 初始化temp文件夹
|
||
if not os.path.exists(temp_path):
|
||
os.makedirs(temp_path)
|
||
print(f"Created directory: {temp_path}")
|
||
|
||
# 初始化user文件夹
|
||
if not os.path.exists(user_path):
|
||
os.makedirs(user_path)
|
||
print(f"Created directory: {user_path}")
|
||
|
||
# 初始化result文件夹
|
||
if not os.path.exists(result_path):
|
||
os.makedirs(result_path)
|
||
print(f"Created directory: {result_path}")
|
||
|
||
|
||
def read_files_in_folder():
|
||
# 获取文件夹下的所有文件
|
||
files = os.listdir(user_path)
|
||
# 如果文件夹为空
|
||
if not files:
|
||
return None
|
||
# 输出文件列表
|
||
print("已登录用户列表:")
|
||
for i, file in enumerate(files):
|
||
print(f"{i + 1}. {file}")
|
||
|
||
# 选择文件
|
||
while True:
|
||
try:
|
||
choice = int(input("请选择要登录的用户序号，重新登录输入0: "))
|
||
if 1 <= choice <= len(files):
|
||
break
|
||
elif choice == 0:
|
||
return None
|
||
else:
|
||
print("无效的选择，请重新输入。")
|
||
except ValueError:
|
||
print("无效的选择，请重新输入。")
|
||
|
||
# 读取选择的文件
|
||
selected_file = files[choice - 1]
|
||
file_path = os.path.join(user_path, selected_file)
|
||
with open(file_path, 'r') as file:
|
||
content = file.read()
|
||
|
||
return content
|