引子
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 \'\\r\\n\'.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库则会将结果放到剪贴板,如没有则输出结果.如果脚本给了图片路径参数则上传给定路径的图片