Added --local argument for local data.json use

This commit is contained in:
pyghost 2020-08-21 05:07:30 +09:30 committed by GitHub
parent 57df1d904b
commit 6a723233a7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -490,6 +490,10 @@ def main():
parser.add_argument("--browse", "-b",
action="store_true", dest="browse", default=False,
help="Browse to all results on default browser.")
parser.add_argument("--local", "-l",
action='store_true', default=None,
help="Forces the use of the local data.json file.")
args = parser.parse_args()
@ -535,7 +539,10 @@ def main():
#Create object with all information about sites we are aware of.
try:
sites = SitesInformation(args.json_file)
if args.local:
sites = SitesInformation(os.path.join(os.path.dirname(__file__), 'resources/data.json'))
else:
sites = SitesInformation(args.json_file)
except Exception as error:
print(f"ERROR: {error}")
sys.exit(1)