]>
Commit | Line | Data |
---|---|---|
53e6db90 DC |
1 | """ |
2 | Formatting numeric literals. | |
3 | """ | |
4 | from blib2to3.pytree import Leaf | |
5 | ||
6 | ||
7 | def format_hex(text: str) -> str: | |
8 | """ | |
9 | Formats a hexadecimal string like "0x12B3" | |
10 | """ | |
11 | before, after = text[:2], text[2:] | |
12 | return f"{before}{after.upper()}" | |
13 | ||
14 | ||
15 | def format_scientific_notation(text: str) -> str: | |
16 | """Formats a numeric string utilizing scentific notation""" | |
17 | before, after = text.split("e") | |
18 | sign = "" | |
19 | if after.startswith("-"): | |
20 | after = after[1:] | |
21 | sign = "-" | |
22 | elif after.startswith("+"): | |
23 | after = after[1:] | |
24 | before = format_float_or_int_string(before) | |
25 | return f"{before}e{sign}{after}" | |
26 | ||
27 | ||
28 | def format_complex_number(text: str) -> str: | |
29 | """Formats a complex string like `10j`""" | |
30 | number = text[:-1] | |
31 | suffix = text[-1] | |
32 | return f"{format_float_or_int_string(number)}{suffix}" | |
33 | ||
34 | ||
35 | def format_float_or_int_string(text: str) -> str: | |
36 | """Formats a float string like "1.0".""" | |
37 | if "." not in text: | |
38 | return text | |
39 | ||
40 | before, after = text.split(".") | |
41 | return f"{before or 0}.{after or 0}" | |
42 | ||
43 | ||
44 | def normalize_numeric_literal(leaf: Leaf) -> None: | |
45 | """Normalizes numeric (float, int, and complex) literals. | |
46 | ||
47 | All letters used in the representation are normalized to lowercase.""" | |
48 | text = leaf.value.lower() | |
49 | if text.startswith(("0o", "0b")): | |
50 | # Leave octal and binary literals alone. | |
51 | pass | |
52 | elif text.startswith("0x"): | |
53 | text = format_hex(text) | |
54 | elif "e" in text: | |
55 | text = format_scientific_notation(text) | |
56 | elif text.endswith("j"): | |
57 | text = format_complex_number(text) | |
58 | else: | |
59 | text = format_float_or_int_string(text) | |
60 | leaf.value = text |