6 Commits

Author SHA1 Message Date
Dave Gallant
258ca59bdf Bump to 0.6.0 (#82) 2020-08-16 19:34:09 -04:00
Dave Gallant
e433a954cd Update README.md (#81) 2020-08-16 16:36:33 -04:00
Dave Gallant
b30a3c5b66 Set default of threads command to 1 page and 40 threads (#80)
* Set default of threads command to 1 page and 40 threads

* Echo posts in a pager
2020-08-16 15:07:48 -04:00
Dave Gallant
cfd7e01e43 Add integration tests (#79)
Adds integration tests in pytest to detect breaking changes.
2020-08-15 20:55:10 -04:00
Dave Gallant
f664cbd9c6 Add download count to README.md 2020-08-03 19:13:46 -04:00
Dave Gallant
ee6939aafe feat: change output to echo via pager 2020-08-03 15:48:39 -04:00
11 changed files with 146 additions and 107 deletions

View File

@@ -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

View File

@@ -5,12 +5,10 @@ Hot deals on the command line.
[![Build Status](https://travis-ci.org/davegallant/rfd.svg?branch=master)](https://travis-ci.org/davegallant/rfd)
[![PyPI version](https://badge.fury.io/py/rfd.svg)](https://badge.fury.io/py/rfd)
[![Dependabot](https://badgen.net/badge/Dependabot/enabled/green?icon=dependabot)](https://dependabot.com/)
[![Language grade: Python](https://img.shields.io/lgtm/grade/python/g/davegallant/rfd.svg?logo=lgtm&logoWidth=18)](https://lgtm.com/projects/g/davegallant/rfd/context:python)
[![Downloads](https://pepy.tech/badge/rfd)](https://pepy.tech/project/rfd)
![screenshot](https://user-images.githubusercontent.com/4519234/85969861-e10a4100-b996-11ea-9a31-6203322c60ee.png)
## 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

View File

@@ -1 +1 @@
0.4.1
0.6.0

View File

@@ -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):

View File

@@ -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))
)

View File

@@ -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

View File

@@ -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) + "] "

View File

@@ -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

View 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)

View File

@@ -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) == []

View File

@@ -9,8 +9,3 @@ envlist = py{27,
passenv = SSH_AUTH_SOCK
commands =
make ci
rfd --version
rfd threads
rfd threads --sort-by score
rfd search 'pizza'
rfd search '(coffee|starbucks)'