引子

Linux下不支持QQ等功能丰富的IM,虽然可以通过wine运行QQ2012,但是还是喜欢在gtalk群中聊天,gtalk群不支持图片方式,这就要靠我们大家自己来解决了,eleven开放了一个Image上传和显示接口,提供了使用curl来解决,但是我们公司的网络使用squid禁止了curl的访问,所以整天看他们这么爽的分享图片我也不甘心阿,所以就使用Python写了一个分享图片的脚本

实现

使用scrot截图,然后使用urllib2库上传图片,如果存在PyQt4库则会将结果放到剪贴板上,如果不存在则输出,自行复制

代码

#!/usr/bin/env python # -*- coding:utf-8 -*- # # Author : cold # E-mail : wh_linux@126.com # Date : 13/01/21 09:54:39 # Desc : 贴代码和图片 # import urllib2, json import mimetools import mimetypes import itertools __host__ = "http://eleveni386.7axu.com" class Form(object): def __init__(self): self.form_fields = [] self.files = [] self.boundary = mimetools.choose_boundary() self.content_type = "application/x-www-form-urlencoded" return def get_content_type(self): return self.content_type def add_field(self, name, value): self.form_fields.append((name, value)) return def add_file(self, fieldname, filename, fileHandle, mimetype=None): body = fileHandle.read() if mimetype is None: mimetype = ( mimetypes.guess_type(filename)[0] or 'applicatioin/octet-stream') self.files.append((fieldname, filename, mimetype, body)) self.content_type = 'multipart/form-data; boundary=%s' % self.boundary return def __str__(self): parts = [] part_boundary = '--' + self.boundary parts.extend( [ part_boundary, 'Content-Disposition: form-data; name="%s"' % name, '', value, ] for name, value in self.form_fields) if self.files: parts.extend([ part_boundary, 'Content-Disposition: form-data; name="%s"; filename="%s"' % (field_name, filename), 'Content-Type: %s' % content_type, '', body, ] for field_name, filename, content_type, body in self.files) flattened = list(itertools.chain(*parts)) flattened.append('--' + self.boundary + '--') flattened.append('') return 'rn'.join(flattened) class HttpHelper(object): def __init__(self, url = None, form = None, method = 'GET'): self._url = url self._form = form self._body = str(form) self._method = method self._dst_url = None if url: self.make_request() def make_request(self): url = self._url if not self._url.startswith('http://'): url = 'http://' + self._url self.request = urllib2.Request(url) if self._form: self.add_header("Content-Type", self._form.get_content_type()) self.add_header("Content-Length", len(self._body)) self.request.add_data(self._body) def add_header(self, key, val): self.request.add_header(key, val) def change(self, url, params = {}, method = 'GET'): self._url = url self._params = params self._method = method self.make_request() def open(self): response = urllib2.urlopen(self.request) content = response.read() self._dst_url = response.geturl() try: return json.loads(content) except: return content if __name__ == "__main__": import argparse import os parser = argparse.ArgumentParser() parser.add_argument(dest="path", nargs="") args = parser.parse_args() if args.path: path = args.path else: path = r"/tmp/tmpscrot.png" os.system("scrot -s {0}".format(path)) form = Form() filename = os.path.split(path)[-1] form.add_file(fieldname='mypic', filename=filename, fileHandle=open(path)) http = HttpHelper( __host__ + '/Image/', form) url = http.open() try: from PyQt4.QtGui import QApplication app = QApplication([]) cb = QApplication.clipboard() cb.setText(url) except: print url

安装

将上面代码保存一个文件,放在PATH路径里,赋予执行权限即可

使用

默认的不跟图片地址则会截图,截图完毕后自动分享,如安装了PyQt4库则会将结果放到剪贴板,如没有则输出结果.如果脚本给了图片路径参数则上传给定路径的图片