complexipy

import os from complexipy import file_complexity, code_complexity # Example 1: Analyze a Python string snippet = """ def calculate_discount(price, quantity, is_member): if is_member: if quantity > 10: return price * quantity * 0.8 # 20% discount else: return price * quantity * 0.9 # 10% discount else: return price * quantity """ print("\n--- Analyzing code snippet ---") result_code = code_complexity(snippet) print(f"Overall snippet complexity: {result_code.complexity}") for func in result_code.functions: print(f" Function '{func.name}': Complexity {func.complexity}, Line {func.line_start}-{func.line_end}") # Example 2: Analyze a temporary Python file file_content = """ def process_data(data): if data.get('status') == 'active': for item in data.get('items', []): if item.get('value', 0) > 100: print(f"Processing high-value item: {item['id']}") else: print(f"Processing regular item: {item['id']}") elif data.get('status') == 'pending': print("Data pending, skipping processing.") else: print("Unknown data status.") """ file_path = "example_module.py" with open(file_path, "w") as f: f.write(file_content) print(f"\n--- Analyzing file: {file_path} ---") result_file = file_complexity(file_path) print(f"File complexity: {result_file.complexity}") for func in result_file.functions: print(f" Function '{func.name}': Complexity {func.complexity}, Line {func.line_start}-{func.line_end}") # Clean up temporary file os.remove(file_path)