def register_proposal(caller_identity):
payload = request.get_json(force=True)
proposal = payload.get('service_proposal', None)
if proposal is None:
return jsonify(error='missing service_proposal'), 400
node_key = proposal.get('provider_id', None)
if node_key is None:
return jsonify(error='missing provider_id'), 400
if node_key.lower() != caller_identity:
message = 'provider_id does not match current identity'
return jsonify(error=message), 403
node = Node.query.get(node_key)
if not node:
node = Node(node_key)
node.ip = request.remote_addr
node.proposal = json.dumps(proposal)
node.mark_activity()
db.session.add(node)
db.session.commit()
return jsonify({})
def node_register():
payload = request.get_json(force=True)
node_key = payload.get('node_key', '')
connection_config = payload.get('connection_config', '')
if node_key == '':
return jsonify(error='node key is empty'), 400
rp = RegisteredParticipant.query.get(node_key)
if rp:
if rp.node_ip != request.remote_addr:
return jsonify(error='node key and ip address mismatch'), 400
node = Node.query.get(node_key)
if not node:
node = Node(node_key)
node.ip = request.remote_addr
node.connection_config = connection_config
node.updated_at = datetime.utcnow()
db.session.add(node)
db.session.commit()
return jsonify({})
def addEditNode(node_id):
node = None
# get choices for node leaders
leader_choices = [(0, 'Self')]
for x in Node.query.filter_by(leader_id=0): # @UndefinedVariable
leader_choices.append((x.id,x.name))
form = NodeForm()
form.leader.choices = leader_choices
form.leader.default = 0
if node_id is not None:
node = Node.query.get(node_id) # @UndefinedVariable
if request.method == 'GET':
if node is None:
form.new.data = True
else:
form.new.data = False
form.id.data = node.id
form.name.data = node.name
form.rid.data = node.rid
form.ip.data = node.ip
form.location.lat.data = node.location.lat
form.location.lon.data = node.location.lon
form.leader.data = node.leader_id
jumppoints = []
for jp in node.jumppoints:
form.jumppoints.append_entry({"jp_id": jp.id, "lat": jp.location.lat, "lon":jp.location.lon })
goals = []
for goal in node.goals:
form.goals.append_entry({"goal_id": goal.id, "lat": goal.location.lat, "lon":goal.location.lon })
elif request.method == 'POST' and form.validate(): # @UndefinedVariable
if node is None:
#new node has passed validation, add to db
location = Location(lat=form.location.lat.data, lon=form.location.lon.data)
db.session.add(location) # @UndefinedVariable
node = Node(name=form.name.data, leader_id=form.leader.data, location=location, rid=form.rid.data, ip=form.ip.data)
db.session.add(node) # @UndefinedVariable
db.session.commit() # @UndefinedVariable
for index, point in enumerate(form.jumppoints.data):
jp = JumpPoint()
location = Location(lat=point['lat'], lon=point['lon'])
db.session.add(location) # @UndefinedVariable
jp.location = location
jp.position = int(point['pos']) + 1
db.session.add(jp)
node.jumppoints.append(jp)
for index, point in enumerate(form.goals.data):
goal = Goal()
location = Location(lat=point['lat'], lon=point['lon'])
db.session.add(location) # @UndefinedVariable
goal.location = location
goal.position = int(point['pos']) + 1
db.session.add(goal)
node.goals.append(goal)
db.session.commit() # @UndefinedVariable
flash("Node has been created")
else:
#node has been updated. save updates
node.name = form.name.data
node.rid = form.rid.data
node.ip = form.ip.data
location = Location.query.get(node.loc_id) # @UndefinedVariable
location.lat = form.location.lat.data
location.lon = form.location.lon.data
node.location = location
node.leader_id = form.leader.data
# create a list of all points already included on this path. will be used to determine if
# any points were deleted from the list.
deleteList = []
for jp in node.jumppoints:
deleteList.append(jp.id)
for index, jp in enumerate(form.jumppoints.data):
if int(jp['jp_id']) == 0:
newjp = JumpPoint()
location = Location(lat=jp['lat'], lon=jp['lon'])
db.session.add(location) # @UndefinedVariable
newjp.location = location
newjp.position = int(jp['pos']) + 1
db.session.add(newjp)
node.jumppoints.append(newjp)
else:
# found existing point. update and remove from delete list
savedjp = JumpPoint.query.get(jp['jp_id']) # @UndefinedVariable
savedjp.position = int(jp['pos']) + 1
savedLoc = Location.query.get(savedjp.loc_id) # @UndefinedVariable
savedLoc.lat = jp['lat']
savedLoc.lon = jp['lon']
deleteList.remove(int(jp['jp_id']))
for id in deleteList:
jp= JumpPoint.query.get(id) # @UndefinedVariable
db.session.delete(jp)
deleteList = []
for goal in node.goals:
deleteList.append(goal.id)
for index, goal in enumerate(form.goals.data):
if int(goal['goal_id']) == 0:
newgoal = Goal()
location = Location(lat=goal['lat'], lon=goal['lon'])
db.session.add(location) # @UndefinedVariable
newgoal.location = location
newgoal.position = int(goal['pos']) + 1
db.session.add(newgoal)
node.goals.append(newgoal)
else:
# found existing point. update and remove from delete list
savedGoal = Goal.query.get(goal['goal_id']) # @UndefinedVariable
savedGoal.position = int(goal['pos']) + 1
savedLoc = Location.query.get(savedGoal.loc_id) # @UndefinedVariable
savedLoc.lat = goal['lat']
savedLoc.lon = goal['lon']
deleteList.remove(int(goal['goal_id']))
for id in deleteList:
goal= Goal.query.get(id) # @UndefinedVariable
db.session.delete(goal)
db.session.commit()
flash("Node has been updated")
# after creating the new state, redirect them back to dce config page
return redirect(url_for("nodePage"))
return render_template("nodeForm.html", form=form)
