コード例 #1
0
    def __init__(self, stdscr):
        self.stdscr = stdscr
        self.reddit_client = RedditClient()
        self.current_posts = []

        # get screen dimensions
        self.__max_y, self.__max_x = self.stdscr.getmaxyx()
コード例 #2
0
class CursesClient(object):
    """
    Handles drawing the text based UI
    """
    __FLUSH = 0
    __MARGIN_Y = 2
    __MARGIN_X = 2

    def __init__(self, stdscr):
        self.stdscr = stdscr
        self.reddit_client = RedditClient()
        self.current_posts = []

        # get screen dimensions
        self.__max_y, self.__max_x = self.stdscr.getmaxyx()

    def __draw_scr_size(self):
        """
        Debug output to determine drawable screen size
        :return: void
        """
        str_scr_size = "{0}, {1}".format(self.__FLUSH, self.__max_x)
        self.stdscr.addstr(
            self.__FLUSH,
            self.__max_x - self.__MARGIN_X - len(str_scr_size),
            str_scr_size
        )

    def __draw_prompt(self):
        """
        Draw placeholder prompt
        :return: void
        """
        self.stdscr.addstr(
            self.__max_y - self.__MARGIN_Y,
            self.__FLUSH,
            ">"
        )

    def __draw_ch(self, input_char):
        """
        Debug output to draw last pressed key
        :param input_char: char of last input key
        :return: void
        """
        self.stdscr.addstr(
            self.__max_y - self.__MARGIN_Y,
            self.__FLUSH,
            ">{0}".format(str(input_char))
        )

    def __draw_page_num(self, page):
        """
        Draw placeholder current page number
        :param page: int page number
        :return: void
        """
        str_page_num = "page: {0}".format(page)
        self.stdscr.addstr(
            self.__max_y - self.__MARGIN_Y,
            self.__max_x - self.__MARGIN_X - len(str_page_num),
            str_page_num
        )

    def __draw_subreddit_name(self, subreddit, sort):
        """
        Draw placeholder subreddit name
        :param subreddit: string subreddit
        :param sort: string sort e.g. new, top, hot, etc
        :return:
        """
        str_scr_size = "r/{subreddit}/{sort}".format(
            subreddit=subreddit,
            sort=sort
        )
        self.stdscr.addstr(
            self.__FLUSH,
            self.__max_x - self.__MARGIN_X - len(str_scr_size),
            str_scr_size
        )

    def __subreddit_menu(self):
        """
        Switch to buffered input and prompt for subreddit name
        :return:  void
        """

        #flush input buffer
        flushinp()

        str_input_subreddit = "subreddit:"
        self.stdscr.addstr(
            self.__max_y - self.__MARGIN_Y,
            self.__FLUSH,
            str_input_subreddit
        )

        # disable cinput mode and enter buffered input mode and echo
        nocbreak()
        echo()
        input_buffer = self.stdscr.getstr()
        if input_buffer == "back":
            # enable cinput mode and disable echo
            cbreak()
            noecho()
        else:
            self.__posts_menu(input_buffer)

    def __posts_menu(self, subreddit):
        """
        Return to cinput mode and wait for key input
        :param subreddit: subreddit string
        :return: void
        """

        # enable cinput mode and disable echo
        cbreak()
        noecho()

        input_char = '_'
        direction = None
        anchor = None
        while input_char != ord('q'):

            # paging
            if input_char == KEY_RIGHT:
                direction = "after"
                anchor = self.current_posts[9]["data"]["name"]
            elif input_char == KEY_LEFT:
                direction = "before"
                anchor = self.current_posts[0]["data"]["name"]

            # post selection
            elif input_char >= 48 and input_char <= 57:
                self.__post_detail(self.current_posts[input_char - 48])

            # subreddit sorting
            elif input_char == ord('n'):
                sort = "new"
            elif input_char == ord('t'):
                sort = "top"
            elif input_char == ord('c'):
                sort = "controversial"
            elif input_char == ord('h'):
                sort = "hot"
            else:
                sort = "hot"

            self.current_posts = self.reddit_client.get_subreddit_posts(
                subreddit,
                sort,
                direction=direction,
                anchor=anchor
            )

            self.stdscr.clear()
            self.__draw_subreddit_name(subreddit, sort)
            self.__draw_ch(input_char)
            self.__draw_page_num(1)

            for post_num in range(len(self.current_posts)):
                post_title = filter_unicode(
                    self.current_posts[post_num]["data"]["title"]
                )
                self.stdscr.addstr(
                    post_num,
                    self.__FLUSH,
                    "{post_num} -> {title}".format(
                        post_num=post_num,
                        title=post_title
                    )
                )

            input_char = self.stdscr.getch()

    def __post_detail(self, post):
        """
        Placeholder display of post text, with placeholder text limit
        :param post: post self text string
        :return: void
        """

        input_char = ''
        while input_char != ord('q'):
            self.stdscr.clear()
            self.__draw_ch(input_char)

            # filter out unicode
            str_post = filter_unicode(post["data"]["selftext"])

            self.stdscr.addstr(
                self.__FLUSH,
                self.__FLUSH,
                str_post[:2000] or "No text found"
            )

            input_char = self.stdscr.getch()

    def top_menu(self):
        """
        Placeholder top level menu, only r for subreddits and q to quit
        :return: void
        """

        input_char = ''
        while input_char != ord('q'):
            self.stdscr.clear()
            self.__draw_scr_size()
            self.__draw_prompt()
            input_char = self.stdscr.getch()

            if input_char == ord('r'):
                self.__subreddit_menu()

    def run(self):
        """
        Main entry point
        :return: void
        """
        self.top_menu()
