From d48a8becba6f701b2d243a29d4ccc7ff03c1ed3a Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Thu, 6 Oct 2016 21:56:23 +0200 Subject: m + add ii MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Mattias Andrée --- src/plugins/ii.py | 142 +++++++++++++++++++++++++++++++++++++++++++++++++ src/plugins/inotify.py | 7 +++ 2 files changed, 149 insertions(+) create mode 100644 src/plugins/ii.py (limited to 'src') diff --git a/src/plugins/ii.py b/src/plugins/ii.py new file mode 100644 index 0000000..5ffb266 --- /dev/null +++ b/src/plugins/ii.py @@ -0,0 +1,142 @@ +# -*- python -*- +''' +xpybar – xmobar replacement written in python +Copyright © 2014, 2015, 2016 Mattias Andrée (maandree@member.fsf.org) + +This program is free software: you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation, either version 3 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program. If not, see . +''' + + +class II: + ''' + IRC interface, requires a running instance of ii + ''' + + + def __init__(self, channel, prefix = None): + ''' + Constructor + + @param channel:str The name of the server or the name of the + server followed by a slash and the name of + the channel including the she + @param prefix:str? The argument associated with ii's -i flag + ''' + import pwd, os + if prefix is None: + prefix = pwd.getpwuid(os.getuid()).pw_dir + '/irc' + self.infile = '%s/%s/in' % (prefix, channel) + self.outfile = '%s/%s/out' % (prefix, channel) + self.log = [] + self.time = '0000-00-00 00:00 ' + + + def write(self, text): + ''' + Write to the channel + + @param text:str The message or command + ''' + text = (text + '\n').encode('utf-8') + with open(self.infile, 'wb') as file: + file.write(text) + file.flush() + + + def read(self): + ''' + Fetch new messages from the channel + + @return :list List of new messages + ''' + with open(self.outfile, 'rb') as file: + text = file.read() + text = text.decode('utf-8', 'replace').split('\n')[:-1] + i = 0 + for line in text: + if line[:17] >= self.time: + break + i += 1 + text = text[i:] + i = 0 + while i < len(text) and i < len(self.log): + if text[i] != self.log[i]: + break + i += 1 + text = text[i:] + if len(text) == 0: + return text + last = text[-1][:17] + if last == self.time: + self.log.extend(text) + return text + self.log = [] + self.time = last + i = 0 + for line in text: + if line[:17] == self.time: + break + i += 1 + self.log = text[i:] + return text + + + def wait(self, timeout = None): + ''' + Wait until more messages are available + + @param timeout:int? The number of seconds to wait + before return unconditionally + ''' + import os + command = ['inotifywait', '-e', 'close_write,modify'] + if timeout is not None: + command += ['-t', str(int(timeout + 0.5))] + command += ['--', self.outfile] + if 'PATH' in os.environ: + path = os.environ['PATH'].split('.') + for p in path: + p += '/pdeath' + if os.access(p, os.X_OK, effective_ids = True): + command = [p, 'HUP'] + command + break + spawn_read(*command) + + + @staticmethod + def list_channels(prefix = None): + ''' + Fetch a list of all joined servers and channels + + @param prefix:str? The argument associated with ii's -i flag + @return :list List of servers and channels + ''' + import pwd, os + if prefix is None: + prefix = pwd.getpwuid(os.getuid()).pw_dir + '/irc' + ret = [] + def isfifo(f): + try: + return os.path.stat.S_ISFIFO(os.stat('f').st_mode) + except: + return False + def recurse(d): + fs = os.listdir(d) + for f in fs: + f = d + '/' + f + if os.path.isfile(f + '/out') and isfifo(f + '/in'): + ret.append(f[len(prefix) + 1:]) + recurse(prefix) + return ret + diff --git a/src/plugins/inotify.py b/src/plugins/inotify.py index f92db48..84b24e9 100644 --- a/src/plugins/inotify.py +++ b/src/plugins/inotify.py @@ -51,6 +51,13 @@ class Inotify: command.append('-e') command.append(event) command += list(arguments) + if 'PATH' in os.environ: + path = os.environ['PATH'].split('.') + for p in path: + p += '/pdeath' + if os.access(p, os.X_OK, effective_ids = True): + command = [p, 'HUP'] + command + break def start(): with LineReader(spawn(*command)) as reader: while True: -- cgit v1.2.3-70-g09d2