Fix update.py script

This commit is contained in:
Lucas Bickel 2017-03-10 15:53:22 +01:00
parent a7d25e1960
commit 6e50af344a
1 changed files with 10 additions and 7 deletions

View File

@ -28,27 +28,30 @@ if __name__ == '__main__':
api_key = config.get(GENERAL_CONFIG_SECTION, 'api_key') api_key = config.get(GENERAL_CONFIG_SECTION, 'api_key')
base_url = config.get(GENERAL_CONFIG_SECTION, 'base_url') base_url = config.get(GENERAL_CONFIG_SECTION, 'base_url')
base_dir = config.get(GENERAL_CONFIG_SECTION, 'base_dir') base_dir = config.get(GENERAL_CONFIG_SECTION, 'base_dir')
base_port = config.get(GENERAL_CONFIG_SECTION, 'base_port', 80)
action = "upgrade" action = "upgrade"
airtime_url = "" station_url = ""
default_url = "http://%s:%s%s" % (base_url, base_port, base_dir)
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--downgrade', help='Downgrade the station', action="store_true") parser.add_argument('--downgrade', help='Downgrade the station', action="store_true")
parser.add_argument('station_url', help='station URL', nargs='?', default='') parser.add_argument('station_url', help='station URL', nargs='?', default=default_url)
args = parser.parse_args() args = parser.parse_args()
if args.downgrade: if args.downgrade:
action = "downgrade" action = "downgrade"
if airtime_url == "": if args.station_url:
airtime_url = "http://%s%s" % (base_url, base_dir) station_url = args.station_url
# Add http:// if you were lazy and didn't pass a scheme to this script # Add http:// if you were lazy and didn't pass a scheme to this script
url = urlparse(airtime_url) url = urlparse(station_url)
if not url.scheme: if not url.scheme:
airtime_url = "http://%s" % airtime_url station_url = "http://%s" % station_url
print "Requesting %s..." % action print "Requesting %s..." % action
r = requests.get("%s/%s" % (airtime_url, action), auth=(api_key, '')) r = requests.get("%s/%s" % (station_url, action), auth=(api_key, ''))
print r.text print r.text
r.raise_for_status() r.raise_for_status()