Skip to content

Remove file listing and moving functions #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
21sec wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Remove file listing and moving functions #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 1 addition & 41 deletions lesson_06/ai_organizer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,46 +20,6 @@
# 获取脚本所在目录的绝对路径，确保不跑偏
BASE_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), TARGET_DIR_NAME)

def list_files(args=None):
"""获取目标文件夹下的所有文件名"""
if not os.path.exists(BASE_PATH):
return json.dumps({"error": f"找不到目录 {TARGET_DIR_NAME}，请先运行生成脚本。"})

files = []
# 遍历目录
for f in os.listdir(BASE_PATH):
f_path = os.path.join(BASE_PATH, f)
# 只看文件，忽略隐藏文件和Python脚本
if os.path.isfile(f_path) and not f.startswith('.'):
files.append(f)

return json.dumps({"files": files}, ensure_ascii=False)

def move_file(args):
"""移动文件到指定分类文件夹"""
filename = args.get("filename")
category = args.get("category")

if not filename or not category:
return json.dumps({"error": "参数缺失"})

source_file = os.path.join(BASE_PATH, filename)
target_folder = os.path.join(BASE_PATH, category)
target_file = os.path.join(target_folder, filename)

try:
if not os.path.exists(source_file):
return json.dumps({"error": f"文件 {filename} 不存在"})

if not os.path.exists(target_folder):
os.makedirs(target_folder) # 自动创建分类文件夹

shutil.move(source_file, target_file)
return json.dumps({"status": "success", "moved_to": category})
except Exception as e:
return json.dumps({"error": str(e)})


# ==========================================
# Agent 主程序 (大脑与循环)
# ==========================================
Expand Down Expand Up @@ -154,4 +114,4 @@ def run_agent():
break

if __name__ == "__main__":
run_agent()
run_agent()