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"

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)