コード例 #3
0
class TestRedditClient(object):

    @classmethod
    def setup_class(self):
        self.reddit_client = RedditClient()

    def test_access_token_not_none(self):
        assert_is_not_none(self.reddit_client.access_token)
        pprint(self.reddit_client.access_token)

    def test_get_subreddit_sidebar(self):
        json = self.reddit_client.get_subreddit_sidebar("civ")
        assert_is_not_none(json)
        assert_is_not_none(json["data"]["description"])
        assert_is_not_none(json["data"]["header_title"])
        assert_is_not_none(json["data"]["public_description"])
        pprint(json)

    def test_get_subreddit_hot(self):
        json = self.reddit_client.get_subreddit_posts("civ", "hot")
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)

    def test_get_subreddit_top(self):
        json = self.reddit_client.get_subreddit_posts("civ", "top")
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)

    def test_get_subreddit_new(self):
        json = self.reddit_client.get_subreddit_posts("civ", "new")
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)

    def test_get_subreddit_controversial(self):
        json = self.reddit_client.get_subreddit_posts("civ", "controversial")
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)

    def test_get_subreddit_page_after(self):
        anchor_json = self.reddit_client.get_subreddit_posts("civ", "new")
        anchor = anchor_json[0]["data"]["name"]
        json = self.reddit_client.get_subreddit_posts(
            "civ",
            "new",
            "after",
            anchor
        )
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)

    def test_get_subreddit_page_before(self):
        anchor_after_json = self.reddit_client.get_subreddit_posts("civ", "new")
        anchor_after = anchor_after_json[0]["data"]["name"]
        anchor_before_json = self.reddit_client.get_subreddit_posts(
            "civ",
            "new",
            "after",
            anchor_after
        )
        anchor_before = anchor_before_json[0]["data"]["name"]
        json = self.reddit_client.get_subreddit_posts(
            "civ",
            "new",
            "before",
            anchor_before
        )
        assert_is_not_none(json)
        assert_is_not_none(json[0]["data"]["name"])
        assert_is_not_none(json[0]["data"]["title"])
        assert_is_not_none(json[0]["data"]["selftext"])
        pprint(json)
コード例 #4
0
 def setup_class(self):
     self.reddit_client = RedditClient()