diff options
Diffstat (limited to 'application.rb')
-rw-r--r-- | application.rb | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/application.rb b/application.rb index eb7db12..b522baf 100644 --- a/application.rb +++ b/application.rb @@ -8,7 +8,16 @@ class ModelStore property :id, Serial property :uri, String, :length => 255 property :yaml, Text, :length => 2**32-1 + property :token_id, String, :length => 255 property :created_at, DateTime + + after :save, :check_policy + + private + def check_policy + OpenTox::Authorization.check_policy(uri, token_id) + end + end class PredictionCache @@ -47,8 +56,17 @@ end delete '/:id/?' do begin + uri = ModelStore.get(params[:id]).uri ModelStore.get(params[:id]).destroy! "Model #{params[:id]} deleted." + if params[:token_id] and !Model.get(params[:id]) and uri + begin + aa = OpenTox::Authorization.delete_policy_from_uri(uri, params[:token_id]) + LOGGER.debug "Policy deleted for Model URI: #{uri} with token_id: #{params[:token_id]} with result: #{aa}" + rescue + LOGGER.warn "Policy delete error for Model URI: #{uri}" + end + end rescue halt 404, "Model #{params[:id]} does not exist." end |