Improved error handling of the default country_code value

InfamousSYN · Dec 18, 2023 · 04b9b1a · 04b9b1a
1 parent 1cddb9d
commit 04b9b1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config.py b/config.py
@@ -3,7 +3,7 @@
 import argparse
 
 # application version
-__version__ = "3.0.4"
+__version__ = "3.0.5"
 
 # application site
 __location__ = "https://rogue.infamoussyn.com/"

diff --git a/core/libs/options.py b/core/libs/options.py
@@ -38,7 +38,7 @@ def set_driver(self):
 
     @classmethod
     def set_country(self):
-        self.country_code = "country_code={}".format(self.country_code) if self.country_code is not None else "#country_code=00"
+        self.country_code = "country_code={}".format(self.country_code) if (self.country_code is not None and self.country_code != '00') else "#country_code=00"
 
     @classmethod
     def check_80211d(self):
@@ -508,7 +508,7 @@ def set_options():
     ieee80211_config.add_argument('--country',
                     dest='country_code',
                     type=str,
-                    default=00,
+                    default='00',
                     choices=config.rogue_country_options,
                     help='Configures of country of operation')