85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
import logging
|
|
import os
|
|
import textwrap
|
|
from enum import Enum
|
|
from types import ModuleType
|
|
|
|
import language
|
|
|
|
import typer
|
|
from rich.logging import RichHandler
|
|
from rich.console import Console
|
|
|
|
app = typer.Typer()
|
|
|
|
app_state = {}
|
|
|
|
default_language = os.environ.get("FANLANG_DEFAULT_LANGUAGE", "common")
|
|
language_pack, supported_languages = language.load_language_pack()
|
|
|
|
Supported = Enum("Supported", ((k, k) for k in supported_languages.keys()))
|
|
|
|
|
|
def print_sample(lang: str, module: ModuleType) -> None:
|
|
summary = module.__doc__.replace("\n", " \n")
|
|
print(f"{summary}")
|
|
print(f" Text: {module.Language.text(10)}")
|
|
print(f" Names: {module.Name}")
|
|
print(f" {module.Name}")
|
|
print(f" {module.Name}")
|
|
if module.NobleName != module.Name:
|
|
print(f" Noble Names: {module.NobleName}")
|
|
print(f" {module.NobleName}")
|
|
print(f" {module.NobleName}")
|
|
print("")
|
|
|
|
|
|
@app.callback()
|
|
def main(
|
|
language: Supported = typer.Option(
|
|
default=default_language,
|
|
help="The language to use."
|
|
),
|
|
):
|
|
app_state["language"] = supported_languages[language.name]
|
|
|
|
debug = os.getenv("FANLANG_DEBUG", None)
|
|
logging.basicConfig(
|
|
format="%(name)s %(message)s",
|
|
level=logging.DEBUG if debug else logging.INFO,
|
|
handlers=[RichHandler(rich_tracebacks=True, tracebacks_suppress=[typer])],
|
|
)
|
|
logging.getLogger('markdown_it').setLevel(logging.ERROR)
|
|
logging.debug(f"Loaded language pack {language_pack}.")
|
|
logging.debug(f"Default language: {default_language}.")
|
|
|
|
|
|
@app.command()
|
|
def text(count: int = typer.Option(50, help="The number of words to generate.")):
|
|
|
|
paragraph = app_state["language"].Language.text(count)
|
|
|
|
console = Console(width=80)
|
|
console.print(paragraph)
|
|
|
|
|
|
@app.command()
|
|
def names(
|
|
count: int = typer.Option(50, help="The number of names to generate."),
|
|
noble: bool = typer.Option(False, help="Generate noble names."),
|
|
):
|
|
generator = app_state["language"].Name if not noble else app_state["language"].NobleName
|
|
for name in generator.name(count):
|
|
print(name["fullname"])
|
|
|
|
|
|
@app.command()
|
|
def list():
|
|
for lang, module in supported_languages.items():
|
|
text = textwrap.shorten(module.Language.text(count=20), width=70, placeholder='')
|
|
print(f"{lang.title():15s}: {text}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|