bib.selection

#!/usr/bin/env python3

import json
import urwid
import urllib.request
import urllib.error
import argparse
import subprocess
import asyncio
import aiohttp

_cache = {}

parser = argparse.ArgumentParser(description="bibtex entries via DBLP search")
parser.add_argument('query', type=str, help='query')
args = parser.parse_args()

data = subprocess.check_output( [ 'bib.get' , args.query ] )
object = json.loads( data.decode() )
results = object['results']
publications = results['publications']

choices = sorted( publications , key = lambda x : ( -int(x['date']) , x['title'] ) )

def menu(title, choices):
    body = [urwid.Text(title), urwid.Divider()]
    for c in choices:
        button = urwid.Button('{date} {title}'.format( **c ) )
        urwid.connect_signal(button, 'click', item_chosen, c)
        body.append(urwid.AttrMap(button, None, focus_map='reversed'))
    return urwid.ListBox(urwid.SimpleFocusListWalker(body))

def item_chosen(button, choice):

    original_widget = main.original_widget

    def back ( button ) :
        main.original_widget = original_widget

    head = urwid.Text(['You chose ', choice['title'], '\n'])
    url = choice['bibtex']['condensed']
    downloading = urwid.Text( ['Downloading ', url, '\n'])
    main.original_widget = urwid.Filler(urwid.Pile([head,downloading]))

    async_loop.create_task(display(url, choice , back))



async def display ( url , choice , back ) :

    try:

        if url not in _cache:

            # with urllib.request.urlopen( url ) as response :
            async with aiohttp.ClientSession() as session:
                async with session.get(url) as response:
                    _cache[url] = await response.read( )

        raw = _cache[url]
        bibtex = raw.decode()
        result = urwid.Text(['Copied to clipboard\n\n',bibtex])
        p = subprocess.Popen( [ 'xsel' , '-bi' ] , stdin = subprocess.PIPE )
        p.communicate( raw )

    # except urllib.error.HTTPError:
    except aiohttp.ClientError:
        result = urwid.Text(['Failed to download ', url, '\n'])

    head = urwid.Text(['You chose ', choice['title'], '\n'])
    done_button = urwid.Button('Exit')
    urwid.connect_signal(done_button, 'click', exit_program)
    back_button = urwid.Button('Back')
    urwid.connect_signal(back_button, 'click', back)
    main.original_widget = urwid.Filler(urwid.Pile([
        head,
        result,
        urwid.AttrMap(back_button, None, focus_map='reversed'),
        urwid.AttrMap(done_button, None, focus_map='reversed'),
    ]))

def exit_program(button):
    raise urwid.ExitMainLoop()

async_loop = asyncio.get_event_loop()
event_loop = urwid.AsyncioEventLoop(loop=async_loop)

main = urwid.Padding(menu('List of publications', choices), left=2, right=2)
top = urwid.Overlay(main, urwid.SolidFill('\N{MEDIUM SHADE}'),
    align='center', width=('relative', 60),
    valign='middle', height=('relative', 60),
    min_width=20, min_height=9)
loop = urwid.MainLoop(
        top,
        palette=[('reversed', 'standout', '')],
        event_loop=event_loop,
)

loop.run()