diff --git a/lib/telegram/bot/client.rb b/lib/telegram/bot/client.rb index 0e2bad6..41d7b65 100644 --- a/lib/telegram/bot/client.rb +++ b/lib/telegram/bot/client.rb @@ -47,7 +47,17 @@ module Telegram end def debug!(dev = STDOUT) - client.debug_dev = dev + if block_given? + begin + old_dev = client.debug_dev + client.debug_dev = dev + yield + ensure + client.debug_dev = old_dev + end + else + client.debug_dev = dev + end end def debug_off!