src/black/numerics.py

   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