1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from typing import Tuple
import shutil
from easyansi._core.codes import CSI as _CSI
from easyansi._core.prnt import prnt
from easyansi._core.validations import validate_at_least_minimum as _validate_at_least_minimum
from easyansi.cursor import locate_code as _locate_code
MINIMUM_ROW = 0
def clear_code() -> str:
"""Return the ANSI code to clear the screen and locate the cursor to 0, 0."""
return _CSI + "2J" + _locate_code(0, 0)
def clear() -> None:
"""Output the ANSI code to clear the screen and locate the cursor to 0, 0."""
prnt(clear_code())
def clear_line_code(row: int) -> str:
"""Return the ANSI code to clear the line at a given row. The cursor will end at the beginning of the row.
If row is not provided, it will be queried from the cursor location."""
_validate_at_least_minimum(number=row, minimum=MINIMUM_ROW, field="Row Number")
return _locate_code(0, row) + _CSI + "2K"
def clear_line(row: int) -> None:
"""Output the ANSI code to clear the line at a given row. The cursor will end at the beginning of the row.
If row is not provided, it will be queried from the cursor location."""
prnt(clear_line_code(row=row))
def get_size() -> Tuple[int, int]:
"""Return the screen size in cols, rows."""
screen_size = shutil.get_terminal_size()
return screen_size.columns, screen_size.lines
|
