# 初始化項(xiàng)目結(jié)構(gòu)
mkdir my_package && cd my_package
touch setup.py  # 包含包元數(shù)據(jù)（名稱(chēng)、版本、依賴(lài)等）

# 安裝可編輯模式
pip install -e .

# pyproject.toml 示例
[build-system]
requires = ["setuptools>=42", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "my_package"
version = "0.1.0"
dependencies = ["requests>=2.25"]

# 使用 PEP 660 標(biāo)準(zhǔn)安裝
pip install -e .

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "Tdouya-AI-Assistant"          # ?? 用戶(hù)指定項(xiàng)目名稱(chēng)
version = "0.1.1"
authors = [{name = "田辛老師", email = "tianxin.xp@gmail.com"}]  # ?? 用戶(hù)信息
description = "基于AI的智能助手開(kāi)發(fā)框架"
readme = "README.md"
requires-python = ">=3.12"

[tool.setuptools]
package-dir = { "" = "src" }  # 聲明包根目錄在src下

from setuptools import setup, find_packages

setup(
    name="Tdouya-AI-Assistant",  # ?? 必須與pyproject.toml一致
    version="0.1.1",
    author="田辛老師",
    author_email="tianxin.xp@gmail.com",
    description="基于AI的智能助手開(kāi)發(fā)框架",
    packages=find_packages(where="src"),  # 指定包搜索路徑
    package_dir={"": "src"},  # 聲明包根目錄
    install_requires=[
        "python-dotenv>=0.19.0", 
        "requests>=2.21.0",  
        "streamlit>=1.0.0", 
        "moviepy>=1.0.3",  
        "colorlog>=6.7.0",
        "langchain_ollama",
        "langchain_core",
        "langchain_community",
        "ollama",
        "dashscope"
    ],
    python_requires=">=3.12",
)

pip install -e .

# ==============================================
# 文件名: init.py
# 描述: 此腳本用于查找項(xiàng)目根路徑，若項(xiàng)目未初始化則進(jìn)行初始化操作，已初始化則給出提示。
# 作者: 田辛
# 創(chuàng)建日期: 2025-05-07
# ==============================================
# 導(dǎo)入操作系統(tǒng)相關(guān)功能模塊，用于文件和目錄操作
import os

# 導(dǎo)入子進(jìn)程模塊，用于創(chuàng)建新進(jìn)程執(zhí)行外部命令
import subprocess

# 導(dǎo)入系統(tǒng)相關(guān)功能模塊，用于獲取Python解釋器信息
import sys

# 導(dǎo)入日期時(shí)間模塊，用于獲取當(dāng)前時(shí)間
import datetime

# 獲取當(dāng)前腳本所在的絕對(duì)路徑目錄
script_dir = os.path.dirname(os.path.abspath(__file__))
# 初始化當(dāng)前搜索目錄為腳本所在目錄
current_dir = script_dir

while True:
    # 定義項(xiàng)目已初始化時(shí)所需的文件列表
    required_files = [
        "setup.py",
        "pyproject.toml",
        "README.md",
        "CONTRIBUTING.md",
        "installed.lock",
    ]
    # 檢查當(dāng)前目錄是否存在所有已初始化所需的文件
    if all(os.path.isfile(os.path.join(current_dir, f)) for f in required_files):
        print(
            f"田豆芽AI助手已安裝， 若想再次安裝請(qǐng)先卸載。卸載命令：python {script_dir}/uninstall.py"
        )
        break
    # 定義項(xiàng)目未初始化時(shí)所需的文件列表
    original_required_files = [
        "setup.py",
        "pyproject.toml",
        "README.md",
        "CONTRIBUTING.md",
    ]
    # 檢查當(dāng)前目錄是否存在所有未初始化所需的文件
    if all(
        os.path.isfile(os.path.join(current_dir, f)) for f in original_required_files
    ):
        try:
            # 切換當(dāng)前工作目錄到項(xiàng)目根路徑
            os.chdir(current_dir)
            # 使用當(dāng)前Python解釋器執(zhí)行 `pip install -e .` 命令進(jìn)行項(xiàng)目安裝
            result = subprocess.run(
                [sys.executable, "-m", "pip", "install", "-e", "."],
                capture_output=True,
                text=True,
                check=True,
            )
            print(result.stdout)
            # 生成鎖定文件路徑
            lock_file_path = os.path.join(current_dir, "installed.lock")
            # 打開(kāi)鎖定文件并寫(xiě)入當(dāng)前時(shí)間戳
            with open(lock_file_path, "w") as f:
                f.write(datetime.datetime.now().strftime("%Y%m%d %H:%M:%S"))
            print(f"田豆芽AI助手安裝完成。")
        except subprocess.CalledProcessError as e:
            # 捕獲子進(jìn)程執(zhí)行錯(cuò)誤并輸出錯(cuò)誤信息
            print(f"執(zhí)行 `pip install -e .` 命令時(shí)出錯(cuò): {e.stderr}")
        except Exception as e:
            # 捕獲其他異常并輸出錯(cuò)誤信息
            print(f"執(zhí)行命令時(shí)出錯(cuò): {e}")
        break
    else:
        # 獲取當(dāng)前目錄的上一級(jí)目錄
        new_dir = os.path.dirname(current_dir)
        # 檢查是否已經(jīng)回到根驅(qū)動(dòng)器，如果是則表示未找到項(xiàng)目根路徑
        if new_dir == current_dir[:3]:
            print("未找到項(xiàng)目根路徑。")
            break
        # 更新當(dāng)前搜索目錄為上一級(jí)目錄
        current_dir = new_dir

# ==============================================
# 文件名: uninstall.py
# 描述: 此腳本用于查找項(xiàng)目根路徑，執(zhí)行卸載操作并刪除鎖定文件。
# 作者: 田辛
# 創(chuàng)建日期: 2025-05-07
# ==============================================
import os
import subprocess
import sys

# 獲取當(dāng)前腳本所在的絕對(duì)路徑目錄
script_dir = os.path.dirname(os.path.abspath(__file__))
# 初始化當(dāng)前搜索目錄為腳本所在目錄
current_dir = script_dir

while True:
    # 定義項(xiàng)目已初始化時(shí)所需的文件列表
    required_files = [
        "setup.py",
        "pyproject.toml",
        "README.md",
        "CONTRIBUTING.md",
        "installed.lock",
    ]
    # 檢查當(dāng)前目錄是否存在所有已初始化所需的文件
    if all(os.path.isfile(os.path.join(current_dir, f)) for f in required_files):
        try:
            # 切換當(dāng)前工作目錄到項(xiàng)目根路徑
            os.chdir(current_dir)
            # 執(zhí)行pip uninstall命令
            result = subprocess.run(
                [sys.executable, "-m", "pip", "uninstall", "-y", "Tdouya-AI-Assistant"],
                capture_output=True,
                text=True,
                check=True,
            )
            print(result.stdout)
            # 刪除鎖定文件
            lock_file_path = os.path.join(current_dir, "installed.lock")
            if os.path.isfile(lock_file_path):
                os.remove(lock_file_path)

            print("田豆芽AI助手卸載成功。")
        except subprocess.CalledProcessError as e:
            # 捕獲子進(jìn)程執(zhí)行錯(cuò)誤并輸出錯(cuò)誤信息
            print(f"執(zhí)行 `pip uninstall Tdouya-AI-Assistant` 命令時(shí)出錯(cuò): {e.stderr}")
        except Exception as e:
            # 捕獲其他異常并輸出錯(cuò)誤信息
            print(f"執(zhí)行命令時(shí)出錯(cuò): {e}")
        break
    else:
        # 獲取當(dāng)前目錄的上一級(jí)目錄
        new_dir = os.path.dirname(current_dir)
        # 檢查是否已經(jīng)回到根驅(qū)動(dòng)器，如果是則表示未找到項(xiàng)目根路徑
        if new_dir == current_dir[:3]:
            print("未找到項(xiàng)目根路徑。")
            break
        # 更新當(dāng)前搜索目錄為上一級(jí)目錄
        current_dir = new_dir