1. 使用临时文件替换/tmp目录下，在使用之后会自动删除\n 2. 支持图片压缩，QQ截图高清格式17M直接压缩为500K\n

Author: tiann

clipboard.py

@@ -1,34 +1,45 @@
 # -*- coding: utf-8 -*-
-import time, os
+import os
+import tempfile
+
 from AppKit import NSPasteboard, NSPasteboardTypePNG, NSPasteboardTypeTIFF, NSPasteboardTypeString
 
 def get_paste_img_file():
+    ''' get a img file from clipboard;
+    the return object is a `tempfile.NamedTemporaryFile`
+    you can use the name field to access the file path.
+    the tmp file will be delete as soon as possible(when gc happened or close explicitly)
+    you can not just return a path, must hold the reference'''
+
     pb = NSPasteboard.generalPasteboard()
     data_type = pb.types()
     # if img file
-    print data_type
-    now = int(time.time() * 1000) # used for filename
-    if NSPasteboardTypePNG in data_type:
-        # png
+    # print data_type
+    # always generate png format img
+    png_file = tempfile.NamedTemporaryFile(suffix="png")
+
+    supported_image_format = (NSPasteboardTypePNG, NSPasteboardTypeTIFF)
+    if any(filter(lambda f: f in data_type, supported_image_format)):
+        # do not care which format it is, we convert it to png finally
+        # system screen shotcut is png, QQ is tiff
+        tmp_img_file = tempfile.NamedTemporaryFile()
         data = pb.dataForType_(NSPasteboardTypePNG)
-        filename = '%s.png' % now
-        filepath = '/tmp/%s' % filename
-        ret = data.writeToFile_atomically_(filepath, False)
-        if ret:
-            return filepath
-    elif NSPasteboardTypeTIFF in data_type:
-        # tiff
-        data = pb.dataForType_(NSPasteboardTypeTIFF)
-        filename = '%s.tiff' % now
-        filepath = '/tmp/%s' % filename
-        ret = data.writeToFile_atomically_(filepath, False)
-        if ret:
-            os.system('sips -s format tiff /tmp/%s.tiff --out /tmp/%s.png' % (now, now))
-            return '/tmp/%s.png' % now
+        ret = data.writeToFile_atomically_(tmp_img_file.name, False)
+        if not ret: return
+
+        # convert it to png file
+        os.system('sips -s format png %s --out %s' % (tmp_img_file.name, png_file.name))
+
+        # close the file explicitly
+        tmp_img_file.close()
+        return png_file
+    
     elif NSPasteboardTypeString in data_type:
         # string todo, recognise url of png & jpg
         pass
 
+if __name__ == '__main__':
+    get_paste_img_file()
 
 
 

info.plist

@@ -85,6 +85,7 @@ import util
 import os
 import subprocess
 import sys
+import time
 
 if not os.path.exists(util.CONFIG_FILE):
     util.generate_config_file()
@@ -99,15 +100,17 @@ url = '%s/%s' % (config['url'], config['prefix'])
 
 img_file = get_paste_img_file()
 if img_file:
-    # has file
-    name = os.path.split(img_file)[1]
-    size_str = subprocess.check_output('sips -g pixelWidth %s | tail -n1 | cut -d" " -f4' % img_file, shell=True)
+    # has image
+
+    # use time to generate a unique upload_file name, we can not use the tmp file name
+    upload_name = "%s.png" % int(time.time() * 1000) 
+    size_str = subprocess.check_output('sips -g pixelWidth %s | tail -n1 | cut -d" " -f4' % img_file.name, shell=True)
     size = int(size_str.strip()) / 2
-    markdown_url = '<img src="%s/%s" width="%d"/>' % (url, name, size)
+    markdown_url = '<img src="%s/%s" width="%d"/>' % (url, upload_name, size)
     # make it to clipboard
     os.system("echo '%s' | pbcopy" % markdown_url)
     os.system('osascript -e \'tell application "System Events" to keystroke "v" using command down\'')
-    ret = upload_qiniu(img_file)
+    ret = upload_qiniu(img_file.name, upload_name)
     if not ret: util.notice("上传图片到图床失败，请检查网络后重试")
 else:
     util.notice("剪切版里没有图片！")

markdown img.alfredworkflow

@@ -6,11 +6,10 @@
 
 config = read_config()
 
-def upload_qiniu(path):
+def upload_qiniu(path, upload_name):
     ''' upload file to qiniu'''
-    q = Auth(config['ak'], config['sk']) 
-    dirname, filename = os.path.split(path)
-    key = '%s/%s' % (config['prefix'], filename) # upload to qiniu's markdown dir
+    q = Auth(config['ak'], config['sk'])
+    key = '%s/%s' % (config['prefix'], upload_name) # upload to qiniu's markdown dir
 
     token = q.upload_token(config['bucket'], key)
     ret, info = put_file(token, key, path, check_crc=True)