import os
import shutil

def export_source_code(src_dir, dst_dir, extensions=('.py',)):
    """導(dǎo)出指定擴(kuò)展名的源代碼文件"""
    if not os.path.exists(dst_dir):
        os.makedirs(dst_dir)
    
    for root, _, files in os.walk(src_dir):
        for file in files:
            if file.endswith(extensions):
                src_path = os.path.join(root, file)
                # 保持相對(duì)目錄結(jié)構(gòu)
                rel_path = os.path.relpath(src_path, src_dir)
                dst_path = os.path.join(dst_dir, rel_path)
                
                # 創(chuàng)建目錄（如果不存在）
                os.makedirs(os.path.dirname(dst_path), exist_ok=True)
                
                # 復(fù)制文件
                shutil.copy2(src_path, dst_path)
                print(f"Exported: {src_path} -> {dst_path}")

# 使用示例：導(dǎo)出所有.py和.html文件
export_source_code('./my_project', './exported_code', ('.py', '.html'))

import os
import zipfile
from getpass import getpass

def zip_source_code(src_dir, zip_path, extensions=('.py',)):
    """將源代碼打包為ZIP文件"""
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, _, files in os.walk(src_dir):
            for file in files:
                if file.endswith(extensions):
                    file_path = os.path.join(root, file)
                    # 在ZIP中創(chuàng)建相同目錄結(jié)構(gòu)
                    arcname = os.path.relpath(file_path, src_dir)
                    zipf.write(file_path, arcname)
                    print(f"Added to zip: {arcname}")

# 使用示例
zip_source_code('./my_project', './code_backup.zip')

def create_encrypted_zip(src_dir, zip_path):
    """創(chuàng)建加密的ZIP文件（需要Python 3.6+）"""
    password = getpass("Enter ZIP password: ").encode('utf-8')
    
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        # 先添加普通文件
        for root, _, files in os.walk(src_dir):
            for file in files:
                if file.endswith('.py'):
                    file_path = os.path.join(root, file)
                    arcname = os.path.relpath(file_path, src_dir)
                    zipf.write(file_path, arcname)
        
        # 設(shè)置密碼（實(shí)際加密需要第三方庫(kù)如pyzipper）
        # 這里僅演示標(biāo)準(zhǔn)庫(kù)的局限性
        print("注意：標(biāo)準(zhǔn)zipfile不支持AES加密，建議使用pyzipper庫(kù)")

def convert_to_utf8(file_path):
    """將文件轉(zhuǎn)換為UTF-8編碼"""
    try:
        with open(file_path, 'r', encoding='utf-8') as f:
            content = f.read()
    except UnicodeDecodeError:
        # 嘗試其他常見編碼
        for encoding in ['gbk', 'latin1', 'big5']:
            try:
                with open(file_path, 'r', encoding=encoding) as f:
                    content = f.read()
                break
            except UnicodeDecodeError:
                continue
        else:
            print(f"Warning: Could not decode {file_path}")
            return
    
    # 重新寫入U(xiǎn)TF-8
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(content)
    print(f"Converted to UTF-8: {file_path}")

def should_exclude(file_path, exclude_patterns):
    """檢查文件是否應(yīng)被排除"""
    for pattern in exclude_patterns:
        if pattern in file_path:
            return True
    return False

# 使用示例
exclude_list = ['config.py', 'secrets/', '__pycache__/']
for root, _, files in os.walk('./my_project'):
    for file in files:
        file_path = os.path.join(root, file)
        if should_exclude(file_path, exclude_list):
            print(f"Excluded (sensitive): {file_path}")
            continue
        # 處理其他文件...

def is_binary_file(file_path):
    """簡(jiǎn)單判斷是否為二進(jìn)制文件"""
    try:
        with open(file_path, 'rb') as f:
            chunk = f.read(1024)
            if b'\x00' in chunk:  # 常見二進(jìn)制標(biāo)志
                return True
            # 簡(jiǎn)單檢查ASCII范圍
            if all(32 <= ord(c) < 127 for c in chunk.decode('ascii', errors='ignore')):
                return False
            return True
    except:
        return True

# 使用示例
if not is_binary_file('image.png'):
    # 處理文本文件
    pass
else:
    # 處理二進(jìn)制文件
    pass

