X-Git-Url: http://gitweb.michael.orlitzky.com/?p=mailshears.git;a=blobdiff_plain;f=lib%2Frm%2Fplugins%2Fdavical.rb;h=77b06b4777714a50a7ea18aa15e2116389128b2b;hp=02e889466849fb7c67aad15e0812afce1e23e33e;hb=20b843bddcd73833d41f98ff79d92ef59bb4d81e;hpb=483d14dc8228a81d12fb109d3ed6510e2d2b03c1 diff --git a/lib/rm/plugins/davical.rb b/lib/rm/plugins/davical.rb index 02e8894..77b06b4 100644 --- a/lib/rm/plugins/davical.rb +++ b/lib/rm/plugins/davical.rb @@ -12,32 +12,22 @@ class DavicalRm include RmPlugin - def delete_account(account) + def delete_user(user) # Delete the given username. DAViCal uses foreign keys properly # and only supports postgres, so we let the ON DELETE CASCADE # trigger handle most of the work. - sql_queries = ['DELETE FROM usr WHERE username = $1'] - - begin - connection = PGconn.connect(@db_host, - @db_port, - @db_opts, - @db_tty, - @db_name, - @db_user, - @db_pass) + raise NonexistentUserError.new(user.to_s()) if not user_exists(user) - sql_queries.each do |sql_query| - connection.query(sql_query, [account]) - end + sql_queries = ['DELETE FROM usr WHERE username = $1'] - connection.close() + connection = PGconn.connect(@db_host, @db_port, @db_opts, @db_tty, + @db_name, @db_user, @db_pass) - rescue PGError => e - # Pretend like we're database-agnostic in case we ever are. - raise DatabaseError.new(e) + sql_queries.each do |sql_query| + connection.query(sql_query, [user.to_s()]) end + connection.close() end