使用Python3实现判断函数的圈复杂度

pip3 install radon

import random
 
def guess_number():
    # 生成一个随机数作为答案
    answer = random.randint(1, 100)
 
    # 初始化猜测次数
    guesses = 0
 
    print("欢迎来到猜数字游戏！我已经想好了一个1到100之间的数字，你需要猜出这个数字是多少。")
 
    # 开始循环，直到玩家猜中数字为止
    while True:
        # 获取玩家的猜测
        guess = int(input("请输入你猜测的数字："))
 
        # 增加猜测次数
        guesses += 1
 
        # 检查玩家猜测的数字与答案的关系
        if guess < answer:
            print("你猜的数字太小了，请继续努力！")
        elif guess > answer:
            print("你猜的数字太大了，请再试一次！")
        else:
            print(f"恭喜你，你猜对了！答案是 {answer}。你一共猜了 {guesses} 次。")
            break  # 结束循环
 
 
# 调用函数开始游戏
guess_number()

from radon.complexity import cc_visit
 
# 定义一个Python文件路径
file_path = 'complex_func.py'
 
# 使用cc_visit函数计算代码的圈复杂度
with open(file_path, 'r') as file:
    code = file.read()
    results = cc_visit(code)
    print(results)
 
# 打印结果
for result in results:
    print(result)

import radon
from radon.complexity import cc_rank, cc_visit
 
 
def calculate_complexity(source_code):
    """
    Calculate the cyclomatic complexity of the given source code.
    Parameters:
    source_code (str): The source code to analyze.
    Returns:
    int: The cyclomatic complexity.
    str: The complexity rating.
    """
    try:
        # Visit the AST and calculate the complexity
        results = cc_visit(source_code)
        complexity = results[0].complexity
        # Get the complexity rating
        rating = cc_rank(complexity)
        return complexity, rating
    except Exception as e:
        print("Error:", e)
        return None, None
 
 
# Example usage:
if __name__ == "__main__":
    code = """
def rank(self):
    rating_num = float(self.rating)
    if rating_num >= 8.5:
        return 'S'
    elif rating_num >= 8:
        return 'A'
    elif rating_num >= 7:
        return 'B'
    elif rating_num >= 6:
        return 'C'
    else:
        return 'D'
    """
    complexity, rating = calculate_complexity(code)
    if complexity is not None and rating is not None:
        print("Cyclomatic Complexity:", complexity)
        print("Complexity Rating:", rating)

def rank(self):
    breakpoints = (6, 7, 8, 8.5)
    grades = ('D', 'C', 'B', 'A', 'S')
    index = bisect.bisect(breakpoints, float(self.rating))
    return grades[index]