Spaces:
Running
Running
Diego Carpintero
commited on
Commit
•
7210215
1
Parent(s):
254da27
return flag after URL check
Browse files
tools.py
CHANGED
@@ -41,7 +41,7 @@ class Tools:
|
|
41 |
except requests.exceptions.RequestException as e:
|
42 |
return url # Return original URL if expansion fails
|
43 |
|
44 |
-
def is_url_safe(self, target_url: str) -> Tuple[
|
45 |
"""Check if URL is safe using Google Safe Browsing API
|
46 |
"""
|
47 |
if not self.safebrowsing_key:
|
@@ -75,7 +75,7 @@ class Tools:
|
|
75 |
result = response.json()
|
76 |
|
77 |
if not result:
|
78 |
-
return
|
79 |
|
80 |
threats = []
|
81 |
if "matches" in result:
|
@@ -85,7 +85,7 @@ class Tools:
|
|
85 |
"threat_url": match.get("threat", {}).get("url"),
|
86 |
})
|
87 |
|
88 |
-
return
|
89 |
|
90 |
except requests.exceptions.RequestException as e:
|
91 |
raise Exception(f"Error checking URL safety: {str(e)}")
|
|
|
41 |
except requests.exceptions.RequestException as e:
|
42 |
return url # Return original URL if expansion fails
|
43 |
|
44 |
+
def is_url_safe(self, target_url: str) -> Tuple[str, List[Dict[str, str]]]:
|
45 |
"""Check if URL is safe using Google Safe Browsing API
|
46 |
"""
|
47 |
if not self.safebrowsing_key:
|
|
|
75 |
result = response.json()
|
76 |
|
77 |
if not result:
|
78 |
+
return "Not Flagged", []
|
79 |
|
80 |
threats = []
|
81 |
if "matches" in result:
|
|
|
85 |
"threat_url": match.get("threat", {}).get("url"),
|
86 |
})
|
87 |
|
88 |
+
return "Flagged", threats
|
89 |
|
90 |
except requests.exceptions.RequestException as e:
|
91 |
raise Exception(f"Error checking URL safety: {str(e)}")
|