about summary refs log tree commit diff stats
path: root/src/loslib.c
diff options
context:
space:
mode:
authorKartik K. Agaram <vc@akkartik.com>2021-12-25 16:43:50 -0800
committerKartik K. Agaram <vc@akkartik.com>2021-12-25 16:43:50 -0800
commit8458ba0202abe802c71ce9adba85833083bb73ea (patch)
treecd4fbbc7cf2346c14399bb824adc5bc6c58a3b78 /src/loslib.c
parent12cb5650772abb9a30f6e0559bcde5c2e2cf800b (diff)
downloadteliva-8458ba0202abe802c71ce9adba85833083bb73ea.tar.gz
fix a couple of colors
Diffstat (limited to 'src/loslib.c')
0 files changed, 0 insertions, 0 deletions
'n91' href='#n91'>91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
# This file is part of ranger, the console file manager.
# License: GNU GPL version 3, see the file "AUTHORS" for details.

# TODO: add a __getitem__ method to get the tag of a file

from __future__ import (absolute_import, division, print_function)

from os.path import isdir, exists, dirname, abspath, realpath, expanduser
import string
import sys

ALLOWED_KEYS = string.ascii_letters + string.digits + string.punctuation


class Tags(object):
    default_tag = '*'

    def __init__(self, filename):

        self._filename = realpath(abspath(expanduser(filename)))

        if isdir(dirname(self._filename)) and not exists(self._filename):
            open(self._filename, 'w')

        self.sync()

    def __contains__(self, item):
        return item in self.tags

    def add(self, *items, **others):
        if 'tag' in others:
            tag = others['tag']
        else:
            tag = self.default_tag
        self.sync()
        for item in items:
            self.tags[item] = tag
        self.dump()

    def remove(self, *items):
        self.sync()
        for item in items:
            try:
                del self.tags[item]
            except KeyError:
                pass
        self.dump()

    def toggle(self, *items, **others):
        if 'tag' in others:
            tag = others['tag']
        else:
            tag = self.default_tag
        tag = str(tag)
        if tag not in ALLOWED_KEYS:
            return
        self.sync()
        for item in items:
            try:
                if item in self and tag in (self.tags[item], self.default_tag):
                    del self.tags[item]
                else:
                    self.tags[item] = tag
            except KeyError:
                pass
        self.dump()

    def marker(self, item):
        if item in self.tags:
            return self.tags[item]
        return self.default_tag

    def sync(self):
        try:
            if sys.version_info[0] >= 3:
                fobj = open(self._filename, 'r', errors='replace')
            else:
                fobj = open(self._filename, 'r')
        except OSError:
            pass
        else:
            self.tags = self._parse(fobj)
            fobj.close()

    def dump(self):
        try:
            fobj = open(self._filename, 'w')
        except OSError:
            pass
        else:
            self._compile(fobj)
            fobj.close()

    def _compile(self, fobj):
        for path, tag in self.tags.items():
            if tag == self.default_tag:
                # COMPAT: keep the old format if the default tag is used
                fobj.write(path + '\n')
            elif tag in ALLOWED_KEYS:
                fobj.write('{0}:{1}\n'.format(tag, path))

    def _parse(self, fobj):
        result = dict()
        for line in fobj:
            line = line.strip()
            if len(line) > 2 and line[1] == ':':
                tag, path = line[0], line[2:]
                if tag in ALLOWED_KEYS:
                    result[path] = tag
            else:
                result[line] = self.default_tag

        return result

    def __nonzero__(self):
        return True
    __bool__ = __nonzero__


class TagsDummy(Tags):
    """A dummy Tags class for use with `ranger --clean`.

    It acts like there are no tags and avoids writing any changes.
    """

    def __init__(self, filename):  # pylint: disable=super-init-not-called
        self.tags = dict()

    def __contains__(self, item):
        return False

    def add(self, *items, **others):
        pass

    def remove(self, *items):
        pass

    def toggle(self, *items, **others):
        pass

    def marker(self, item):
        return self.default_tag

    def sync(self):
        pass

    def dump(self):
        pass

    def _compile(self, fobj):
        pass

    def _parse(self, fobj):
        pass