diff options
Diffstat (limited to 'py3status')
| -rw-r--r-- | py3status/modules/screenshot.py | 123 |
1 files changed, 69 insertions, 54 deletions
diff --git a/py3status/modules/screenshot.py b/py3status/modules/screenshot.py index 1f20478..9ede108 100644 --- a/py3status/modules/screenshot.py +++ b/py3status/modules/screenshot.py @@ -1,75 +1,90 @@ # -*- coding: utf-8 -*- """ -Display a 'SHOT' button in your i3bar allowing you to take screenshot and directly send (if wanted) the file to your online server, as long the screenshot - has been taken, 'SHOT' is replaced by the file_name generated +Take a screenshot and optionally upload it to your online server. + +Display a 'SHOT' button in your i3bar allowing you to take a screenshot and +directly send (if wanted) the file to your online server. +When the screenshot has been taken, 'SHOT' is replaced by the file_name. + +This modules uses the 'gnome-screenshot' program to take the screenshot. Configuration parameters: - - save_path: Directory where to store your screenshots. - file_length: generated file_name length - push: True/False if yo want to push your screenshot to your server - - upload_server: your server address + - save_path: Directory where to store your screenshots. - upload_path: the remote path where to push the screenshot - - upload_user: your ssh user + - upload_server: your server address + - upload_user: your ssh user @author Amaury Brisou <py3status AT puzzledge.org> """ -import time, os, string, random, subprocess +import os +import random +import string +import subprocess +import time + class Py3status: - cache_timeout = 5 - - save_path = "%s%s" % (os.environ['HOME'] , "/Pictures/") - file_length = 4 - upload_path = "/files" - upload_server = 'puzzledge.org' - upload_user = 'erol' - push = True + """ + """ + # available configuration parameters + cache_timeout = 5 + save_path = '%s%s' % (os.environ['HOME'], '/Pictures/') + file_length = 4 + upload_path = "/files" + upload_server = 'puzzledge.org' + upload_user = 'erol' + push = True - full_text = "" + def __init__(self): + self.full_text = '' - def __init__(self): - pass + def on_click(self, i3s_output_list, i3s_config, event): - def kill(self, i3s_output_list, i3s_config): - pass - - def on_click(self, i3s_output_list, i3s_config, event): - - file_name = self._filename_generator(self.file_length) + file_name = self._filename_generator(self.file_length) - command = "%s %s/%s%s" % ('gnome-screenshot -f', self.save_path, file_name, '.jpg') - - subprocess.Popen(command.split()) + command = '%s %s/%s%s' % ('gnome-screenshot -f', self.save_path, + file_name, '.jpg') - self.full_text = "%s%s" % (file_name, '.jpg') + subprocess.Popen(command.split()) - if self.push and self.upload_server and self.upload_user and self.upload_path: - command = "scp %s/%s%s %s@%s:%s" % (self.save_path, file_name, '.jpg', self.upload_user, self.upload_server, self.upload_path) - subprocess.Popen(command.split()) - - def _filename_generator(self, size=6, chars=string.ascii_lowercase + string.digits): - return ''.join(random.choice(chars) for _ in range(size)) + self.full_text = '%s%s' % (file_name, '.jpg') + + if (self.push and self.upload_server and self.upload_user and + self.upload_path): + command = 'scp %s/%s%s %s@%s:%s' % (self.save_path, file_name, + '.jpg', self.upload_user, + self.upload_server, + self.upload_path) + subprocess.Popen(command.split()) + + def _filename_generator(self, + size=6, + chars=string.ascii_lowercase + string.digits): + return ''.join(random.choice(chars) for _ in range(size)) + + def screenshot(self, i3s_output_list, i3s_config): + if self.full_text == '': + self.full_text = 'SHOT' + + response = { + 'cached_until': time.time() + self.cache_timeout, + 'color': i3s_config['color_good'], + 'full_text': self.full_text + } + return response - def screenshot(self, i3s_output_list, i3s_config): - if self.full_text == "": - self.full_text = "SHOT" - response = { - 'cached_until': time.time() + self.cache_timeout, - 'full_text': self.full_text, - 'color': i3s_config['color_good'], - } - return response - if __name__ == "__main__": - - from time import sleep - x = Py3status() - config = { - 'color_bad': '#FF0000', - 'color_degraded': '#FFFF00', - 'color_good': '#00FF00', - } - while True: - print(x.screenshot([], config)) - sleep(1)
\ No newline at end of file + + from time import sleep + x = Py3status() + config = { + 'color_bad': '#FF0000', + 'color_degraded': '#FFFF00', + 'color_good': '#00FF00', + } + while True: + print(x.screenshot([], config)) + sleep(1) |