Arthur de Jong

Open Source / Free Software developer

summaryrefslogtreecommitdiffstats
path: root/ls.py
blob: 718aab8a925157ca0f820c2e67884ea8baeb2be1 (plain)
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
# ls.py - functions for listing file information
#
# Copyright (C) 2015 Arthur de Jong
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
#
# The files produced as output from the software do not automatically fall
# under the copyright of the software, unless explicitly stated otherwise.

import stat
import sys
import time


def supports_colors(stream):
    """Check whether the specified stream supports colors."""
    if stream.isatty():
        try:
            import curses
            curses.setupterm()
            return curses.tigetnum('colors') > 2
        except:
            pass
    return False


# see if stdout supports colors
_stdout_colors = supports_colors(sys.stdout)


def ls_format(info, use_colors=_stdout_colors):
    info['date'] = time.strftime(
        '%Y-%m-%d %H:%M:%S', time.localtime(info['mtime']))
    mode = info.get('mode', 0)
    ftype = info.get('type', '?').lower()
    color = ''
    # get file type
    if ftype == 'd':
        color = '\x1b[01;34m'  # blue
    elif ftype == 'l':
        color = '\x1b[01;36m'  # cyan
    elif ftype == 'r':
        ftype = '-'
        if mode & (stat.S_IXGRP | stat.S_IXUSR | stat.S_IXOTH):
            color = '\x1b[01;32m'  # green
    # build file permissions
    perms = ''
    for who in 'USR', 'GRP', 'OTH':
        for what in 'R', 'W', 'X':
            if mode & getattr(stat, 'S_I' + what + who):
                perms += what.lower()
            else:
                perms += '-'
    info['ftype'] = ftype
    info['perms'] = perms
    info['color'] = color if use_colors else ''
    info['endcolor'] = '\x1b[00m' if info['color'] else ''
    print(('%(ftype)s%(perms)s %(uid)-8s %(gid)-8s %(size)8d %(date)s ' +
           '%(color)s%(path)s%(endcolor)s') % info)