update and delete location

api/endpoints/location.py

 from flask import request, jsonify
 from flask_restful import Resource
-from flask_jwt_extended import jwt_required
+from flask_jwt_extended import jwt_required, get_jwt_identity
+from api.models.dbmodels import User
+
 from api import db
 from api import limiter
-from api.models.dbmodels import Location
+from api.models.dbmodels import Location, Salamander
 import re
 import os
 from api.endpoints.matchsalamander import sanitize_int_str
@@ -42,6 +44,55 @@ class LocationEndpoint(Resource):
             location_list.append(loc)
         return jsonify({"locations": location_list, "status": 200})
 
+    @staticmethod
+    @jwt_required
+    def delete():
+        user_id = get_jwt_identity()
+        user = db.session.query(User).filter_by(id=user_id).first()
+        data = request.form
+        if user.admin:
+            if "id" in data:
+                salamanders = db.session.query(Salamander).filter_by(location_id=data['id'])
+                if salamanders and len(salamanders) == 0:
+                    location = db.session.query(Location).filter_by(id=data['id']).first()
+                    if location:
+                        db.session.delete(location)
+                        db.session.commit()
+                        return jsonify({"location": str(location.id), "deleted": 200})
+                    else:
+                        return jsonify({"message": "location doesn't exist", "status": 400})
+                else:
+                    return jsonify({"message": "location needs to be empty", "status": 400})
+            else:
+                return jsonify({"message": "wrong data", "status": 400})
+        else:
+            return jsonify({"message": "user not admin", "status": 400})
+
+    @staticmethod
+    @jwt_required
+    def put():
+        user_id = get_jwt_identity()
+        user = db.session.query(User).filter_by(id=user_id).first()
+        data = request.form
+        if user.admin:
+            if "id" in data:
+                location = db.session.query(Location).filter_by(id=data['id']).first()
+                if location:
+                    if "new_name" in data:
+                        location.name = data['new_radius'].lower()
+                    if "new_radius" in data:
+                        location.radius = sanitize_int_str(str(data['new_radius']))
+                    if "new_latitude" in data:
+                        location.latitude = data['new_latitude']
+                    if "new_longitude" in data:
+                        location.longitude = data['new_longitude']
+                    db.session.commit()
+                    return jsonify({"location": str(location.id), " updated.": 200})
+                else:
+                    return jsonify({"message": "location doesn't exist", "status": 400})
+            else:
+                return jsonify({"message": "wrong data", "status": 400})
+        else: return jsonify({"message": "user not admin", "status": 400})
 
 def create_directory_folders(location):
     dir_smooth_male = "images/" + location + "/smooth_newt/" + "male/"