From ee425b7d30a226a6514d89ccabd2f5b0d58cec43 Mon Sep 17 00:00:00 2001
From: MoMMde <mommde@pm.me>
Date: Mon, 28 Feb 2022 20:28:50 +0100
Subject: [PATCH] Added an option for checking a specific status code

---
 sherlock/sherlock.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/sherlock/sherlock.py b/sherlock/sherlock.py
index e29a56a4..0de5f649 100644
--- a/sherlock/sherlock.py
+++ b/sherlock/sherlock.py
@@ -331,6 +331,7 @@ def sherlock(username, site_data, query_notify,
 
         # Get the expected error type
         error_type = net_info["errorType"]
+        error_code = net_info.get("errorCode")
 
         # Retrieve future and ensure it has finished
         future = net_info["request_future"]
@@ -395,8 +396,15 @@ def sherlock(username, site_data, query_notify,
                                      QueryStatus.AVAILABLE,
                                      query_time=response_time)
         elif error_type == "status_code":
+            # Checks if the Status Code is equal to the optional "errorCode" given in 'data.json'
+            if error_code == r.status_code:
+                result = QueryResult(username,
+                                     social_network,
+                                     url,
+                                     QueryStatus.AVAILABLE,
+                                     query_time=response_time)
             # Checks if the status code of the response is 2XX
-            if not r.status_code >= 300 or r.status_code < 200:
+            elif not r.status_code >= 300 or r.status_code < 200:
                 result = QueryResult(username,
                                      social_network,
                                      url,