def generate_toc(root_dir, output_file):
    """生成Markdown格式的目錄"""
    with open(output_file, 'w', encoding='utf-8') as f:
        f.write("# 代碼目錄\n\n")
        for root, _, files in os.walk(root_dir):
            for file in files:
                if file.endswith('.py'):
                    rel_path = os.path.relpath(os.path.join(root, file), root_dir)
                    # 替換路徑分隔符為Markdown鏈接格式
                    md_path = rel_path.replace('\', '/')
                    f.write(f"- [{md_path}]({md_path})\n")
    print(f"TOC generated at {output_file}")

from fpdf import FPDF

def code_to_pdf(input_file, output_pdf):
    """將單個(gè)代碼文件導(dǎo)出為PDF"""
    pdf = FPDF()
    pdf.add_page()
    pdf.set_font("Courier", size=10)  # 等寬字體適合代碼
    
    with open(input_file, 'r', encoding='utf-8') as f:
        for line in f:
            pdf.cell(0, 5, txt=line.rstrip(), ln=True)
    
    pdf.output(output_pdf)
    print(f"PDF generated: {output_pdf}")

# 使用示例（需先導(dǎo)出單個(gè)文件）
code_to_pdf('./my_project/main.py', './main_code.pdf')

import ast
import os

def find_imports(file_path):
    """解析Python文件中的import語(yǔ)句"""
    imports = set()
    with open(file_path, 'r', encoding='utf-8') as f:
        tree = ast.parse(f.read())
    
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            for alias in node.names:
                imports.add(alias.name.split('.')[0])
        elif isinstance(node, ast.ImportFrom):
            imports.add(node.module.split('.')[0])
    
    return imports

# 使用示例
imports = find_imports('./my_project/main.py')
print(f"Imports in main.py: {imports}")

import os
import shutil
import zipfile
from datetime import datetime

class CodeExporter:
    def __init__(self, src_dir, exclude_patterns=None):
        self.src_dir = os.path.abspath(src_dir)
        self.exclude_patterns = exclude_patterns or []
        self.export_time = datetime.now().strftime("%Y%m%d_%H%M%S")
    
    def should_exclude(self, file_path):
        for pattern in self.exclude_patterns:
            if pattern in file_path:
                return True
        return False
    
    def export_files(self, dst_dir, extensions=('.py', '.html', '.js', '.css')):
        """導(dǎo)出文件到目錄"""
        if not os.path.exists(dst_dir):
            os.makedirs(dst_dir)
        
        exported_files = []
        for root, _, files in os.walk(self.src_dir):
            for file in files:
                if file.endswith(extensions):
                    file_path = os.path.join(root, file)
                    if self.should_exclude(file_path):
                        print(f"Excluded: {file_path}")
                        continue
                    
                    rel_path = os.path.relpath(file_path, self.src_dir)
                    dst_path = os.path.join(dst_dir, rel_path)
                    
                    os.makedirs(os.path.dirname(dst_path), exist_ok=True)
                    shutil.copy2(file_path, dst_path)
                    exported_files.append(dst_path)
                    print(f"Exported: {dst_path}")
        
        return exported_files
    
    def zip_exported(self, dst_dir, zip_name):
        """將導(dǎo)出的目錄打包為ZIP"""
        zip_path = os.path.join(dst_dir, f"{zip_name}_{self.export_time}.zip")
        with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
            for root, _, files in os.walk(dst_dir):
                for file in files:
                    if file.endswith('.zip'):  # 避免打包自己
                        continue
                    file_path = os.path.join(root, file)
                    arcname = os.path.relpath(file_path, dst_dir)
                    zipf.write(file_path, arcname)
        print(f"Zip created: {zip_path}")
        return zip_path
    
    def generate_toc(self, exported_files, toc_path):
        """生成Markdown目錄"""
        with open(toc_path, 'w', encoding='utf-8') as f:
            f.write(f"# 代碼導(dǎo)出目錄 ({self.export_time})\n\n")
            for file in sorted(exported_files):
                rel_path = os.path.relpath(file, os.path.dirname(toc_path))
                md_path = rel_path.replace('\', '/')
                f.write(f"- [{md_path}]({md_path})\n")
        print(f"TOC generated: {toc_path}")

# 使用示例
if __name__ == "__main__":
    exporter = CodeExporter(
        src_dir='./my_project',
        exclude_patterns=['config.py', 'secrets/', '__pycache__/']
    )
    
    # 創(chuàng)建導(dǎo)出目錄
    export_dir = f"./exported_code_{exporter.export_time}"
    os.makedirs(export_dir, exist_ok=True)
    
    # 導(dǎo)出文件
    exported_files = exporter.export_files(export_dir)
    
    # 生成目錄
    exporter.generate_toc(exported_files, os.path.join(export_dir, 'TOC.md'))
    
    # 打包為ZIP
    exporter.zip_exported(export_dir, "my_project_source")