Coverage for thumbhash / cli.py: 67%

1import io (empty)

2from base64 import b64encode (empty)

3from pathlib import Path (empty)

4from typing import Annotated, Any (empty)

5 

6import typer (empty)

7from rich import print (empty)

8from rich.console import Console (empty)

9 

10from thumbhash import image_to_thumbhash, thumbhash_to_image (empty)

11 

12app = typer.Typer() (empty)

13 

14 

15@app.command() (empty)

16def encode( (empty)

17 image_path: Annotated[ 

18 Path, 

19 typer.Argument( 

20 help="The path of the image to convert", 

21 exists=True, 

22 file_okay=True, 

23 dir_okay=False, 

24 readable=True, 

25 resolve_path=True, 

26 ), 

27 ], 

28) -> None: 

29 """ 

30 Get thumbhash from image 

31 """ 

32 hash = image_to_thumbhash(image_path) (empty)

33 

34 print(f"Thumbhash (base64): [green]{hash}[/green]") (empty)

35 

36 

37@app.command() (empty)

38def decode( (empty)

39 image_path: Annotated[ 

40 Path, 

41 typer.Argument( 

42 help="The path where the image created from the hash will be saved, '-' for stdout (png base64)", 

43 file_okay=True, 

44 dir_okay=False, 

45 resolve_path=True, 

46 allow_dash=True, 

47 ), 

48 ], 

49 hash: Annotated[str, typer.Argument(help="The base64-encoded thumbhash")], 

50 size: Annotated[ 

51 int, 

52 typer.Option("--size", "-s", help="The base size of the output image", min=1), 

53 ] = 32, 

54 saturation: Annotated[ 

55 float, typer.Option(help="The saturation boost factor to use") 

56 ] = 1.25, 

57) -> None: 

58 """ 

59 Save thumbnail image from thumbhash 

60 """ 

61 image = thumbhash_to_image(hash, size, saturation) 

62 

63 if str(image_path) == "-": 

64 b = io.BytesIO() 

65 console = Console(markup=False, no_color=True, highlight=False) 

66 

67 image.save(b, "png") 

68 console.print(b64encode(b.getvalue()).decode("utf-8")) 

69 else: 

70 image.save(image_path) 

71 

72 

73def main() -> Any: (empty)

74 return app()