Linux 下使用Python截图自动分享

Tue 22 January 2013

引子

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

Category: Python Tagged: Linux scrot 截图 python

Comments

Fork me on GitHub