Пример #1
0
def main():
    # Setup argument parser
    parser = ArgumentParser()
    parser.add_argument('pdfs', 
                        help='Input PDFs (directory or file with path list)')
    parser.add_argument('--with-unknown', action='store_true', default=False, 
                        help='Display queries for unknown files')
    parser.add_argument('--with-waiting', action='store_true', default=False, 
                        help='Display pending queries')
    
    # Process arguments
    args = parser.parse_args()
    
    pdfs = sorted(utility.get_pdfs(args.pdfs))
    handler = PdfrateQueryHandler()
    for pdf in pdfs:
        report = handler.poll(pdf)
        if not args.with_unknown and report['status'] == 'unknown':
            continue
        if not args.with_waiting and report['status'] == 'waiting':
            continue
        sys.stdout.write('{}: {}'.format(report['filename'], report['status']))
        if report['status'] in ['success', 'nometadata']:
            r = max(report['results'].keys())
            sys.stdout.write(' [{}%]'.format(report['results'][r]
                                             ['contagio_bm']))
        sys.stdout.write('\n')
    return 0