Update utils/functions.py

utils/functions.py

@@ -23,6 +23,7 @@ logging.basicConfig(
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 DATA_DIR = os.path.join(BASE_DIR, '..', 'data')
 FAKE_NUMBERS_FILE = os.path.join(DATA_DIR, 'fake_numbers.json')
+STATS_FILE = os.path.join(DATA_DIR, 'stats.json')
 
 # Upewnij się, że katalog 'data' istnieje
 os.makedirs(DATA_DIR, exist_ok=True)
@@ -409,121 +410,101 @@ Provide your analysis and conclusions following the guidelines above."""
         logging.error(f"Błąd połączenia z API: {e}")
         return f"Błąd połączenia z API: {e}", "Błąd analizy.", "Błąd analizy."
 
-# Funkcje analizy stron internetowych
+# Funkcje analizy strony internetowej
 
-def check_url_with_phishtank(url_to_check, phishtank_api_key):
+def analyze_website(url, language):
     """
-    Sprawdza, czy podany URL znajduje się w bazie PhishTank.
+    Analizuje zawartość strony internetowej pod kątem zagrożeń.
     """
-    if not phishtank_api_key:
-        logging.error("Brak klucza API PhishTank.")
-        return False, "Brak klucza API PhishTank."
-
-    api_url = "https://checkurl.phishtank.com/checkurl/"
-    data = {
-        'format': 'json',
-        'app_key': phishtank_api_key,
-        'url': url_to_check
-    }
-
     try:
-        response = requests.post(api_url, data=data)
+        response = requests.get(url)
         if response.status_code == 200:
-            result = response.json()
-            is_phish = result.get('results', {}).get('valid', False)
-            return is_phish, None
+            content = response.text
+            # Wstaw tu kod analizy zawartości strony
+            return content  # Na razie zwracamy tylko zawartość
         else:
-            logging.error(f"Błąd podczas sprawdzania URL w PhishTank: {response.status_code}")
-            return False, f"Błąd podczas sprawdzania URL w PhishTank: {response.status_code}"
-    except Exception as e:
-        logging.error(f"Błąd podczas sprawdzania URL w PhishTank: {e}")
-        return False, f"Błąd podczas sprawdzania URL w PhishTank: {e}"
+            logging.error(f"Błąd podczas pobierania strony: {response.status_code}")
+            return "Błąd podczas pobierania strony."
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Błąd żądania: {e}")
+        return "Błąd żądania."
 
-def get_google_safe_browsing(url_to_check, google_api_key):
+def check_urls_with_phishtank(urls):
     """
-    Sprawdza, czy URL jest bezpieczny za pomocą Google Safe Browsing API.
+    Sprawdza URL w bazie PhishTank.
     """
-    if not google_api_key:
-        logging.error("Brak klucza API Google Safe Browsing.")
-        return False, "Brak klucza API Google Safe Browsing."
-
-    api_url = "https://safebrowsing.googleapis.com/v4/threatMatches:find"
-    params = {
-        'key': google_api_key
-    }
-    payload = {
-        "client": {
-            "clientId": "yourcompanyname",
-            "clientVersion": "1.5.2"
+    phishing_urls = []
+    for url in urls:
+        params = {
+            'format': 'json',
+            'url': url
+        }
+        try:
+            response = requests.post('https://checkurl.phishtank.com/checkurl/', data=params)
+            if response.status_code == 200:
+                data = response.json()
+                in_database = data.get('results', {}).get('in_database', False)
+                valid = data.get('results', {}).get('valid', False)
+                if in_database and valid:
+                    phishing_urls.append(url)
+            else:
+                logging.warning(f"Błąd podczas sprawdzania URL w PhishTank: {response.status_code}")
+        except Exception as e:
+            logging.error(f"Błąd podczas sprawdzania URL w PhishTank: {e}")
+    return phishing_urls
+
+def check_urls_with_safe_browsing(urls):
+    """
+    Sprawdza URL w Google Safe Browsing.
+    """
+    api_key = os.getenv('GOOGLE_SAFE_BROWSING_API_KEY')
+    if not api_key:
+        return None
+    unsafe_urls = []
+    headers = {'Content-Type': 'application/json'}
+    client_body = {
+        'client': {
+            'clientId': 'yourcompanyname',
+            'clientVersion': '1.0'
         },
-        "threatInfo": {
-            "threatTypes": ["MALWARE", "SOCIAL_ENGINEERING"],
-            "platformTypes": ["WINDOWS"],
-            "threatEntryTypes": ["URL"],
-            "threatEntries": [
-                {"url": url_to_check}
-            ]
+        'threatInfo': {
+            'threatTypes': ["MALWARE", "SOCIAL_ENGINEERING", "UNWANTED_SOFTWARE", "POTENTIALLY_HARMFUL_APPLICATION"],
+            'platformTypes': ["ANY_PLATFORM"],
+            'threatEntryTypes': ["URL"],
+            'threatEntries': [{'url': url} for url in urls]
         }
     }
-
     try:
-        response = requests.post(api_url, params=params, json=payload)
+        response = requests.post(
+            f'https://safebrowsing.googleapis.com/v4/threatMatches:find?key={api_key}',
+            headers=headers,
+            json=client_body
+        )
         if response.status_code == 200:
-            result = response.json()
-            is_threat = bool(result)
-            return is_threat, None
+            data = response.json()
+            matches = data.get('matches', [])
+            unsafe_urls = [match['threat']['url'] for match in matches]
         else:
-            logging.error(f"Błąd podczas sprawdzania URL w Google Safe Browsing: {response.status_code}")
-            return False, f"Błąd podczas sprawdzania URL w Google Safe Browsing: {response.status_code}"
+            logging.error(f"Błąd podczas komunikacji z Google Safe Browsing API: {response.status_code}")
     except Exception as e:
         logging.error(f"Błąd podczas sprawdzania URL w Google Safe Browsing: {e}")
-        return False, f"Błąd podczas sprawdzania URL w Google Safe Browsing: {e}"
+    return unsafe_urls
 
-def analyze_url(url, phishtank_api_key, google_api_key):
+def update_stats(fraud_detected=False):
     """
-    Analizuje stronę internetową pod kątem phishingu i innych zagrożeń.
+    Aktualizuje statystyki analiz w pliku stats.json.
     """
-    # Sprawdzenie PhishTank
-    is_phish, error = check_url_with_phishtank(url, phishtank_api_key)
-    if error:
-        phishing_result = error
-    elif is_phish:
-        phishing_result = "Zagrożenie phishingiem wykryte w PhishTank."
-    else:
-        phishing_result = "Brak zagrożeń phishingowych według PhishTank."
-
-    # Sprawdzenie Google Safe Browsing
-    is_threat, error = get_google_safe_browsing(url, google_api_key)
-    if error:
-        google_result = error
-    elif is_threat:
-        google_result = "Zagrożenie wykryte przez Google Safe Browsing."
-    else:
-        google_result = "Brak zagrożeń według Google Safe Browsing."
-
-    # Analiza zawartości strony
-    try:
-        response = requests.get(url)
-        if response.status_code == 200:
-            content = response.text[:500]  # Pobierz pierwsze 500 znaków
-            logging.info(f"Pobrano zawartość strony {url}.")
-            return phishing_result, google_result, content
-        else:
-            logging.error(f"Nie udało się pobrać zawartości strony {url}. Kod statusu HTTP: {response.status_code}")
-            return f"Nie udało się pobrać zawartości strony. Kod statusu HTTP: {response.status_code}", google_result, ""
-    except Exception as e:
-        logging.error(f"Błąd podczas pobierania URL {url}: {e}")
-        return f"Błąd podczas pobierania URL: {e}", google_result, ""
-
-def extract_text_from_image(image_file):
+    stats = load_json(STATS_FILE)
+    stats["total_analyses"] += 1
+    if fraud_detected:
+        stats["total_frauds_detected"] += 1
+    save_json(STATS_FILE, stats)
+    logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats['total_analyses']}, Oszustw {stats['total_frauds_detected']}.")
+
+def get_stats():
     """
-    Ekstraktuje tekst z obrazu za pomocą Tesseract OCR.
+    Pobiera statystyki analiz z pliku stats.json.
     """
-    try:
-        image = Image.open(image_file)
-        text = pytesseract.image_to_string(image, lang='pol')  # Upewnij się, że masz zainstalowane języki dla Tesseract
-        logging.info("Tekst został wyodrębniony z obrazu.")
-        return text
-    except Exception as e:
-        logging.error(f"Błąd podczas ekstrakcji tekstu z obrazu: {e}")
-        return f"Błąd podczas ekstrakcji tekstu z obrazu: {e}"
+    stats = load_json(STATS_FILE)
+    logging.info("Statystyki zostały pobrane pomyślnie.")
+    return stats