python 放入必备的几个代码示例

python 放入必备的几个代码示例

1. 文件批量重命名工具import os def batch_rename_files(folder_path, prefix): """ 批量重命名指定文件夹中的所有文件 :param folder_path: 文件夹路径 :param prefix: 新文件名前缀 """ if not os.path.isdir(folder_path): print(f"错误: {folder_path} 不是一个有效的文件夹") return file_list = os.listdir(folder_path) count = 1 for file_name in file_list: old_path = os.path.join(folder_path, file_name) # 跳过文件夹，只处理文件 if os.path.isdir(old_path): continue # 获取文件扩展名 file_ext = os.path.splitext(file_name)[1] # 构建新文件名 new_file_name = f"{prefix}_{count:03d}{file_ext}" new_path = os.path.join(folder_path, new_file_name) # 重命名文件 os.rename(old_path, new_path) print(f"重命名: {file_name} -> {new_file_name}") count += 1 # 使用示例 if __name__ == "__main__": batch_rename_files("/path/to/your/files", "file")
2. 简单的文本内容搜索工具import os def search_text_in_files(folder_path, search_text, case_sensitive=False): """ 在指定文件夹的所有文件中搜索文本 :param folder_path: 文件夹路径 :param search_text: 要搜索的文本 :param case_sensitive: 是否区分大小写 """ if not os.path.isdir(folder_path): print(f"错误: {folder_path} 不是一个有效的文件夹") return # 遍历文件夹中的所有文件 for root, dirs, files in os.walk(folder_path): for file in files: file_path = os.path.join(root, file) # 尝试打开并读取文件 try: with open(file_path, 'r', encoding='utf-8') as f: content = f.read() # 根据是否区分大小写进行搜索 if case_sensitive: if search_text in content: print(f"在文件中找到: {file_path}") else: if search_text.lower() in content.lower(): print(f"在文件中找到: {file_path}") except UnicodeDecodeError: # 跳过二进制文件 continue except Exception as e: print(f"读取文件 {file_path} 时出错: {e}") # 使用示例 if __name__ == "__main__": search_text_in_files("/path/to/your/files", "example", case_sensitive=False)
3. 简易的HTTP服务器import http.server import socketserver import argparse def start_server(port=8000, directory='.'): """ 启动一个简单的HTTP服务器 :param port: 端口号 :param directory: 要共享的目录 """ Handler = http.server.SimpleHTTPRequestHandler Handler.directory = directory with socketserver.TCPServer(("", port), Handler) as httpd: print(f"服务器启动在 http://localhost:{port}") print(f"共享目录: {os.path.abspath(directory)}") try: httpd.serve_forever() except KeyboardInterrupt: print("\n服务器已停止") # 使用示例 if __name__ == "__main__": parser = argparse.ArgumentParser(description='启动简易HTTP服务器') parser.add_argument('-p', '--port', type=int, default=8000, help='端口号') parser.add_argument('-d', '--dir', type=str, default='.', help='要共享的目录') args = parser.parse_args() start_server(args.port, args.dir)

import os
import shutil
import datetime

def backup_files(source, destination):
    """
    备份文件或目录到指定位置
    :param source: 源文件或目录路径
    :param destination: 备份目标路径
    """
    # 检查源是否存在
    if not os.path.exists(source):
        print(f"错误: 源 {source} 不存在")
        return
    
    # 获取当前日期时间作为备份文件名的一部分
    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
    
    # 构建备份文件名
    source_name = os.path.basename(source)
    backup_name = f"{source_name}_backup_{timestamp}"
    backup_path = os.path.join(destination, backup_name)
    
    try:
        # 如果是目录，使用copytree
        if os.path.isdir(source):
            shutil.copytree(source, backup_path)
            print(f"目录备份成功: {backup_path}")
        # 如果是文件，使用copy2保留元数据
        else:
            shutil.copy2(source, backup_path)
            print(f"文件备份成功: {backup_path}")
            
    except Exception as e:
        print(f"备份失败: {e}")

# 使用示例
if __name__ == "__main__":
    # 要备份的文件或目录列表
    sources_to_backup = [
        "/path/to/important/file.txt",
        "/path/to/important/folder"
    ]
    
    # 备份存储位置
    backup_destination = "/path/to/backup/location"
    
    # 确保备份目录存在
    if not os.path.exists(backup_destination):
        os.makedirs(backup_destination)
    
    # 执行备份
    for source in sources_to_backup:
        backup_files(source, backup_destination)