linux编写python脚本_使用Python编写一个在Linux下实现截图分享的脚本的教程-CSDN博客

引子

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]

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