Merge branch 'main' into refactor-01

blocklist.py

@@ -0,0 +1,11 @@
+class BlockList:
+    """A class holding methods to determine if a company is blocklisted"""
+
+    BLOCKLISTED_COMPANIES = set(
+        ["Pattern Learning AI - Career & Tech Recruitment Reimagined!"]
+    )
+
+    def is_blacklisted_company(self, company: str) -> bool:
+        """Determines if the company is blacklisted or not"""
+
+        return True if company in self.BLOCKLISTED_COMPANIES else False

utility/utils.py

@@ -5,7 +5,8 @@
 import argparse
 import json
 from bs4 import BeautifulSoup
-from opportunity import Opportunity, OpportunityType
+from opportunity import Opportunity
+from blocklist import BlockList
 
 # ----------------- FOR CLI LIBRARY COMMAND -----------------
 
@@ -93,23 +94,25 @@ def blueprint_opportunity_formatter(
     internship_list = []
     for elem in div:
         company = elem.find(class_=company_elem).text.strip()
-        title = elem.find(class_=title_elem).text.strip()
-        location = elem.find(class_=location_elem).text.strip()
-        link = elem.find(class_=link_elem)["href"].split("?")[0]
-        processed = 0
-
-        date_difference = calculate_day_difference(elem)
-        if len(internship_list) < len_of_jobs:
-            if date_limit and int(days_needed_command_value) >= date_difference:
-                opportunity = Opportunity(
-                    company, title, location, link, processed, opp_type
-                )
-            else:
-                opportunity = Opportunity(
-                    company, title, location, link, processed, opp_type
-                )
-
-            internship_list.append(opportunity)
+        if not BlockList().is_blacklisted_company(company):
+            title = elem.find(class_=title_elem).text.strip()
+            location = elem.find(class_=location_elem).text.strip()
+            link = elem.find(class_=link_elem)["href"].split("?")[0]
+            processed = 0
+
+            date_difference = calculate_day_difference(elem)
+
+            if len(internship_list) < len_of_jobs:
+                if date_limit and int(days_needed_command_value) >= date_difference:
+                    opportunity = Opportunity(
+                        company, title, location, link, processed, opp_type
+                    )
+                else:
+                    opportunity = Opportunity(
+                        company, title, location, link, processed, opp_type
+                    )
+
+                internship_list.append(opportunity)
 
     return internship_list