Simplify sort-by 'total_views' -> 'views' (#76)

* Add --sort-by flag to threads sub-command

* Add some sanity test commands to tox

.github/PULL_REQUEST_TEMPLATE.md

@@ -1,9 +1 @@
-
-**What this PR does / why we need it:**
--
-
-**Which issue(s) this PR fixes:**
--
-
-**Additional Notes:**
--
+###### Motivation for this change

.github/workflows/codeql-analysis.yml

@@ -0,0 +1,27 @@
+name: "Code scanning - action"
+
+on:
+  push:
+  pull_request:
+  schedule:
+    - cron: '0 4 * * 1'
+
+jobs:
+  CodeQL-Build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v2
+      with:
+        fetch-depth: 2
+
+    - run: git checkout HEAD^2
+      if: ${{ github.event_name == 'pull_request' }}
+
+    - name: Initialize CodeQL
+      uses: github/codeql-action/init@v1
+
+    - name: Perform CodeQL Analysis
+      uses: github/codeql-action/analyze@v1

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v2.3.0
+    rev: v2.5.0
     hooks:
     -   id: check-added-large-files
     -   id: check-ast

.travis.yml

@@ -1,5 +1,5 @@
 language: python
-dist: xenial
+dist: bionic
 sudo: false
 cache: false
 stages:
@@ -18,15 +18,14 @@ script:
 jobs:
   include:
     - python: "2.7"
-
     - python: "3.5"
-
     - python: "3.6"
-
     - python: "3.7"
+    - python: "3.8"
+    - python: "3.9-dev"
 
     - stage: deploy
-      python: "3.7"
+      python: "3.8"
       deploy:
         on:
           repo: davegallant/rfd

README.md

@@ -5,7 +5,6 @@ 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/)
-[![Total alerts](https://img.shields.io/lgtm/alerts/g/davegallant/rfd.svg?logo=lgtm&logoWidth=18)](https://lgtm.com/projects/g/davegallant/rfd/alerts/)
 [![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)
 
 
@@ -37,29 +36,46 @@ Commands:
 
 ## Examples
 
-### view hot deals
-```shell
-rfd threads
+### View Hot Deals
+```console
+$ rfd threads
 ```
 
-### search for pizza
-```shell
-rfd search 'pizza'
+### View and Sort Hot Deals
+
+```console
+$ rfd threads --sort-by score
 ```
 
-## Tab Completion
+```console
+$ rfd threads --sort-by views --limit 40
+```
 
-To enable:
+### Simple Search
+```console
+$ rfd search 'pizza'
+```
+
+### RegEx Search
+
+Regular expressions can be used for search.
+
+```console
+$ rfd search '(coffee|starbucks)' --num-pages 100
+```
+
+## Shell Completion
+
+Completion can be enabled if using `bash` or `zsh`.
 
 ### bash
 
-```bash
-echo 'eval "$(_RFD_COMPLETE=source rfd)"' >> ~/.profile
+```console
+$ echo 'eval "$(_RFD_COMPLETE=source rfd)"' >> ~/.profile
 ```
 
 ### zsh
 
-
-```zsh
-echo 'eval "$(_RFD_COMPLETE=source_zsh rfd)"' >> ~/.zshrc
+```console
+$ echo 'eval "$(_RFD_COMPLETE=source_zsh rfd)"' >> ~/.zshrc
 ```

rfd/VERSION

@@ -1 +1 @@
-0.3.5
+0.4.1

rfd/__init__.py

@@ -6,3 +6,5 @@ __title__ = "RFD CLI"
 __author__ = "Dave Gallant"
 __license__ = "Apache 2.0"
 __copyright__ = "(c) 2018 Dave Gallant"
+
+API_BASE_URL = "https://forums.redflagdeals.com"

rfd/api.py

@@ -6,11 +6,10 @@ except ImportError:
     JSONDecodeError = ValueError
 import logging
 import requests
-from .constants import API_BASE_URL
-from .format import strip_html, is_valid_url
-from .models import Post
+from . import API_BASE_URL
+from .posts import Post
 from .scores import calculate_score
-from .utils import is_int
+from .utils import is_int, strip_html, is_valid_url
 
 
 def extract_post_id(url):

rfd/cli.py

@@ -7,8 +7,7 @@ import sys
 import click
 from colorama import init, Fore, Style
 from .api import get_threads, get_posts
-from .search import search_threads
-from .parsing import parse_threads
+from .threads import parse_threads, search_threads, sort_threads
 from .__version__ import version as current_version
 
 init()
@@ -20,7 +19,7 @@ logging.getLogger().addHandler(logging.StreamHandler())
 
 
 def get_version():
-    return "rfd " + current_version
+    return "rfd v" + current_version
 
 
 def get_terminal_width():
@@ -39,10 +38,31 @@ def get_vote_color(score):
 def print_version(ctx, value):
     if not value or ctx.resilient_parsing:
         return
-    click.echo(get_version())
+    click.echo(get_version(), nl=False)
     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",
@@ -94,9 +114,10 @@ def posts(post_id):
 
 
 @cli.command(short_help="Displays threads in the forum. Defaults to hot deals.")
-@click.option("--limit", default=10, help="Number of topics.")
 @click.option("--forum-id", default=9, help="The forum id number")
-def threads(limit, forum_id):
+@click.option("--limit", default=10, help="Number of threads.")
+@click.option("--sort-by", default=None, help="Sort threads by")
+def threads(limit, forum_id, sort_by):
     """Display threads in the specified forum id. Defaults to 9 (hot deals).
 
     Popular forum ids:
@@ -113,21 +134,11 @@ def threads(limit, forum_id):
     74 \t shopping discussion
     88 \t cell phones
     """
-    _threads = parse_threads(get_threads(forum_id, limit), limit)
+    _threads = sort_threads(
+        parse_threads(get_threads(forum_id, limit), limit), sort_by=sort_by
+    )
     for count, thread in enumerate(_threads, 1):
-        click.echo(
-            " "
-            + str(count)
-            + "."
-            + get_vote_color(thread.score)
-            + Fore.RESET
-            + "[%s] %s" % (thread.dealer_name, thread.title)
-            + Fore.LIGHTYELLOW_EX
-            + " (%d views)" % thread.total_views
-            + Fore.RESET
-        )
-        click.echo(Fore.BLUE + " {}".format(thread.url))
-        click.echo(Style.RESET_ALL)
+        display_thread(click, thread, count)
 
 
 @cli.command(short_help="Search deals based on a regular expression.")
@@ -159,13 +170,4 @@ def search(num_pages, forum_id, regex):
         _threads = parse_threads(get_threads(forum_id, 100, page=page), limit=100)
         for thread in search_threads(threads=_threads, regex=regex):
             count += 1
-            click.echo(
-                " "
-                + str(count)
-                + "."
-                + get_vote_color(thread.score)
-                + Fore.RESET
-                + "[%s] %s" % (thread.dealer_name, thread.title)
-            )
-            click.echo(Fore.BLUE + " {}".format(thread.url))
-            click.echo(Style.RESET_ALL)
+            display_thread(click, thread, count)

rfd/constants.py

@@ -1 +0,0 @@
-API_BASE_URL = "https://forums.redflagdeals.com"

rfd/format.py

@@ -1,16 +0,0 @@
-"""Formatting utils"""
-
-try:
-    from urllib.parse import urlparse  # python 3
-except ImportError:
-    from urlparse import urlparse  # python 2
-from bs4 import BeautifulSoup
-
-
-def strip_html(text):
-    return BeautifulSoup(text, "html.parser").get_text()
-
-
-def is_valid_url(url):
-    result = urlparse(url)
-    return all([result.scheme, result.netloc, result.path])

rfd/models.py

@@ -1,18 +0,0 @@
-# pylint: disable=old-style-class
-class Thread:
-    def __init__(self, title, dealer_name, score, url, total_views):
-        self.dealer_name = dealer_name
-        self.score = score
-        self.title = title
-        self.url = url
-        self.total_views = total_views
-
-    def __repr__(self):
-        return "Thread(%s)" % self.title
-
-
-class Post:
-    def __init__(self, body, score, user):
-        self.body = body
-        self.score = score
-        self.user = user

rfd/parsing.py

@@ -1,35 +0,0 @@
-from .constants import API_BASE_URL
-from .scores import calculate_score
-from .models import Thread
-
-
-def build_web_path(slug):
-    return "{}{}".format(API_BASE_URL, slug)
-
-
-def parse_threads(threads, limit):
-    """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
-    """
-    parsed_threads = []
-    if threads is None:
-        return []
-    for count, topic in enumerate(threads.get("topics"), start=1):
-        if count > limit:
-            break
-        parsed_threads.append(
-            Thread(
-                title=topic.get("title"),
-                dealer_name=topic["offer"].get("dealer_name"),
-                score=calculate_score(topic),
-                url=build_web_path(topic.get("web_path")),
-                total_views=topic.get("total_views"),
-            )
-        )
-    return parsed_threads

rfd/posts.py

@@ -0,0 +1,8 @@
+# pylint: disable=old-style-class
+
+
+class Post:
+    def __init__(self, body, score, user):
+        self.body = body
+        self.score = score
+        self.user = user

rfd/search.py

@@ -1,14 +0,0 @@
-import re
-
-
-def search_threads(threads, regex):
-    """Match deal title and dealer names with regex specified."""
-
-    regexp = re.compile(str(regex).lower())
-
-    for deal in threads:
-
-        if regexp.search(deal.title.lower()) or (
-            deal.dealer_name and regexp.search(deal.dealer_name.lower())
-        ):
-            yield deal

rfd/threads.py

@@ -0,0 +1,76 @@
+import re
+from . import API_BASE_URL
+from .scores import calculate_score
+
+# pylint: disable=old-style-class
+class Thread:
+    def __init__(self, title, dealer_name, score, url, views):
+        self.dealer_name = dealer_name
+        self.score = score
+        self.title = title
+        self.url = url
+        self.views = views
+
+    def __repr__(self):
+        return "Thread(%s)" % self.title
+
+
+def build_web_path(slug):
+    return "{}{}".format(API_BASE_URL, slug)
+
+
+def get_dealer(topic):
+    dealer = None
+    if topic.get("offer"):
+        dealer = topic.get("offer").get("dealer_name")
+    return dealer
+
+
+def parse_threads(threads, limit):
+    """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
+    """
+    parsed_threads = []
+    if threads is None:
+        return []
+    for count, topic in enumerate(threads.get("topics"), start=1):
+        if count > limit:
+            break
+        parsed_threads.append(
+            Thread(
+                title=topic.get("title"),
+                dealer_name=get_dealer(topic),
+                score=calculate_score(topic),
+                url=build_web_path(topic.get("web_path")),
+                views=topic.get("total_views"),
+            )
+        )
+    return parsed_threads
+
+
+def sort_threads(threads, sort_by):
+    """Sort threads by an attribute"""
+    if sort_by is None:
+        return threads
+    assert sort_by in ["views", "score", "title"]
+    threads = sorted(threads, key=lambda x: getattr(x, sort_by), reverse=True)
+    return threads
+
+
+def search_threads(threads, regex):
+    """Match deal title and dealer names with regex specified."""
+
+    regexp = re.compile(str(regex).lower())
+
+    for deal in threads:
+
+        if regexp.search(deal.title.lower()) or (
+            deal.dealer_name and regexp.search(deal.dealer_name.lower())
+        ):
+            yield deal

rfd/utils.py

@@ -1,4 +1,18 @@
 """This module provides utility functions that are used within rfd"""
+try:
+    from urllib.parse import urlparse  # python 2
+except ImportError:
+    from urlparse import urlparse  # python 1
+from bs4 import BeautifulSoup
+
+
+def strip_html(text):
+    return BeautifulSoup(text, "html.parser").get_text()
+
+
+def is_valid_url(url):
+    result = urlparse(url)
+    return all([result.scheme, result.netloc, result.path])
 
 
 def is_int(number):

tests/test_api.py

@@ -1,5 +1,5 @@
 from rfd.api import extract_post_id
-from rfd.parsing import build_web_path, parse_threads
+from rfd.threads import build_web_path, parse_threads
 
 
 def test_build_web_path():

tox.ini

@@ -9,3 +9,8 @@ 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)'