Add functions for CJK string alignment

archinstall/lib/output.py

@@ -1,6 +1,7 @@
 import logging
 import os
 import sys
+import unicodedata
 from enum import Enum
 
 from pathlib import Path
@@ -326,3 +327,33 @@ def log(
 		if level != logging.DEBUG or storage.get('arguments', {}).get('verbose', False):
 			sys.stdout.write(f"{text}\n")
 			sys.stdout.flush()
+
+def _count_cjk_chars(string: str) -> int:
+	"Count the total number of CJK characters contained in a string"
+	return sum(unicodedata.east_asian_width(c) in 'FW' for c in string)
+
+def cjk_ljust(string: str, width: int, fillbyte: str = ' ') -> str:
+	"""Return a left-justified CJK string of length width.
+	>>> cjkljust('Hello', 15, '*')
+	'Hello**********'
+	>>> cjkljust('你好', 15, '*')
+	'你好***********'
+	>>> cjkljust('안녕하세요', 15, '*')
+	'안녕하세요*****'
+	>>> cjkljust('こんにちは', 15, '*')
+	'こんにちは*****'
+	"""
+	return string.ljust(width - _count_cjk_chars(string), fillbyte)
+
+def cjk_rjust(string: str, width: int, fillbyte: str = ' ') -> str:
+	"""Return a right-justified CJK string of length width.
+	>>> cjkrjust('Hello', 15, '*')
+	'**********Hello'
+	>>> cjkrjust('你好', 15, '*')
+	'***********你好'
+	>>> cjkrjust('안녕하세요', 15, '*')
+	'*****안녕하세요'
+	>>> cjkrjust('こんにちは', 15, '*')
+	'*****こんにちは'
+	"""
+	return string.rjust(width - _count_cjk_chars(string), fillbyte)