aboutsummaryrefslogtreecommitdiffstats
path: root/src/featherweight.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/featherweight.py')
-rwxr-xr-xsrc/featherweight.py223
1 files changed, 3 insertions, 220 deletions
diff --git a/src/featherweight.py b/src/featherweight.py
index f51aeef..c153797 100755
--- a/src/featherweight.py
+++ b/src/featherweight.py
@@ -23,6 +23,7 @@ import sys
from subprocess import Popen, PIPE
from flocker import *
+from trees import *
old_stty = Popen('stty --save'.split(' '), stdout = PIPE, stderr = PIPE).communicate()[0]
@@ -36,8 +37,6 @@ root = '%s/.featherweight' % home
if not os.path.exists(root):
os.makedirs(root)
-islinux = ('TERM' not in os.environ) or (os.environ['TERM'] == 'linux')
-
feeds = None
with touch('%s/feeds' % root) as feeds_flock:
flock(feeds_flock, False)
@@ -49,228 +48,12 @@ with touch('%s/feeds' % root) as feeds_flock:
unflock(feeds_flock)
-def count_new(feeds):
- rc = 0
- for feed in feeds:
- count = 0
- if 'inner' in feed:
- count = count_new(feed['inner'])
- feed['new'] = count
- else:
- count = feed['new']
- rc += count
- return rc
-
-
-def is_expanded(feed):
- return ('expanded' not in feed) or feed['expanded']
-
-
-def print_node(feed, last, indent):
- global line, curline, lineoff, height
- title = feed['title']
- prefix = indent + ('└' if last else '├')
- collapsed = False
- if ('inner' not in feed) or (is_expanded(feed)):
- prefix += '── ' if islinux else '─╼ '
- else:
- collapsed = True
- prefix += '─┘ ' if islinux else '─┚ '
- if feed['new'] > 0:
- prefix += '\033[01;31m(%i)\033[00m ' % feed['new']
- if select_stack[-1][0] is feed:
- title = '\033[01;34m%s\033[00m' % title
- if lineoff <= curline < lineoff + height:
- if curline > lineoff:
- print()
- print(prefix + title, end = '')
- curline += 1
- if line >= 0:
- line += 1
- if select_stack[-1][0] is feed:
- line = ~line
- if ('inner' in feed) and not collapsed:
- inner = feed['inner']
- for feed in inner:
- print_node(feed, feed is inner[-1], indent + (' ' if last else '│ '))
-
-
-count = count_new(feeds)
-
-global height, width
-height_width = Popen('stty size'.split(' '), stdout = PIPE, stderr = PIPE).communicate()[0]
-(height, width) = height_width.decode('utf-8', 'error')[:-1].split(' ')
-height, width = int(height), int(width)
-
-global line, curline, lineoff
-line = 0
-curline = 0
-lineoff = 0
-def print_tree():
- global line, curline, lineoff, height, width
- line = 0
- curline = 0
- height_width = Popen('stty size'.split(' '), stdout = PIPE, stderr = PIPE).communicate()[0]
- (height, width) = height_width.decode('utf-8', 'error')[:-1].split(' ')
- height, width = int(height), int(width)
-
- print('\033[H\033[2J', end = '')
- title = 'My Feeds'
- if len(select_stack) == 1:
- title = '\033[01;34m%s\033[00m' % title
- if lineoff <= curline < lineoff + height:
- if count > 0:
- print('\033[01;31m(%i)\033[00m ' % count, end = '')
- print(title, end = '')
- line += 1
- curline += 1
- if len(select_stack) == 1:
- line = ~line
- for feed in feeds:
- print_node(feed, feed is feeds[-1], '')
- sys.stdout.flush()
-
- line = ~line
- if not (lineoff < line <= lineoff + height):
- lineoff = line - height // 2
- if not (lineoff < line <= lineoff + height):
- lineoff -= 1
- if lineoff < 0:
- lineoff = 0
- print_tree()
-
-
print('\033[?1049h\033[?25l', end = '')
-select_stack = [(None, None)]
-global collapsed_count
-collapsed_count = 0
-print_tree()
try:
- buf = ''
- while True:
- buf += sys.stdin.read(1)
- buf = buf[-10:]
- if buf.endswith('\033[A'):
- if select_stack[-1][0] is not None:
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- select_stack.pop()
- if curi > 0:
- par = select_stack[-1][0]
- par = feeds if par is None else par['inner']
- curi -= 1
- cur = par[curi]
- select_stack.append((cur, curi))
- while ('inner' in cur) and is_expanded(cur):
- curi = len(cur['inner']) - 1
- cur = cur['inner'][curi]
- select_stack.append((cur, curi))
- print_tree()
- elif buf.endswith('\033[1;5A'):
- if select_stack[-1][0] is not None:
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- select_stack.pop()
- if curi > 0:
- par = select_stack[-1][0]
- par = feeds if par is None else par['inner']
- select_stack.append((par[curi - 1], curi - 1))
- print_tree()
- elif buf.endswith('\033[B'):
- if select_stack[-1][0] is None:
- if len(feeds) > 0:
- select_stack.append((feeds[0], 0))
- print_tree()
- else:
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- if ('inner' in cur) and is_expanded(cur):
- select_stack.append((cur['inner'][0], 0))
- print_tree()
- else:
- backup = select_stack[:]
- while len(select_stack) > 1:
- par = select_stack[-2][0]
- par = feeds if par is None else par['inner']
- select_stack.pop()
- if curi + 1 < len(par):
- select_stack.append((par[curi + 1], curi + 1))
- backup = None
- print_tree()
- break
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- if backup is not None:
- select_stack[:] = backup
- elif buf.endswith('\033[1;5B'):
- while select_stack[-1][0] is not None:
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- par = select_stack[-2][0]
- par = feeds if par is None else par['inner']
- if curi + 1 < len(par):
- select_stack.pop()
- select_stack.append((par[curi + 1], curi + 1))
- print_tree()
- break
- elif select_stack[-2][0] is not None:
- select_stack.pop()
- else:
- break
- elif buf.endswith('\033[C'):
- if select_stack[-1][0] is None:
- if len(feeds) > 0:
- select_stack.append((feeds[0], 0))
- print_tree()
- else:
- cur = select_stack[-1][0]
- curi = select_stack[-1][1]
- if 'inner' in cur:
- if not is_expanded(cur):
- cur['expanded'] = True
- collapsed_count -= 1
- select_stack.append((cur['inner'][0], 0))
- print_tree()
- elif buf.endswith('\033[D'):
- if len(select_stack) > 1:
- select_stack.pop()
- print_tree()
- elif buf.endswith('\033[1;5D'):
- select_stack[:] = select_stack[:1]
- print_tree()
- elif buf.endswith(' '):
- cur = select_stack[-1][0]
- if cur is None:
- def expand(feed, value):
- global collapsed_count
- if 'inner' in feed:
- cur_value = is_expanded(feed)
- if cur_value != value:
- feed['expanded'] = value
- collapsed_count += -1 if value else 1
- for inner in feed['inner']:
- expand(inner, value)
- value = collapsed_count != 0
- for feed in feeds:
- expand(feed, value)
- else:
- if 'inner' in cur:
- value = not is_expanded(cur)
- collapsed_count += -1 if value else 1
- cur['expanded'] = value
- print_tree()
- elif buf.endswith(chr(ord('L') - ord('@'))):
- print_tree()
- elif buf.endswith('q'):
- break
- elif buf.endswith('\t'):
- print('Tab')
- elif buf.endswith('\n'):
- print('Enter')
+ Tree('My Feeds', feeds).interact()
-except Exception as err:
- raise err
+except:
pass
finally:
Popen(['stty', old_stty], stdout = PIPE, stderr = PIPE).communicate()
generated by cgit v1.2.3-70-g09d2 (git 2.51.0) at 2026-02-21 06:33:51 +0000