mirror of
https://github.com/davegallant/rfd.git
synced 2025-08-06 08:43:41 +00:00
dcd7ad5838
* instructions how to add completion. * argument completion won't come until click 7.0 * version bump. * drop 3.4 support. * remove python 3.7 * new gif. * new gif and use limit not count.
126 lines
3.4 KiB
Python
126 lines
3.4 KiB
Python
from __future__ import unicode_literals
|
|
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
import click
|
|
from colorama import init, Fore, Style
|
|
from rfd.api import parse_threads, get_threads, get_posts
|
|
from rfd.__version__ import __version__
|
|
|
|
init()
|
|
print()
|
|
|
|
logging.getLogger()
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
logging.getLogger().addHandler(logging.StreamHandler())
|
|
|
|
|
|
def get_version():
|
|
return 'rfd ' + __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) + "] "
|
|
elif score < 0:
|
|
return Fore.RED + " [" + str(score) + "] "
|
|
return Fore.BLUE + " [" + str(score) + "] "
|
|
|
|
|
|
@click.group(invoke_without_command=True)
|
|
@click.option('--version/--no-version', default=False)
|
|
@click.pass_context
|
|
def cli(ctx, version):
|
|
"""Welcome to the RFD CLI. (RedFlagDeals.com)"""
|
|
if version:
|
|
click.echo(get_version())
|
|
elif not ctx.invoked_subcommand:
|
|
click.echo(ctx.get_help())
|
|
|
|
|
|
@cli.command('version')
|
|
def display_version():
|
|
click.echo(get_version())
|
|
|
|
|
|
@cli.command(short_help="Displays posts in a specific thread.")
|
|
@click.option('--head', default=0, help='Number of topics. Default is 0, for all topics')
|
|
@click.option('--tail', default=0, help='Number of topics. Default is disabled. This will override head.')
|
|
@click.argument('post_id')
|
|
def posts(post_id, head, tail):
|
|
"""Displays posts in a specific thread.
|
|
|
|
post_id can be a full url or post id only
|
|
|
|
Example:
|
|
|
|
\b
|
|
url: https://forums.redflagdeals.com/koodo-targeted-public-mobile-12-120-koodo-5gb-40-no-referrals-2173603
|
|
post_id: 2173603
|
|
"""
|
|
if head < 0:
|
|
click.echo("Invalid head.")
|
|
sys.exit(1)
|
|
|
|
if tail < 0:
|
|
click.echo("Invalid tail.")
|
|
sys.exit(1)
|
|
|
|
# Tail overrides head
|
|
if tail > 0:
|
|
count = tail
|
|
else:
|
|
count = head
|
|
|
|
try:
|
|
click.echo("-" * get_terminal_width())
|
|
for post in get_posts(post=post_id, count=count, tail=tail > 0):
|
|
click.echo(" -" + get_vote_color(post.get('score')) + Fore.RESET +
|
|
post.get('body') + Fore.YELLOW + " ({})".format(post.get('user')))
|
|
click.echo(Style.RESET_ALL)
|
|
click.echo("-" * get_terminal_width())
|
|
except ValueError:
|
|
click.echo("Invalid post id.")
|
|
sys.exit(1)
|
|
except AttributeError:
|
|
click.echo("AttributeError: RFD API did not return expected data.")
|
|
|
|
|
|
@cli.command(short_help="Displays threads in the specified forum.")
|
|
@click.option('--limit', default=10, help='Number of topics.')
|
|
@click.argument('forum_id', default=9)
|
|
def threads(limit, forum_id):
|
|
"""Displays threads in the specified forum id. Defaults to 9.
|
|
|
|
Popular forum ids:
|
|
|
|
\b
|
|
9 \t hot deals
|
|
14 \t computer and electronics
|
|
15 \t offtopic
|
|
17 \t entertainment
|
|
18 \t food and drink
|
|
40 \t automotive
|
|
53 \t home and garden
|
|
67 \t fashion and apparel
|
|
74 \t shopping discussion
|
|
88 \t cell phones
|
|
"""
|
|
_threads = parse_threads(get_threads(forum_id, limit), limit)
|
|
for i, thread in enumerate(_threads, 1):
|
|
click.echo(" " + str(i) + "." +
|
|
get_vote_color(thread.get('score')) + Fore.RESET + thread.get('title'))
|
|
click.echo(Fore.BLUE + " {}".format(thread.get('url')))
|
|
click.echo(Style.RESET_ALL)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|