From ccb84c744a483943d72205913f4a306187d40f64 Mon Sep 17 00:00:00 2001 From: Max Melentiev Date: Fri, 26 Feb 2016 16:27:43 +0600 Subject: [PATCH] Client#debug! supports block --- lib/telegram/bot/client.rb | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) 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!