diff --git a/connectors/reddit_connector.py b/connectors/reddit_api.py similarity index 89% rename from connectors/reddit_connector.py rename to connectors/reddit_api.py index 053576e..0488ab1 100644 --- a/connectors/reddit_connector.py +++ b/connectors/reddit_api.py @@ -2,7 +2,7 @@ from dto.post import Post from dto.user import User import requests -class RedditConnector: +class RedditAPI: def __init__(self): self.url = "https://www.reddit.com/" self.source_name = "Reddit" @@ -29,6 +29,14 @@ class RedditConnector: data = self._fetch_data(url, params) return self._parse_posts(data) + def get_new_subreddit_posts(self, subreddit: str, limit: int = 10) -> list[Post]: + params = { + 'limit': limit + } + url = f"r/{subreddit}/new.json" + data = self._fetch_data(url, params) + return self._parse_posts(data) + def get_user(self, username: str) -> User: data = self._fetch_data(f"user/{username}/about.json", {}) return self._parse_user(data) diff --git a/server/app.py b/server/app.py index 917df87..a1c1d79 100644 --- a/server/app.py +++ b/server/app.py @@ -1,12 +1,12 @@ from flask import Flask from db.database import Database -from connectors.reddit_connector import RedditConnector +from connectors.reddit_api import RedditAPI from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer app = Flask(__name__) db = Database(db_name='ethnograph', user='ethnograph_user', password='ethnograph_pass') -reddit_connector = RedditConnector() +reddit_connector = RedditAPI() @app.route('/fetch_subreddit//', methods=['GET']) def fetch_subreddit(subreddit, limit = 10):