rename reddit_connecter to reddit_api

This commit is contained in:
2026-01-13 14:45:20 +00:00
parent 7c01c335fa
commit b5624035ec
2 changed files with 11 additions and 3 deletions

View File

@@ -2,7 +2,7 @@ from dto.post import Post
from dto.user import User from dto.user import User
import requests import requests
class RedditConnector: class RedditAPI:
def __init__(self): def __init__(self):
self.url = "https://www.reddit.com/" self.url = "https://www.reddit.com/"
self.source_name = "Reddit" self.source_name = "Reddit"
@@ -29,6 +29,14 @@ class RedditConnector:
data = self._fetch_data(url, params) data = self._fetch_data(url, params)
return self._parse_posts(data) 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: def get_user(self, username: str) -> User:
data = self._fetch_data(f"user/{username}/about.json", {}) data = self._fetch_data(f"user/{username}/about.json", {})
return self._parse_user(data) return self._parse_user(data)

View File

@@ -1,12 +1,12 @@
from flask import Flask from flask import Flask
from db.database import Database from db.database import Database
from connectors.reddit_connector import RedditConnector from connectors.reddit_api import RedditAPI
from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer
app = Flask(__name__) app = Flask(__name__)
db = Database(db_name='ethnograph', user='ethnograph_user', password='ethnograph_pass') db = Database(db_name='ethnograph', user='ethnograph_user', password='ethnograph_pass')
reddit_connector = RedditConnector() reddit_connector = RedditAPI()
@app.route('/fetch_subreddit/<string:subreddit>/<int:limit>', methods=['GET']) @app.route('/fetch_subreddit/<string:subreddit>/<int:limit>', methods=['GET'])
def fetch_subreddit(subreddit, limit = 10): def fetch_subreddit(subreddit, limit = 10):