Coverage for osmclient/common/print_output.py: 23%

1####################################################################################### 

2# Copyright ETSI Contributors and Others. 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 

13# implied. 

14# See the License for the specific language governing permissions and 

15# limitations under the License. 

16####################################################################################### 

17import click 

18import logging 

19import json 

20import yaml 

21from prettytable import PrettyTable 

22 

23from jsonpath_ng import parse 

24 

25 

26logger = logging.getLogger("osmclient") 

27 

28 

29def evaluate_output_format(ctx, param, value): 

30 logger.debug("") 

31 if value in ["table", "json", "yaml", "csv"]: 

32 return value 

33 elif value.startswith("jsonpath="): 

34 return value 

35 else: 

36 raise click.BadParameter( 

37 f"{value} is not one of 'table', 'json', 'yaml', 'csv', 'jsonpath'" 

38 ) 

39 

40 

41output_option = click.option( 

42 "-o", 

43 "--output", 

44 default="table", 

45 is_eager=True, 

46 callback=evaluate_output_format, 

47 help="output format (default: table)", 

48) 

49 

50 

51literal_option = click.option( 

52 "--literal", is_flag=True, help="print literally, no pretty table" 

53) 

54 

55 

56def print_output(format, headers, rows): 

57 logger.debug("") 

58 if format == "table": 

59 table = PrettyTable(headers) 

60 table.align = "l" 

61 for row in rows: 

62 table.add_row(row) 

63 print(table) 

64 elif format == "csv": 

65 table = PrettyTable(headers) 

66 for row in rows: 

67 table.add_row(row) 

68 print(table.get_csv_string()) 

69 elif format == "json" or format == "yaml" or format.startswith("jsonpath="): 

70 data = [] 

71 for row in rows: 

72 item = {} 

73 for i in range(len(row)): 

74 item[headers[i]] = row[i] 

75 data.append(item) 

76 if format == "json": 

77 print(json.dumps(data, indent=4)) 

78 elif format == "yaml": 

79 print( 

80 yaml.safe_dump( 

81 data, indent=4, default_flow_style=False, sort_keys=False 

82 ) 

83 ) 

84 elif format.startswith("jsonpath="): 

85 # JSONPath expression 

86 json_path_expression = format.partition("=")[-1] 

87 logger.debug(f"Jsonpath expression: {json_path_expression}") 

88 json_path = parse(json_path_expression) 

89 # Apply JSONPath expression on the JSON object 

90 results = [match.value for match in json_path.find(data)] 

91 for i in results: 

92 print(i)