Skip to content

Commit 9a01ed6

Browse files
committed
halonotebook no longer overwrite spinner on Windows
resolves #141
1 parent 89c7934 commit 9a01ed6

1 file changed

Lines changed: 24 additions & 2 deletions

File tree

halo/halo_notebook.py

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
import halo.cursor as cursor
77

88
from halo import Halo
9-
from halo._utils import colored_frame, decode_utf_8_text
10-
9+
from halo._utils import colored_frame, decode_utf_8_text, is_text_type
10+
from spinners.spinners import Spinners
1111

1212
class HaloNotebook(Halo):
1313
def __init__(
@@ -120,3 +120,25 @@ def stop_and_persist(self, symbol=" ", text=None):
120120

121121
with self.output:
122122
self.output.outputs = self._output(output)
123+
124+
def _get_spinner(self, spinner):
125+
"""Extracts spinner value from options and returns value
126+
containing spinner frames and interval, defaults to 'dots' spinner.
127+
Parameters
128+
----------
129+
spinner : dict, str
130+
Contains spinner value or type of spinner to be used
131+
Returns
132+
-------
133+
dict
134+
Contains frames and interval defining spinner
135+
"""
136+
default_spinner = Spinners['dots'].value
137+
138+
if spinner and type(spinner) == dict:
139+
return spinner
140+
141+
if all([is_text_type(spinner), spinner in Spinners.__members__]):
142+
return Spinners[spinner].value
143+
else:
144+
return default_spinner

0 commit comments

Comments
 (0)