ChatGPT写脚本实战指南：从自动化到生产环境部署

1次阅读

没有评论

共计 3197 个字符，预计需要花费 8 分钟才能阅读完成。

直接使用 ChatGPT 生成脚本时，开发者经常会遇到以下问题：

代码片段不完整，缺少必要的导入语句或函数定义
缺乏错误处理机制，代码健壮性不足
变量命名不规范，可读性差
依赖特定环境配置，可移植性差
安全风险，可能生成包含漏洞的代码

这些问题导致生成的脚本往往需要大量人工修改才能投入实际使用。

构建有效的 prompt 模板是提高生成质量的关键。一个好的 prompt 应该包含：

明确的任务描述
期望的编程语言和版本
必要的依赖项说明
代码风格要求
错误处理要求
性能考虑

例如：

 请生成一个 Python 3.8 脚本，用于批量重命名指定目录下的图片文件。要求：1. 使用 Pillow 库获取图片元数据中的拍摄日期
2. 按照 "年 - 月 - 日_序号.jpg" 格式重命名
3. 包含完整的错误处理
4. 遵循 PEP8 规范
5. 添加必要的注释

生成的代码需要进行以下后处理步骤：

使用 flake8 或 pylint 进行静态分析
人工审查关键逻辑
添加单元测试
性能基准测试
安全扫描

 请编写一个 Python 脚本，用于监控指定目录的文件变化，当新增.log 文件时，自动压缩并移动到备份目录。要求：1. 使用 watchdog 库监测文件系统事件
2. 压缩使用 gzip 格式
3. 记录操作日志到 syslog
4. 处理文件权限异常
5. 添加详细注释

import gzip
import shutil
import syslog
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler

class LogHandler(FileSystemEventHandler):
    def on_created(self, event):
        if event.src_path.endswith('.log'):
            try:
                with open(event.src_path, 'rb') as f_in:
                    with gzip.open(event.src_path + '.gz', 'wb') as f_out:
                        shutil.copyfileobj(f_in, f_out)
                syslog.syslog(f'Compressed {event.src_path}')
            except PermissionError as e:
                syslog.syslog(f'Error: {str(e)}')

observer = Observer()
event_handler = LogHandler()
observer.schedule(event_handler, path='.')
observer.start()

try:
    while True:
        pass
except KeyboardInterrupt:
    observer.stop()
observer.join()

#!/usr/bin/env python3
"""
Log file monitor and compressor

Monitors a directory for new .log files, compresses them using gzip,
and moves them to a backup directory with proper error handling.
"""

import gzip
import os
import shutil
import syslog
from datetime import datetime
from pathlib import Path
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler

# Configuration
WATCH_DIR = Path('/var/log/myapp')
BACKUP_DIR = Path('/var/log/backup')
BACKUP_DIR.mkdir(exist_ok=True, parents=True)

class LogHandler(FileSystemEventHandler):
    """Handle filesystem events for log files."""

    def on_created(self, event):
        """Process newly created files."""
        src_path = Path(event.src_path)

        # Only process .log files
        if src_path.suffix != '.log':
            return

        try:
            # Create timestamped backup filename
            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
            backup_path = BACKUP_DIR / f"{src_path.stem}_{timestamp}.gz"

            # Compress file
            with open(src_path, 'rb') as f_in:
                with gzip.open(backup_path, 'wb') as f_out:
                    shutil.copyfileobj(f_in, f_out)

            # Remove original after successful compression
            src_path.unlink()

            # Log success
            syslog.syslog(syslog.LOG_INFO, 
                         f'Compressed {src_path} to {backup_path}')

        except (PermissionError, OSError) as e:
            syslog.syslog(syslog.LOG_ERR, 
                         f'Failed to process {src_path}: {str(e)}')


def main():
    """Start monitoring directory."""
    observer = Observer()
    event_handler = LogHandler()
    observer.schedule(event_handler, path=str(WATCH_DIR))
    observer.start()

    syslog.syslog(syslog.LOG_INFO, 
                 f'Started monitoring {WATCH_DIR} for log files')

    try:
        while True:
            pass
    except KeyboardInterrupt:
        observer.stop()
        syslog.syslog(syslog.LOG_INFO, 'Shutting down monitor')

    observer.join()


if __name__ == '__main__':
    main()