Added support for context manager 'as' clause.

Signed-off-by: Andreas Maier <andras.r.maier@gmx.de>

Author: Andreas Maier

click_spinner/__init__.py

@@ -32,6 +32,7 @@ def init_spin(self):
 
     def __enter__(self):
         self.start()
+        return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.stop()

tests/test_spinner.py

@@ -78,3 +78,15 @@ def cli():
     result = runner.invoke(cli, [])
     assert result.exception is None
 
+
+def test_spinner_as():
+    @click.command()
+    def cli():
+       spinner = click_spinner.spinner()
+       with spinner as sp:
+           assert sp == spinner
+
+    runner = CliRunner()
+    result = runner.invoke(cli, [])
+    assert result.exception is None
+