Snipaste 如何通过命令行实现截图后自动上传到图床并生成链接

pip install requests
# 如果使用腾讯云COS，可能还需要其SDK
pip install cos-python-sdk-v5

Snipaste.exe snipping [output_file]

Snipaste.exe snip --output "C:\path\to\your\screenshot.png"

import subprocess
import os
import time

def take_snipaste_screenshot(output_path):
    """
    调用 Snipaste 进行区域截图并保存到指定路径。
    注意：此命令会启动截图交互，用户需要手动选择区域。
    """
    # 确保输出目录存在
    os.makedirs(os.path.dirname(output_path), exist_ok=True)

    # 构建命令
    # 假设 Snipaste.exe 已在 PATH 中，否则需使用完整路径，如 r"C:\Program Files\Snipaste\Snipaste.exe"
    command = ['Snipaste.exe', 'snip', '--output', output_path]

    try:
        # 启动 Snipaste 截图进程
        process = subprocess.Popen(command, shell=True)
        # 这里可以等待一段时间，或者通过其他方式判断截图是否完成
        # 一个简单的方法是检查输出文件是否被创建/修改
        timeout = 30  # 最多等待30秒
        start_time = time.time()
        while not os.path.exists(output_path):
            if time.time() - start_time > timeout:
                print("截图超时或用户取消。")
                process.terminate()
                return False
            time.sleep(0.5)
        print(f"截图已保存至：{output_path}")
        return True
    except FileNotFoundError:
        print("错误：未找到 Snipaste.exe。请确保 Snipaste 已安装且路径正确。")
        return False
    except Exception as e:
        print(f"调用截图命令时发生未知错误：{e}")
        return False

# 使用示例
if __name__ == "__main__":
    screenshot_path = os.path.join(os.getenv('TEMP'), 'auto_upload_snip.png')
    if take_snipaste_screenshot(screenshot_path):
        print("截图成功，准备上传...")
        # 此处可衔接上传函数
    else:
        print("截图失败。")

import requests

def upload_to_smms(file_path, api_token):
    """
    上传图片到 SM.MS 图床。
    :param file_path: 本地图片文件路径
    :param api_token: 你的 SM.MS API Token
    :return: 成功返回图片链接，失败返回 None
    """
    url = 'https://sm.ms/api/v2/upload'
    headers = {'Authorization': api_token}
    
    try:
        with open(file_path, 'rb') as f:
            files = {'smfile': f}
            response = requests.post(url, files=files, headers=headers, timeout=30)
        result = response.json()
        
        if response.status_code == 200 and result.get('success'):
            image_url = result['data']['url']
            print(f"上传成功！链接：{image_url}")
            return image_url
        else:
            error_msg = result.get('message', '未知错误')
            print(f"上传失败：{error_msg}")
            # 检查是否是重复上传，SM.MS会返回已存在的链接
            if 'Image upload repeated' in error_msg:
                # 从错误信息中提取已有链接（根据API实际返回格式调整）
                # 例如，有时错误信息里会包含链接，或者 `result['images']` 里有
                print("图片已存在于图床。")
                # 这里需要根据实际的错误响应结构来解析，本例作简化处理
            return None
    except requests.exceptions.RequestException as e:
        print(f"网络请求错误：{e}")
        return None
    except IOError as e:
        print(f"文件读取错误：{e}")
        return None

# 示例使用腾讯云COS Python SDK
from qcloud_cos import CosConfig
from qcloud_cos import CosS3Client
import sys
import os

def upload_to_tencent_cos(file_path, config):
    """
    上传图片到腾讯云COS。
    :param file_path: 本地图片文件路径
    :param config: 包含 secret_id, secret_key, region, bucket 等配置的字典
    :return: 成功返回图片的公开访问链接，失败返回 None
    """
    secret_id = config['secret_id']
    secret_key = config['secret_key']
    region = config['region']
    bucket = config['bucket']
    # 可以自定义上传后的文件名，这里使用时间戳+原文件名
    import time
    object_name = f"snipaste/{int(time.time())}_{os.path.basename(file_path)}"

    cos_config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key)
    client = CosS3Client(cos_config)

    try:
        response = client.upload_file(
            Bucket=bucket,
            LocalFilePath=file_path,
            Key=object_name,
            PartSize=1, # 大于1MB的文件会启用分块上传，这里可调整
            MAXThread=10 # 并发线程数
        )
        # 生成公开访问链接（前提是存储桶和对象是公有读，或配置了CDN域名）
        # 如果配置了CDN，请使用CDN域名，否则使用COS默认域名
        cdn_domain = config.get('cdn_domain', f'{bucket}.cos.{region}.myqcloud.com')
        image_url = f'https://{cdn_domain}/{object_name}'
        print(f"上传成功！链接：{image_url}")
        return image_url
    except Exception as e:
        print(f"上传至腾讯云COS失败：{e}")
        return None

