mirror of
https://github.com/davegallant/rfd.git
synced 2025-08-06 00:33:40 +00:00
Compare commits
6 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
258ca59bdf | ||
|
e433a954cd | ||
|
b30a3c5b66 | ||
|
cfd7e01e43 | ||
|
f664cbd9c6 | ||
|
ee6939aafe |
9
Makefile
9
Makefile
@@ -32,15 +32,10 @@ lint:
|
||||
.PHONY: lint
|
||||
|
||||
## test: Run all unit tests
|
||||
test: tmp/.tests-passed.sentinel
|
||||
test:
|
||||
> pytest -vvv tests
|
||||
.PHONY: test
|
||||
|
||||
# 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)
|
||||
> pytest -v
|
||||
> touch $@
|
||||
|
||||
## pr: Run pre-commit, lint and test
|
||||
pr: precommit lint test
|
||||
.PHONY: pr
|
||||
|
24
README.md
24
README.md
@@ -5,12 +5,10 @@ Hot deals on the command line.
|
||||
[](https://travis-ci.org/davegallant/rfd)
|
||||
[](https://badge.fury.io/py/rfd)
|
||||
[](https://dependabot.com/)
|
||||
[](https://lgtm.com/projects/g/davegallant/rfd/context:python)
|
||||
|
||||
[](https://pepy.tech/project/rfd)
|
||||
|
||||

|
||||
|
||||
|
||||
## Install
|
||||
|
||||
```bash
|
||||
@@ -36,7 +34,14 @@ Commands:
|
||||
|
||||
## Examples
|
||||
|
||||
All commands open up in a pager.
|
||||
|
||||
Search can be done using `/`.
|
||||
|
||||
Close pager with `q`.
|
||||
|
||||
### View Hot Deals
|
||||
|
||||
```console
|
||||
$ rfd threads
|
||||
```
|
||||
@@ -48,20 +53,27 @@ $ rfd threads --sort-by score
|
||||
```
|
||||
|
||||
```console
|
||||
$ rfd threads --sort-by views --limit 40
|
||||
$ rfd threads --sort-by views --pages 10
|
||||
```
|
||||
|
||||
### Simple Search
|
||||
|
||||
```console
|
||||
$ rfd search 'pizza'
|
||||
```
|
||||
|
||||
### RegEx Search
|
||||
### Advanced Search
|
||||
|
||||
Regular expressions can be used for search.
|
||||
|
||||
```console
|
||||
$ rfd search '(coffee|starbucks)' --num-pages 100
|
||||
$ rfd search '(coffee|starbucks)' --pages 10 --sort-by views
|
||||
```
|
||||
|
||||
### View Posts
|
||||
|
||||
```console
|
||||
$ rfd posts https://forums.redflagdeals.com/kobo-vs-kindle-2396227/
|
||||
```
|
||||
|
||||
## Shell Completion
|
||||
|
@@ -1 +1 @@
|
||||
0.4.1
|
||||
0.6.0
|
||||
|
24
rfd/api.py
24
rfd/api.py
@@ -33,28 +33,30 @@ def create_user_map(users):
|
||||
return m
|
||||
|
||||
|
||||
def get_threads(forum_id, limit, page=1):
|
||||
def get_threads(forum_id, pages):
|
||||
"""Get threads from rfd api
|
||||
|
||||
Arguments:
|
||||
forum_id {int} -- forum id
|
||||
limit {[type]} -- limit number of threads returned
|
||||
pages {int} -- the number of pages of threads to collect
|
||||
|
||||
Returns:
|
||||
dict -- api response
|
||||
"""
|
||||
threads = []
|
||||
try:
|
||||
response = requests.get(
|
||||
"{}/api/topics?forum_id={}&per_page={}&page={}".format(
|
||||
API_BASE_URL, forum_id, get_safe_per_page(limit), page
|
||||
for page in range(1, pages + 1):
|
||||
response = requests.get(
|
||||
"{}/api/topics?forum_id={}&per_page=40&page={}".format(
|
||||
API_BASE_URL, forum_id, page
|
||||
)
|
||||
)
|
||||
)
|
||||
if response.status_code == 200:
|
||||
return response.json()
|
||||
logging.error("Unable to retrieve threads. %s", response.text)
|
||||
if response.status_code != 200:
|
||||
raise Exception("When collecting threads, received a status code: %s" % response.status_code)
|
||||
threads += response.json().get("topics")
|
||||
except JSONDecodeError as err:
|
||||
logging.error("Unable to retrieve threads. %s", err)
|
||||
return None
|
||||
logging.error("Unable to decode threads. %s", err)
|
||||
return threads
|
||||
|
||||
|
||||
def get_posts(post):
|
||||
|
85
rfd/cli.py
85
rfd/cli.py
@@ -2,16 +2,15 @@ from __future__ import unicode_literals
|
||||
|
||||
|
||||
import logging
|
||||
import os
|
||||
import sys
|
||||
import click
|
||||
from colorama import init, Fore, Style
|
||||
from colorama import init
|
||||
from .api import get_threads, get_posts
|
||||
from .threads import parse_threads, search_threads, sort_threads
|
||||
from .threads import parse_threads, search_threads, sort_threads, generate_thread_output
|
||||
from .posts import generate_posts_output
|
||||
from .__version__ import version as current_version
|
||||
|
||||
init()
|
||||
print()
|
||||
|
||||
logging.getLogger()
|
||||
logging.getLogger().setLevel(logging.INFO)
|
||||
@@ -21,20 +20,6 @@ logging.getLogger().addHandler(logging.StreamHandler())
|
||||
def get_version():
|
||||
return "rfd v" + current_version
|
||||
|
||||
|
||||
def get_terminal_width():
|
||||
_, columns = os.popen("stty size", "r").read().split()
|
||||
return int(columns)
|
||||
|
||||
|
||||
def get_vote_color(score):
|
||||
if score > 0:
|
||||
return Fore.GREEN + " [+" + str(score) + "] "
|
||||
if score < 0:
|
||||
return Fore.RED + " [" + str(score) + "] "
|
||||
return Fore.BLUE + " [" + str(score) + "] "
|
||||
|
||||
|
||||
def print_version(ctx, value):
|
||||
if not value or ctx.resilient_parsing:
|
||||
return
|
||||
@@ -42,27 +27,6 @@ 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)
|
||||
|
||||
|
||||
@click.group(invoke_without_command=True)
|
||||
@click.option(
|
||||
"-v",
|
||||
@@ -93,18 +57,7 @@ def posts(post_id):
|
||||
"""
|
||||
|
||||
try:
|
||||
click.echo("-" * get_terminal_width())
|
||||
for post in get_posts(post=post_id):
|
||||
click.echo(
|
||||
" -"
|
||||
+ get_vote_color(post.score)
|
||||
+ Fore.RESET
|
||||
+ post.body
|
||||
+ Fore.YELLOW
|
||||
+ " ({})".format(post.user)
|
||||
)
|
||||
click.echo(Style.RESET_ALL)
|
||||
click.echo("-" * get_terminal_width())
|
||||
click.echo_via_pager(generate_posts_output(get_posts(post=post_id)))
|
||||
except ValueError:
|
||||
click.echo("Invalid post id.")
|
||||
sys.exit(1)
|
||||
@@ -115,9 +68,9 @@ def posts(post_id):
|
||||
|
||||
@cli.command(short_help="Displays threads in the forum. Defaults to hot deals.")
|
||||
@click.option("--forum-id", default=9, help="The forum id number")
|
||||
@click.option("--limit", default=10, help="Number of threads.")
|
||||
@click.option("--pages", default=1, help="Number of pages to show. Defaults to 1.")
|
||||
@click.option("--sort-by", default=None, help="Sort threads by")
|
||||
def threads(limit, forum_id, sort_by):
|
||||
def threads(forum_id, pages, sort_by):
|
||||
"""Display threads in the specified forum id. Defaults to 9 (hot deals).
|
||||
|
||||
Popular forum ids:
|
||||
@@ -134,20 +87,18 @@ def threads(limit, forum_id, sort_by):
|
||||
74 \t shopping discussion
|
||||
88 \t cell phones
|
||||
"""
|
||||
_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)
|
||||
_threads = sort_threads(parse_threads(get_threads(forum_id, pages)), sort_by=sort_by)
|
||||
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 +116,11 @@ def search(num_pages, forum_id, regex):
|
||||
88 \t cell phones
|
||||
"""
|
||||
|
||||
count = 0
|
||||
for page in range(1, num_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 = []
|
||||
|
||||
_threads = parse_threads(get_threads(forum_id, pages=pages))
|
||||
for thread in search_threads(threads=_threads, regex=regex):
|
||||
matched_threads.append(thread)
|
||||
click.echo_via_pager(
|
||||
generate_thread_output(sort_threads(matched_threads, sort_by=sort_by))
|
||||
)
|
||||
|
26
rfd/posts.py
26
rfd/posts.py
@@ -1,8 +1,32 @@
|
||||
# pylint: disable=old-style-class
|
||||
|
||||
import os
|
||||
from colorama import Fore, Style
|
||||
from .scores import get_vote_color
|
||||
|
||||
class Post:
|
||||
def __init__(self, body, score, user):
|
||||
self.body = body
|
||||
self.score = score
|
||||
self.user = user
|
||||
|
||||
def get_terminal_width():
|
||||
_, columns = os.popen("stty size", "r").read().split()
|
||||
return int(columns)
|
||||
|
||||
def generate_posts_output(posts):
|
||||
output = ""
|
||||
output += ("-" * get_terminal_width())
|
||||
for post in posts:
|
||||
output += (
|
||||
" -"
|
||||
+ get_vote_color(post.score)
|
||||
+ Fore.RESET
|
||||
+ post.body
|
||||
+ Fore.YELLOW
|
||||
+ " ({})".format(post.user)
|
||||
)
|
||||
output += (Style.RESET_ALL)
|
||||
output += "\n"
|
||||
output += ("-" * get_terminal_width())
|
||||
output += "\n"
|
||||
return output
|
||||
|
@@ -1,3 +1,6 @@
|
||||
from colorama import Fore
|
||||
|
||||
|
||||
def calculate_score(post):
|
||||
"""Calculate either topic or post score. If votes cannot be retrieved, the score is 0.
|
||||
|
||||
@@ -16,3 +19,11 @@ def calculate_score(post):
|
||||
pass
|
||||
|
||||
return score
|
||||
|
||||
|
||||
def get_vote_color(score):
|
||||
if score > 0:
|
||||
return Fore.GREEN + " [+" + str(score) + "] "
|
||||
if score < 0:
|
||||
return Fore.RED + " [" + str(score) + "] "
|
||||
return Fore.BLUE + " [" + str(score) + "] "
|
||||
|
@@ -1,6 +1,7 @@
|
||||
import re
|
||||
from colorama import Fore, Style
|
||||
from . import API_BASE_URL
|
||||
from .scores import calculate_score
|
||||
from .scores import calculate_score, get_vote_color
|
||||
|
||||
# pylint: disable=old-style-class
|
||||
class Thread:
|
||||
@@ -26,12 +27,11 @@ def get_dealer(topic):
|
||||
return dealer
|
||||
|
||||
|
||||
def parse_threads(threads, limit):
|
||||
def parse_threads(threads):
|
||||
"""Parse topics list api response into digestible list.
|
||||
|
||||
Arguments:
|
||||
threads {dict} -- topics response from rfd api
|
||||
limit {int} -- limit number of threads returned
|
||||
|
||||
Returns:
|
||||
list(dict) -- digestible list of threads
|
||||
@@ -39,9 +39,7 @@ def parse_threads(threads, limit):
|
||||
parsed_threads = []
|
||||
if threads is None:
|
||||
return []
|
||||
for count, topic in enumerate(threads.get("topics"), start=1):
|
||||
if count > limit:
|
||||
break
|
||||
for topic in threads:
|
||||
parsed_threads.append(
|
||||
Thread(
|
||||
title=topic.get("title"),
|
||||
@@ -74,3 +72,28 @@ def search_threads(threads, regex):
|
||||
deal.dealer_name and regexp.search(deal.dealer_name.lower())
|
||||
):
|
||||
yield deal
|
||||
|
||||
|
||||
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
|
||||
|
25
tests/integration/test_cli.py
Normal file
25
tests/integration/test_cli.py
Normal file
@@ -0,0 +1,25 @@
|
||||
from subprocess import Popen, PIPE
|
||||
import pytest
|
||||
|
||||
|
||||
def run_cli(args):
|
||||
cmd = ["python", "-m", "rfd"] + args.split()
|
||||
p = Popen(cmd, stdout=PIPE)
|
||||
stdout, _ = p.communicate()
|
||||
assert p.returncode == 0
|
||||
return stdout
|
||||
|
||||
|
||||
def test_version():
|
||||
stdout = run_cli("--version")
|
||||
assert b"rfd v" in stdout
|
||||
|
||||
|
||||
@pytest.mark.parametrize("args", ["", "--sort-by score"])
|
||||
def test_threads(args):
|
||||
run_cli("threads " + args)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("args", ["'pizza'", "'(coffee|starbucks)'"])
|
||||
def test_search(args):
|
||||
run_cli("search " + args)
|
@@ -19,11 +19,10 @@ def test_extract_post_id():
|
||||
|
||||
def test_parse_threads(threads_api_response):
|
||||
|
||||
limit = 10
|
||||
threads = parse_threads(threads_api_response, limit)
|
||||
assert len(threads) == limit
|
||||
threads = parse_threads(threads_api_response.get("topics"))
|
||||
assert len(threads) == 10
|
||||
|
||||
|
||||
def test_parse_threads_empty():
|
||||
|
||||
assert parse_threads(None, 10) == []
|
||||
assert parse_threads(None) == []
|
||||
|
Reference in New Issue
Block a user