diff --git a/update_checker.py b/update_checker.py index 05ca69880dce78cb7176b4f0e4607127e8d2f176..e3427742baeaa8f9b8ccfd08f372680b68d6b93d 100755 --- a/update_checker.py +++ b/update_checker.py @@ -7,8 +7,13 @@ import os import sys import pprint import requests +import argparse MAC_URL = 'http://macvendors.co/api/%s' +ap = argparse.ArgumentParser() +ap.add_argument("--raw", type=argparse.FileType("r", encoding="utf-8"), required=True) +args = ap.parse_args() + def getHardwareModelFromEntry(d): try: hardware_model = d["nodeinfo"]["hardware"]["model"] @@ -90,10 +95,7 @@ for f in allNewFirmwareDownloads: json.dump(allFirmwareDownloads,open("firmwareDownloads.json","w"),sort_keys=True, indent=4, separators=(',', ': ')) -nodesdbFilename = "/home/www/html/netinfo/json/nodesdb.json" -if not os.path.isfile(nodesdbFilename): - nodesdbFilename = "nodesdb.json" -data = json.load(open(nodesdbFilename, encoding='utf-8')) +data = json.load(args.raw) for download in allFirmwareDownloads: try: