# Copyright (C) 2009, 2010 Roman Zimbelmann <romanz@lavabit.com>
#
# 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/>.
"""
The statusbar displays information about the current file and directory.
On the left side, there is a display similar to what "ls -l" would
print for the current file. The right side shows directory information
such as the space used by all the files in this directory.
"""
from pwd import getpwuid
from grp import getgrgid
from os import getuid
from time import strftime, localtime
from ranger.ext.human_readable import human_readable
from . import Widget
from ranger.gui.bar import Bar
class StatusBar(Widget):
__doc__ = __doc__
owners = {}
groups = {}
timeformat = '%Y-%m-%d %H:%M'
hint = None
msg = None
old_cf = None
old_mtime = None
old_du = None
old_hint = None
result = None
def __init__(self, win, column=None):
Widget.__init__(self, win)
self.column = column
def notify(self, text, duration=4, bad=False):
self.msg = Message(text, duration, bad)
def draw(self):
"""Draw the statusbar"""
if self.hint and isinstance(self.hint, str):
if self.old_hint != self.hint:
self.need_redraw = True
if self.need_redraw:
self._draw_hint()
return
if self.old_hint and not self.hint:
self.old_hint = None
self.need_redraw = True
if self.msg:
if self.msg.is_alive():
self._draw_message()
return
else:
self.msg = None
self.need_redraw = True
try:
mtime = self.env.cf.stat.st_mtime
except:
mtime = -1
if not self.result:
self.need_redraw = True
if self.old_du and not self.env.cwd.disk_usage:
self.old_du = self.env.cwd.disk_usage
self.need_redraw = True
if self.old_cf != self.env.cf:
self.old_cf = self.env.cf
self.need_redraw = True
if self.old_mtime != mtime:
self.old_mtime = mtime
self.need_redraw = True
if self.need_redraw:
self.need_redraw = False
self._calc_bar()
self._print_result(self.result)
def _calc_bar(self):
bar = Bar('in_statusbar')
self._get_left_part(bar)
self._get_right_part(bar)
bar.shrink_by_removing(self.wid)
self.result = bar.combine()
def _draw_message(self):
self.win.erase()
self.color('in_statusbar', 'message',
self.msg.bad and 'bad' or 'good')
self.addnstr(0, 0, self.msg.text, self.wid)
def _draw_hint(self):
self.win.erase()
highlight = True
space_left = self.wid
starting_point = self.x
for string in self.hint.split('//'):
highlight = not highlight
if highlight:
self.color('in_statusbar', 'text', 'highlight')
else:
self.color('in_statusbar', 'text')
try:
self.addnstr(0, starting_point, string, space_left)
except:
break
space_left -= len(string)
starting_point += len(string)
def _get_left_part(self, bar):
left = bar.left
if self.column is not None and self.column.target is not None\
and self.column.target.is_directory:
target = self.column.target.pointed_obj
else:
target = self.env.at_level(0).pointed_obj
if target is None:
return
if target.accessible is False:
return
perms = target.get_permission_string()
how = getuid() == target.stat.st_uid and 'good' or 'bad'
left.add(perms, 'permissions', how)
left.add_space()
left.add(str(target.stat.st_nlink), 'nlink')
left.add_space()
left.add(self._get_owner(target), 'owner')
left.add_space()
left.add(self._get_group(target), 'group')
left.add_space()
if target.islink:
how = target.exists and 'good' or 'bad'
left.add('-> ' + target.readlink, 'link', how)
else:
left.add(strftime(self.timeformat,
localtime(target.stat.st_mtime)), 'mtime')
def _get_owner(self, target):
uid = target.stat.st_uid
try:
return self.owners[uid]
except KeyError:
try:
self.owners[uid] = getpwuid(uid)[0]
return self.owners[uid]
except KeyError:
return str(uid)
def _get_group(self, target):
gid = target.stat.st_gid
try:
return self.groups[gid]
except KeyError:
try:
self.groups[gid] = getgrgid(gid)[0]
return self.groups[gid]
except KeyError:
return str(gid)
def _get_right_part(self, bar):
right = bar.right
if self.column is None:
return
target = self.column.target
if target is None:
return
if not target.content_loaded or not target.accessible:
return
pos = target.scroll_begin
max_pos = len(target) - self.column.hei
base = 'scroll'
if target.marked_items:
if len(target.marked_items) == len(target.files):
right.add(human_readable(target.disk_usage, seperator=''))
else:
right.add(human_readable(sum(f.size \
for f in target.marked_items \
if f.is_file), seperator=''))
right.add(" / " + str(len(target.marked_items)))
else:
right.add(human_readable(target.disk_usage, seperator=''))
right.add(", ", "space")
right.add(human_readable(self.env.get_free_space( \
target.mount_path), seperator=''))
right.add(" ", "space")
if target.marked_items:
# Indicate that there are marked files. Useful if you scroll
# away and don't see them anymore.
right.add('Mrk', base, 'marked')
elif max_pos > 0:
if pos == 0:
right.add('Top', base, 'top')
elif pos >= max_pos:
right.add('Bot', base, 'bot')
else:
right.add('{0:0>.0f}%'.format(100.0 * pos / max_pos),
base, 'percentage')
else:
right.add('All', base, 'all')
def _print_result(self, result):
import _curses
self.win.move(0, 0)
for part in result:
self.color(*part.lst)
self.addstr(part.string)
self.color_reset()
from time import time
class Message(object):
elapse = None
text = None
bad = False
def __init__(self, text, duration, bad):
self.text = text
self.bad = bad
self.elapse = time() + duration
def is_alive(self):
return time() <= self.elapse