aurwatcher.py

#!/pusr/bin/env python3
#Usage example: aurwatcher p=emacs

import os
import sys
import argparse
import subprocess
import readline
import curses

from functools import reduce
from getch import read_single_char

def prepare_arguments():
    '''create CLI parser'''
    parser = argparse.ArgumentParser(prog="aurwatcher",
                                     description="script looks for packages in AUR or in official repository.")
    parser.add_argument("p",
                        help="defines package-name substring to look for")
    parser.add_argument("-s","--source",
                        help="choose source of finding.",
                        choices=["AUR","off"])
    parser.add_argument("-o",
                        "--output",
                        help   ="choose output mode(plain text xor inner pager)",
                        choices=["pt","ip"])
    return parser.parse_args()

def parse_arguments():
    '''This function is weird, I know, but it is required, because it's convert passed arguments to
       old format i used before started to use argparse.

       Note: i think it can be refactored.'''
    args = vars(prepare_arguments())
    result = {}
    result["o"] = "pt" if args["output"] is None else args["output"]
    result["s"] = "AUR" if args["source"] is None else args["source"]
    result["p"] =args["p"]
    return result

def compute_request(args):
    '''generate rpc-request'''
    source = args["s"]
    package = args["p"]

    if source.lower() not in ("off","aur"):
        print("{} is unknown source".format(source))
        sys.exit(1)

    if source == "AUR":
        return "https://aur.archlinux.org/rpc/?v=5&type=search&arg={}".format(package)
    else:
        return  "https://archlinux.org/packages/search/json/?q={}".format(package)


def get_source_related_keys_list(source):
    '''since response is big json file, then we need only specific keys'''
    if source == "AUR":
        return ("Name","Description","Maintainer","URL","Version","OutOfData")
    else:
        return ("pkgname","pkgdesc","packager","repo","arch","url")
    
def extract_required_info(response_result,source):
    '''response is json-file and this function filters it to return only required data'''
    data = {}
    for key,value in response_result.items():
        if key in get_source_related_keys_list(source):
            data[key] = value
    return data

def __print_items(item):
    print("-"*40)
    for k in item.keys():
        print("{}:{}".format(k,item[k]))
    print("-"*40)

def __save_result(item,filename):
    values = list()
    for k in item.keys():
        values.append("{}:{}".format(k,item[k]))
    data = str(reduce(lambda a,b: a+'\n'+b,values))
    try:
        with open(filename,"w") as f:
            f.write(data)
    except Exception:
        print("failed write to file!")


def print_result(response, paging_mode=False):
    '''print output as plain text or using "inner pager"'''
    paging_mode = True if paging_mode == "ip" else False

    fix_counter = False
    for i,item in enumerate(response):
        counter = int(i/3) if fix_counter else i
        print("found item #{}".format(counter))
        __print_items(item)
        if paging_mode:
            answer = read_single_char(fix_counter)
            if str(answer) == 'q':
                sys.exit(0)
            elif answer == 'w':
                subprocess.run(["clear"])
                infile = os.path.expanduser(input("file to write:"))
                if infile == "q":
                    __print_items(item)
                    sys.exit()
                    
                __save_result(item,infile)
                subprocess.run(["clear"])
                __print_items(item)
                sys.exit()
            else:
                subprocess.run(["clear"]) #check isn't required here, because clear won't fail