diff --git a/server/connectors/reddit_api.py b/server/connectors/reddit_api.py index 0ec6100..61f3656 100644 --- a/server/connectors/reddit_api.py +++ b/server/connectors/reddit_api.py @@ -5,10 +5,11 @@ import time from dto.post import Post from dto.user import User from dto.comment import Comment +from server.connectors.base import BaseConnector logger = logging.getLogger(__name__) -class RedditAPI: +class RedditAPI(BaseConnector): def __init__(self): self.url = "https://www.reddit.com/" self.source_name = "Reddit" diff --git a/server/connectors/youtube_api.py b/server/connectors/youtube_api.py index d0e00a3..71ce6ed 100644 --- a/server/connectors/youtube_api.py +++ b/server/connectors/youtube_api.py @@ -6,12 +6,13 @@ from googleapiclient.discovery import build from googleapiclient.errors import HttpError from dto.post import Post from dto.comment import Comment +from server.connectors.base import BaseConnector load_dotenv() API_KEY = os.getenv("YOUTUBE_API_KEY") -class YouTubeAPI: +class YouTubeAPI(BaseConnector): def __init__(self): self.youtube = build('youtube', 'v3', developerKey=API_KEY)