Rewrite of startups script in python - UNTESTED!!

This commit is contained in:
jpmeijers 2017-02-26 18:28:58 +01:00
parent 27ab7f6f34
commit e2789618cb

245
run.py Normal file
View File

@ -0,0 +1,245 @@
#!/usr/bin/python
"""
Author: JP Meijers
Date: 2017-02-26
Based on: https://github.com/rayozzie/ttn-resin-gateway-rpi/blob/master/run.sh
"""
import os.environ
import os.path
import sys
import urllib2
import time
import uuid
import subprocess
try:
import RPi.GPIO as GPIO
except RuntimeError:
print("Error importing RPi.GPIO! This is probably because you need superuser privileges. You can achieve this by using 'sudo' to run your script")
GWID_PREFIX="FFFE"
if not os.path.exists("mp_pkg_fwd"):
print ("ERROR: gateway executable not found. Is it built yet?")
sys.exit(0)
if os.environ.get('HALT') != None:
print ("*** HALT asserted - exiting ***")
sys.exit(0)
# Show info about the machine we're running on
print ("*** Resin Machine Info:")
print ("*** Type: "+os.environ.get('RESIN_MACHINE_NAME'))
print ("*** Arch: "+os.environ.get('RESIN_ARCH'))
if os.environ.get("RESIN_HOST_CONFIG_core_freq")!=None:
print ("*** Core freq: "+os.environ.get('RESIN_HOST_CONFIG_core_freq'))
if os.environ.get("RESIN_HOST_CONFIG_dtoverlay")!=None:
print ("*** UART mode: "+os.environ.get('RESIN_HOST_CONFIG_dtoverlay'))
# Check if the correct environment variables are set
print ("*******************")
print ("*** Configuration:")
print ("*******************")
if os.environ.get("GW_TYPE")==None:
print ("ERROR: GW_TYPE required")
print ("See https://github.com/rayozzie/ttn-resin-gateway-rpi/blob/master/README.md")
sys.exit(0)
if os.environ.get("GW_ID")==None:
print ("ERROR: GW_ID required")
print ("See https://www.thethingsnetwork.org/docs/gateways/registration.html#via-gateway-connector")
sys.exit(0)
if os.environ.get("GW_KEY")==None:
print ("ERROR: GW_KEY required")
print ("See https://www.thethingsnetwork.org/docs/gateways/registration.html#via-gateway-connector")
sys.exit(0)
my_eui = GWID_PREFIX + format(uuid.getnode(), '012x')
print ("Gateway Type:\t"+os.environ.get("GW_TYPE"))
print ("Gateway ID:\t"+os.environ.get("GW_ID"))
print ("Gateway EUI:\t"+my_eui)
print ("Gateway Key:\t"+os.environ.get("GW_KEY"))
print ("Has hardware GPS: "+os.getenv('GW_GPS', False))
print ("Hardware GPS port: "+os.getenv('GW_GPS_PORT', "/dev/ttyAMA0"))
print ("*******************")
print ("*** Fetching config from TTN account server")
print ("*******************")
# Define default configs
description = ""
placement = ""
latitude = os.getenv('GW_REF_LATITUDE', 0)
longitude = os.getenv('GW_REF_LONGITUDE', 0)
altitude = os.getenv('GW_REF_ALTITUDE', 0)
frequency_plan_url = "https://account.thethingsnetwork.org/api/v2/frequency-plans/EU_863_870"
# Fetch config from TTN if TTN is enabled
if(os.getenv('SERVER_TTN', True)):
# Fetch the URL, if it fails try 30 seconds later again.
config_response = ""
try:
req = urllib2.Request('https://account.thethingsnetwork.org/gateways/'+os.environ.get("GW_ID"))
req.add_header('Authorization', 'Key '+os.environ.get("GW_KEY"))
response = urllib2.urlopen(req, timeout=30)
config_response = response.read()
except urllib2.URLError as err:
print ("Unable to fetch configuration from TTN")
sys.exit(0)
# Parse config
ttn_config = {}
try:
ttn_config = json.loads(config_response)
except:
print ("Unable to parse configuration from TTN")
sys.exit(0)
frequency_plan = ttn_config.get('frequency_plan')#, "EU_863_870")
frequency_plan_url = ttn_config.get('frequency_plan_url')#, "https://account.thethingsnetwork.org/api/v2/frequency-plans/EU_863_870")
router = ttn_config.get('router')#, "router.dev.thethings.network:1883")
if "attributes" in ttn_config:
description = ttn_config['attributes'].get('description', "")
placement = ttn_config['attributes'].get('placement', "unknown")
if "location" in ttn_config:
latitude = ttn_config['location'].get('lat', 0)
longitude = ttn_config['location'].get('lng', 0)
altitude = ttn_config.get('altitude', 0)
print ("Router:\t"+router)
print ("Frequency plan:\t"+frequency_plan)
print ("Frequency plan url:\t"+frequency_plan_url)
print ("Gateway description:\t"+description)
print ("Gateway placement:\t"+placement)
print ("Latitude:\t"+latitude)
print ("Longitude:\t"+longitude)
# Done fetching config from TTN
# Retrieve global_conf
global_conf = ""
try:
response = urllib2.urlopen(frequency_plan_url, timeout=30)
global_conf = response.read()
except urllib2.URLError as err:
print ("Unable to fetch global conf from Github")
sys.exit(0)
# Write global_coonf
with open('global_conf.json', 'w') as the_file:
the_file.write(global_conf)
# Build local_conf
gateway_conf = {}
gateway_conf['gateway_ID'] = my_eui
gateway_conf['contact_email'] = os.getenv('GW_CONTACT_EMAIL', "")
gateway_conf['description'] = description
# Use hardware GPS
if(os.getenv('GW_GPS', True):
gateway_conf['gps'] = True
gateway_conf['fake_gps'] = False
gateway_conf['gps_tty_path'] = os.getenv('GW_GPS_PORT', "/dev/ttyAMA0")
# Use fake GPS with coordinates from TTN
if(os.getenv('GW_GPS', False) and latitude!=0 and longitude!=0))
gateway_conf['gps'] = True
gateway_conf['fake_gps'] = True
gateway_conf['ref_latitude'] = latitude
gateway_conf['ref_longitude'] = longitude
gateway_conf['ref_altitude'] = altitude
# No GPS coordinates
else:
gateway_conf['gps'] = False
gateway_conf['fake_gps'] = False
# Add server configuration
gateway_conf['servers'] = []
# Add TTN server
if(os.getenv('SERVER_TTN', True)):
server = {}
server['serv_type'] = "ttn"
server['server_address'] = router
server['serv_gw_id'] = os.environ.get("GW_ID")
server['serv_gw_key'] = os.environ.get("GW_KEY")
server['serv_enabled'] = True
gateway_conf['servers'].append(server)
# Add up to 3 additional servers
if(os.getenv('SERVER_1', True)):
server = {}
if(os.getenv('SERVER_1_TYPE', "semtech")=="ttn"):
server['serv_type'] = "ttn"
server['server_address'] = os.environ.get("SERVER_1_ADDRESS")
server['serv_gw_id'] = os.environ.get("SERVER_1_GWID")
server['serv_gw_key'] = os.environ.get("SERVER_1_GWKEY")
if(os.getenv('SERVER_1_ENABLED', "true")=="true"):
server['serv_enabled'] = True
else:
server['serv_enabled'] = False
gateway_conf['servers'].append(server)
if(os.getenv('SERVER_2', True)):
server = {}
if(os.getenv('SERVER_1_TYPE', "semtech")=="ttn"):
server['serv_type'] = "ttn"
server['server_address'] = os.environ.get("SERVER_1_ADDRESS")
server['serv_gw_id'] = os.environ.get("SERVER_1_GWID")
server['serv_gw_key'] = os.environ.get("SERVER_1_GWKEY")
if(os.getenv('SERVER_1_ENABLED', "true")=="true"):
server['serv_enabled'] = True
else:
server['serv_enabled'] = False
gateway_conf['servers'].append(server)
if(os.getenv('SERVER_3', True)):
server = {}
if(os.getenv('SERVER_1_TYPE', "semtech")=="ttn"):
server['serv_type'] = "ttn"
server['server_address'] = os.environ.get("SERVER_1_ADDRESS")
server['serv_gw_id'] = os.environ.get("SERVER_1_GWID")
server['serv_gw_key'] = os.environ.get("SERVER_1_GWKEY")
if(os.getenv('SERVER_1_ENABLED', "true")=="true"):
server['serv_enabled'] = True
else:
server['serv_enabled'] = False
gateway_conf['servers'].append(server)
# Now write the local_conf out to a file
# Write global_coonf
local_conf = {'gateway_conf': gateway_conf}
with open('local_conf.json', 'w') as the_file:
the_file.write(json.dumps(local_conf))
# Endless loop to reset and restart packet forwarder
while True:
# Reset the gateway board
GPIO.setmode(GPIO.BCM) # hardware pin numbers, just like gpio -1
if (os.environ.get("GW_TYPE")=="imst-ic880a-spi"):
print ("[TTN Gateway]: Toggling reset pin on IMST iC880A-SPI Board")
GPIO.setup(22, GPIO.OUT, initial=GPIO.LOW)
GPIO.output(22, 0)
time.sleep(0.1)
GPIO.output(22, 1)
time.sleep(0.1)
GPIO.output(22, 0)
time.sleep(0.1)
#TODO: reset other gateway boards
# Start forwarder
subprocess.call(["./mp_pkt_fwd"])
time.sleep(15)