Script: unread_buffer.py

Mark buffer as unread.
Author: nils_2 — Version: 2 — License: GPL3
For WeeChat ≥ 1.0.
Tags: buffer, py2, py3
Added: 2015-06-14 — Updated: 2018-06-07

Download GitHub Repository

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# -*- coding: utf-8 -*-
#
# Copyright (c) 2015-2018 by nils_2 <weechatter@arcor.de>
#
# mark buffer as unread
#
# 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 <http://www.gnu.org/licenses/>.
#
# idea by Phyks
#
# 2018-06-07: nils_2, (freenode.#weechat)
#       2 : add "remove"
# 2015-03-13: nils_2, (freenode.#weechat)
#       1 : initial release

try:
    import weechat
    import time

except Exception:
    print("This script must be run under WeeChat.")
    print("Get WeeChat now at: https://weechat.org")
    quit()

SCRIPT_NAME     = "unread_buffer"
SCRIPT_AUTHOR   = "nils_2 <weechatter@arcor.de>"
SCRIPT_VERSION  = "2"
SCRIPT_LICENSE  = "GPL"
SCRIPT_DESC     = "mark buffer as unread"

def unread_buffer_cb(data, buffer, args):

    arguments = args.lower().split(' ')
    if not len(arguments) == 2:
        return weechat.WEECHAT_RC_OK

    if arguments[0] == 'low':
        priority = "0"
    elif arguments[0] == 'message':
        priority = "1"
    elif arguments[0] == 'private':
        priority = "2"
    elif arguments[0] == 'highlight':
        priority = "3"
    elif arguments[0] == 'remove':
        priority = "-1"
    else:
        return weechat.WEECHAT_RC_OK
    # search for either buffer_name or buffer_number
    ptr_buffer = buffer_infolist(arguments[1])

    weechat.buffer_set(ptr_buffer, "hotlist", "-1")
    weechat.buffer_set(ptr_buffer, "hotlist", priority)

    return weechat.WEECHAT_RC_OK

def buffer_infolist(buf_name):
    ptr_buffer = 0
    infolist = weechat.infolist_get('buffer', '', '')
    while weechat.infolist_next(infolist):
        short_name = weechat.infolist_string(infolist, 'short_name')
        name = weechat.infolist_string(infolist, 'name')
        number = weechat.infolist_integer(infolist, 'number')
        matching = name.lower().find(buf_name) >= 0
        if matching:
            ptr_buffer = weechat.infolist_pointer(infolist, 'pointer')
            break
        if not matching and buf_name.isdigit():
            matching = str(number).startswith(buf_name)
            if len(buf_name) == 0 or matching:
                ptr_buffer = weechat.infolist_pointer(infolist, 'pointer')
                break
    weechat.infolist_free(infolist)
    if ptr_buffer:
        buf_type = weechat.buffer_get_string(ptr_buffer, "localvar_type")
        # buffer has no type (e.g. channel, private, server, weechat)
        if buf_type == '':
            ptr_buffer = 0
    return ptr_buffer
# ================================[ main ]===============================
if __name__ == "__main__":
    if weechat.register(SCRIPT_NAME, SCRIPT_AUTHOR, SCRIPT_VERSION, SCRIPT_LICENSE, SCRIPT_DESC, '', ''):
        WEECHAT_VERSION = weechat.info_get("version_number", "") or 0

        if int(WEECHAT_VERSION) >= 0x01000000:
            weechat.hook_command(SCRIPT_NAME, SCRIPT_DESC, 'low <buffer>||message <buffer>||private <buffer>||highlight <buffer>',
                                 'Notify levels:\n'
                                 '  low      : message with low importance (for example irc join/part/quit)\n'
                                 '  message  : message from a user\n'
                                 '  private  : message in a private buffer\n'
                                 '  highlight: message with highlight\n'
                                 '     remove: remove buffer from hotlist\n'
                                 '\n'
                                 '<buffer> can be a buffer name or a buffer number\n'
                                 '\n'
                                 'Example:\n'
                                 '  /unread_buffer highlight 3\n'
                                 '  /unread_buffer message freenode.#weechat\n',
                                 'low|message|private|highlight|remove %(buffers_names)|%(buffers_numbers) %-',
                                 'unread_buffer_cb',
                                 '')
        else:
            weechat.prnt('','%s%s %s' % (weechat.prefix('error'),SCRIPT_NAME,': needs version 1.0 or higher'))