mirror of
https://github.com/davegallant/rfd.git
synced 2025-08-05 08:23:38 +00:00
feat: change output to echo via pager
This commit is contained in:
11
Makefile
11
Makefile
@@ -35,6 +35,15 @@ lint:
|
||||
test: tmp/.tests-passed.sentinel
|
||||
.PHONY: test
|
||||
|
||||
## examples: Run basic commands
|
||||
examples: tmp/.tests-passed.sentinel
|
||||
> rfd --version
|
||||
> rfd threads >/dev/null
|
||||
> rfd threads --sort-by score >/dev/null
|
||||
> rfd search 'pizza' >/dev/null
|
||||
> rfd search '(coffee|starbucks)' >/dev/null
|
||||
.PHONY: examples
|
||||
|
||||
# Tests - re-ran if any file under src has been changed since tmp/.tests-passed.sentinel was last touched
|
||||
tmp/.tests-passed.sentinel: $(shell find ${SRC} -type f)
|
||||
> mkdir -p $(@D)
|
||||
@@ -45,7 +54,7 @@ tmp/.tests-passed.sentinel: $(shell find ${SRC} -type f)
|
||||
pr: precommit lint test
|
||||
.PHONY: pr
|
||||
|
||||
ci: lint test
|
||||
ci: lint test examples
|
||||
.PHONY: ci
|
||||
|
||||
## help: Print this help message
|
||||
|
@@ -7,10 +7,8 @@ Hot deals on the command line.
|
||||
[](https://dependabot.com/)
|
||||
[](https://lgtm.com/projects/g/davegallant/rfd/context:python)
|
||||
|
||||
|
||||

|
||||
|
||||
|
||||
## Install
|
||||
|
||||
```bash
|
||||
@@ -37,6 +35,7 @@ Commands:
|
||||
## Examples
|
||||
|
||||
### View Hot Deals
|
||||
|
||||
```console
|
||||
$ rfd threads
|
||||
```
|
||||
@@ -52,6 +51,7 @@ $ rfd threads --sort-by views --limit 40
|
||||
```
|
||||
|
||||
### Simple Search
|
||||
|
||||
```console
|
||||
$ rfd search 'pizza'
|
||||
```
|
||||
@@ -61,7 +61,7 @@ $ rfd search 'pizza'
|
||||
Regular expressions can be used for search.
|
||||
|
||||
```console
|
||||
$ rfd search '(coffee|starbucks)' --num-pages 100
|
||||
$ rfd search '(coffee|starbucks)' --pages 10 --sort-by views
|
||||
```
|
||||
|
||||
## Shell Completion
|
||||
|
@@ -1 +1 @@
|
||||
0.4.1
|
||||
0.5.0
|
||||
|
61
rfd/cli.py
61
rfd/cli.py
@@ -42,25 +42,29 @@ def print_version(ctx, value):
|
||||
ctx.exit()
|
||||
|
||||
|
||||
def display_thread(click, thread, count): # pylint: disable=redefined-outer-name
|
||||
dealer = thread.dealer_name
|
||||
if dealer and dealer is not None:
|
||||
dealer = "[" + dealer + "] "
|
||||
else:
|
||||
dealer = ""
|
||||
click.echo(
|
||||
" "
|
||||
+ str(count)
|
||||
+ "."
|
||||
+ get_vote_color(thread.score)
|
||||
+ Fore.RESET
|
||||
+ "%s%s" % (dealer, thread.title)
|
||||
+ Fore.LIGHTYELLOW_EX
|
||||
+ " (%d views)" % thread.views
|
||||
+ Fore.RESET
|
||||
)
|
||||
click.echo(Fore.BLUE + " {}".format(thread.url))
|
||||
click.echo(Style.RESET_ALL)
|
||||
def generate_thread_output(_threads):
|
||||
for count, thread in enumerate(_threads, 1):
|
||||
output = ""
|
||||
dealer = thread.dealer_name
|
||||
if dealer and dealer is not None:
|
||||
dealer = "[" + dealer + "] "
|
||||
else:
|
||||
dealer = ""
|
||||
output += (
|
||||
" "
|
||||
+ str(count)
|
||||
+ "."
|
||||
+ get_vote_color(thread.score)
|
||||
+ Fore.RESET
|
||||
+ "%s%s" % (dealer, thread.title)
|
||||
+ Fore.LIGHTYELLOW_EX
|
||||
+ " (%d views)" % thread.views
|
||||
+ Fore.RESET
|
||||
)
|
||||
output += Fore.BLUE + " {}".format(thread.url)
|
||||
output += Style.RESET_ALL
|
||||
output += "\n\n"
|
||||
yield output
|
||||
|
||||
|
||||
@click.group(invoke_without_command=True)
|
||||
@@ -137,17 +141,17 @@ def threads(limit, forum_id, sort_by):
|
||||
_threads = sort_threads(
|
||||
parse_threads(get_threads(forum_id, limit), limit), sort_by=sort_by
|
||||
)
|
||||
for count, thread in enumerate(_threads, 1):
|
||||
display_thread(click, thread, count)
|
||||
click.echo_via_pager(generate_thread_output(_threads))
|
||||
|
||||
|
||||
@cli.command(short_help="Search deals based on a regular expression.")
|
||||
@click.option("--num-pages", default=5, help="Number of pages to search.")
|
||||
@click.option("--pages", default=5, help="Number of pages to search.")
|
||||
@click.option(
|
||||
"--forum-id", default=9, help="The forum id number. Defaults to 9 (hot deals)."
|
||||
)
|
||||
@click.option("--sort-by", default=None, help="Sort threads by")
|
||||
@click.argument("regex")
|
||||
def search(num_pages, forum_id, regex):
|
||||
def search(pages, forum_id, sort_by, regex):
|
||||
"""Search deals based on regex.
|
||||
|
||||
Popular forum ids:
|
||||
@@ -165,9 +169,12 @@ def search(num_pages, forum_id, regex):
|
||||
88 \t cell phones
|
||||
"""
|
||||
|
||||
count = 0
|
||||
for page in range(1, num_pages):
|
||||
matched_threads = []
|
||||
|
||||
for page in range(1, pages):
|
||||
_threads = parse_threads(get_threads(forum_id, 100, page=page), limit=100)
|
||||
for thread in search_threads(threads=_threads, regex=regex):
|
||||
count += 1
|
||||
display_thread(click, thread, count)
|
||||
matched_threads.append(thread)
|
||||
click.echo_via_pager(
|
||||
generate_thread_output(sort_threads(matched_threads, sort_by=sort_by))
|
||||
)
|
||||
|
Reference in New Issue
Block a user