Merge pull request #25 from sundowndev/hotfix/countryScan

[Fix #24] Country code scan

phoneinfoga.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-__version__ = 'v1.1.2-rc1'
+__version__ = 'v1.0.0-rc3'
 
 try:
     import sys
@@ -233,7 +233,7 @@ def search(req, stop):
 
         return links
     except:
-        print(code_error + 'Request failed. Please retry or open an issue on GitHub.')
+        print(code_error + 'Request failed. Please retry or open an issue on https://github.com/sundowndev/PhoneInfoga.')
 
 
 def formatNumber(InputNumber):
@@ -264,9 +264,17 @@ def localScan(InputNumber):
         numberCountryCode = phonenumbers.format_number(
             PhoneNumberObject, phonenumbers.PhoneNumberFormat.INTERNATIONAL).split(' ')[0]
 
-        countryRequest = json.loads(requests.request(
-            'GET', 'https://restcountries.eu/rest/v2/callingcode/{}'.format(numberCountryCode.replace('+', ''))).content)
-        numberCountry = countryRequest[0]['alpha2Code']
+        try:
+            countries = json.load(open('data/CountryCodes.json'))
+
+            for country in countries:
+                if (country['dial_code'].replace(' ', '') == numberCountryCode):
+                    print(code_info + 'Country code found: {} ({})'.format(country['name'],country['code']))
+                    numberCountry = country['code']
+                    break
+        except:
+            print(code_error + 'Unable to find country code.')
+            print(numberCountry)
 
         localNumber = phonenumbers.format_number(
             PhoneNumberObject, phonenumbers.PhoneNumberFormat.E164).replace(numberCountryCode, '')
@@ -371,10 +379,13 @@ def ovhScan():
         'cache-control': "no-cache"
     }
 
+    try:
         response = requests.request(
             "GET", "https://api.ovh.com/1.0/telephony/number/detailedZones", data="", headers=headers, params=querystring)
-
         data = json.loads(response.content)
+    except:
+        print(code_error + 'OVH API is unreachable. Maybe retry later.')
+        return -1
 
     if isinstance(data, list):
         askedNumber = "0" + localNumber.replace(localNumber[-4:], 'xxxx')
@@ -490,7 +501,6 @@ def osintScan():
     global localNumber
     global internationalNumber
     global numberCountryCode
-    global numberCountry
     global customFormatting
 
     if not args.osint:
@@ -576,7 +586,8 @@ def osintScan():
 
     osintIndividualScan()
 
-    retry_input = input(code_info + "Would you like to rerun OSINT scan ? (e.g to use a different format) (y/N) ")
+    retry_input = input(
+        code_info + "Would you like to rerun OSINT scan ? (e.g to use a different format) (y/N) ")
 
     if retry_input.lower() == 'y' or retry_input.lower() == 'yes':
         osintScan()
@@ -622,6 +633,7 @@ def scanNumber(InputNumber):
     else:
         print('\n')
 
+
 try:
     if args.no_ansi or args.output:
         code_info = '[*] '