def get(self, id=None, name=None):
if id:
site = SiteModel.find_by_id(id)
else:
site = SiteModel.find_by_name(name)
if site:
return site.json()
return {'message': 'Site not found'}, 404
def delete(self, id=None, name=None):
if id:
site = SiteModel.find_by_id(id)
else:
site = SiteModel.find_by_name(name)
if site:
site.delete_from_db()
return {'message': 'Site deleted'}
def post(self):
data = CreateSite.parser2.parse_args()
current_user = current_identity.id
site = SiteModel(name=data['name'], admin=current_user)
if SiteModel.find_by_name(data['name']):
return {
'message':
"A site with name '{}' already exists.".format(data['name'])
}, 400
site.save_to_db()
return site.json(), 201
def post(self, name):
if SiteModel.find_by_name(name):
return {
'message': "A site with name '{}' already exists.".format(name)
}, 400
current_user = current_identity.id
site = SiteModel(name, admin=current_user)
try:
site.save_to_db()
except:
return {"message": "An error occurred creating the site."}, 500
return site.json(), 201
def put(self, sitename):
data = Site.parser.parse_args()
site = SiteModel.find_by_name(sitename)
if site:
site.street = data['street']
site.city = data['city']
site.state = data['state']
else:
site = SiteModel(sitename, data['street'], data['city'],
data['state'])
site.save_to_db()
return site.json()
def post(self, sitename):
data = Site.parser.parse_args()
if SiteModel.find_by_name(sitename):
return {
'message':
"A cell site with name '{}' already exists.".format(sitename)
}, 400
site = SiteModel(sitename, data['street'], data['city'], data['state'])
try:
site.save_to_db()
except:
return {
"message":
"An error occurred creating the {} site.".format(sitename)
}, 500
return site.json(), 201
def delete(self, sitename):
site = SiteModel.find_by_name(sitename)
if site:
site.delete_from_db()
return {'message': "Site {} deleted".format(sitename)}
def get(self, sitename):
site = SiteModel.find_by_name(sitename)
if site:
return site.json()
return {'message': "Site {} not found".format(sitename)}, 404