import subprocess
import os
import time
import requests
import pyperclip  # 用于操作剪贴板，需安装：pip install pyperclip
import sys

class SnipasteAutoUploader:
    def __init__(self, upload_config, use_smms=True):
        """
        初始化上传器。
        :param upload_config: 图床配置字典。
                              对于SM.MS: {'api_token': 'your_token_here'}
                              对于COS: {'secret_id': '...', 'secret_key': '...', 'region': '...', 'bucket': '...', 'cdn_domain': '...'(可选)}
        :param use_smms: True使用SM.MS，False使用腾讯云COS。
        """
        self.upload_config = upload_config
        self.use_smms = use_smms
        # 定义临时文件存放目录
        self.temp_dir = os.path.join(os.getenv('TEMP'), 'SnipasteAutoUpload')
        os.makedirs(self.temp_dir, exist_ok=True)

    def take_screenshot(self):
        """步骤1：调用 Snipaste 截图并保存到临时文件。"""
        timestamp = int(time.time())
        self.screenshot_path = os.path.join(self.temp_dir, f'snip_{timestamp}.png')
        
        print("正在启动 Snipaste 截图... (请用鼠标选择区域)")
        command = ['Snipaste.exe', 'snip', '--output', self.screenshot_path]
        try:
            process = subprocess.Popen(command, shell=True)
            # 等待截图完成（通过检查文件是否存在）
            timeout = 60
            start_time = time.time()
            while not os.path.exists(self.screenshot_path):
                if time.time() - start_time > timeout:
                    print("错误：截图等待超时。用户可能取消了操作。")
                    process.terminate()
                    return False
                time.sleep(0.5)
            # 额外等待一小会儿，确保文件写入完成
            time.sleep(0.5)
            print(f"截图已临时保存：{self.screenshot_path}")
            return True
        except Exception as e:
            print(f"截图过程出错：{e}")
            return False

    def upload_screenshot(self):
        """步骤2：将临时图片文件上传到指定的图床。"""
        if not os.path.exists(self.screenshot_path):
            print("错误：未找到截图文件。")
            return None
        
        file_size = os.path.getsize(self.screenshot_path) / 1024  # KB
        print(f"准备上传图片，大小：{file_size:.2f} KB")
        
        if self.use_smms:
            api_token = self.upload_config.get('api_token')
            if not api_token:
                print("错误：未配置 SM.MS API Token。")
                return None
            image_url = upload_to_smms(self.screenshot_path, api_token)
        else:
            # 使用腾讯云COS
            image_url = upload_to_tencent_cos(self.screenshot_path, self.upload_config)
        
        return image_url

    def cleanup(self):
        """步骤4：清理临时截图文件。"""
        try:
            if os.path.exists(self.screenshot_path):
                os.remove(self.screenshot_path)
                print(f"已清理临时文件：{self.screenshot_path}")
        except OSError as e:
            print(f"清理临时文件时出错：{e}")

    def run(self):
        """执行完整工作流：截图 -> 上传 -> 复制链接 -> 清理。"""
        print("=== Snipaste 自动上传图床脚本启动 ===")
        
        # 1. 截图
        if not self.take_screenshot():
            print("工作流因截图失败而终止。")
            return
        
        # 2. 上传
        image_url = self.upload_screenshot()
        
        # 3. 处理结果
        if image_url:
            # 将链接复制到剪贴板
            try:
                pyperclip.copy(image_url)
                print(f"图片链接已复制到剪贴板：{image_url}")
                # 可选：在屏幕上显示一个轻量级通知（Windows可用 toast-notifier 等库）
            except Exception as e:
                print(f"复制到剪贴板失败：{e}")
                print(f"请手动复制链接：{image_url}")
            # 成功，可以保留临时文件稍后清理，或立即清理
            # 这里选择立即清理
            self.cleanup()
            print("自动化流程成功完成！")
        else:
            print("上传失败。临时截图文件保留以供手动处理：", self.screenshot_path)
            # 上传失败时，不自动删除文件，让用户可以手动处理

# 配置部分（请根据你的实际情况修改）
if __name__ == "__main__":
    # 示例：使用 SM.MS
    smms_config = {
        'api_token': 'YOUR_SMMS_API_TOKEN_HERE'  # 请替换为你的真实 Token
    }
    uploader = SnipasteAutoUploader(smms_config, use_smms=True)
    
    # 示例：使用腾讯云 COS (取消注释并配置)
    # cos_config = {
    #     'secret_id': 'YOUR_SECRET_ID',
    #     'secret_key': 'YOUR_SECRET_KEY',
    #     'region': 'ap-shanghai',  # 你的存储桶地域
    #     'bucket': 'your-bucket-name-1250000000',
    #     'cdn_domain': 'cdn.yourdomain.com'  # 如果配置了CDN加速域名
    # }
    # uploader = SnipasteAutoUploader(cos_config, use_smms=False)
    
    uploader.run()