使用python写一个随机生产python代码的程序

原创

爱写代码的阿木 2024-11-18 14:02:17 ©著作权

文章标签 Python 类定义函数定义 文章分类 Python 后端开发 私藏项目实操分享

©著作权归作者所有：来自51CTO博客作者爱写代码的阿木的原创作品，请联系作者获取转载授权，否则将追究法律责任

下面是一个简单的Python程序，它会生成随机的Python代码片段。这个程序会随机生成一些常见的Python语句，如变量赋值、简单的算术运算、条件语句和循环语句。

python复制代码
 import random
 
 import string
 
  
 
 def generate_variable_name():
 
     # 生成一个随机的变量名
 
     return ''.join(random.choices(string.ascii_letters + string.digits, k=5))
 
  
 
 def generate_number():
 
     # 生成一个随机的数字
 
     return str(random.randint(0, 100))
 
  
 
 def generate_statement():
 
     # 生成一个随机的Python语句
 
     statements = [
 
         f"{generate_variable_name()} = {generate_number()}",
 
         f"{generate_variable_name()} = {generate_variable_name()} + {generate_number()}",
 
         f"{generate_variable_name()} = {generate_variable_name()} * {generate_number()}",
 
         f"print({generate_variable_name()})",
 
         f"if {generate_variable_name()} > {generate_number()}: print('true')",
 
         f"for i in range({generate_number()}): print(i)"
 
     ]
 
     return random.choice(statements)
 
  
 
 def generate_code(num_statements):
 
     # 生成包含指定数量语句的Python代码
 
     code = []
 
     for _ in range(num_statements):
 
         code.append(generate_statement())
 
     return '\n'.join(code)
 
  
 
 if __name__ == "__main__":
 
     num_statements = random.randint(5, 15)  # 随机生成5到15个语句
 
     random_code = generate_code(num_statements)
 
     print("Generated Python Code:")
 
     print(random_code)
 
  
 
     # 将生成的代码写入一个文件
 
     with open("random_code.py", "w") as file:
 
         file.write(random_code)
 
     print("Code has been written to random_code.py")

说明

generate_variable_name: 生成一个随机的变量名，由字母和数字组成，长度为5。
generate_number: 生成一个0到100之间的随机整数。
generate_statement: 随机生成一个Python语句，包括变量赋值、简单的算术运算、打印语句、条件语句和循环语句。
generate_code: 根据指定的语句数量，生成一个包含多个随机语句的Python代码片段。
主程序: 随机生成5到15个语句的Python代码，并将其打印出来，同时写入一个名为random_code.py的文件中。我们可以编写一个更复杂的程序来生成更高级的Python代码片段。这个程序将能够生成包含函数定义、类定义、导入语句、异常处理以及更多复杂逻辑的代码。

以下是一个示例程序，它使用了更多的随机性和结构来生成复杂的Python代码：

python复制代码
 import random
 
 import string
 
  
 
 def generate_variable_name():
 
     return ''.join(random.choices(string.ascii_letters + string.digits, k=5))
 
  
 
 def generate_number():
 
     return str(random.randint(0, 100))
 
  
 
 def generate_string():
 
     return f'"{random.choice(["hello", "world", "python", "code", "random"])}"'
 
  
 
 def generate_import_statement():
 
     imports = [
 
         "import os",
 
         "import sys",
 
         "import random",
 
         "import math",
 
         "from datetime import datetime"
 
     ]
 
     return random.choice(imports)
 
  
 
 def generate_function_definition():
 
     func_name = generate_variable_name()
 
     params = [f"{generate_variable_name()}" for _ in range(random.randint(0, 3))]
 
     params_str = ', '.join(params) if params else ''
 
     body = '\n    '.join([generate_statement() for _ in range(random.randint(1, 5))])
 
     return f"def {func_name}({params_str}):\n    {body}"
 
  
 
 def generate_class_definition():
 
     class_name = generate_variable_name().capitalize()
 
     methods = [generate_function_definition() for _ in range(random.randint(1, 3))]
 
     init_method = f"    def __init__(self, {', '.join([generate_variable_name() for _ in range(random.randint(1, 2)])}):\n        pass"
 
     body = '\n\n'.join([init_method] + methods) if methods else init_method
 
     return f"class {class_name}:\n{body}"
 
  
 
 def generate_try_except_block():
 
     try_body = '\n    '.join([generate_statement() for _ in range(random.randint(1, 3))])
 
     except_type = random.choice(["Exception", "ValueError", "TypeError", "ZeroDivisionError"])
 
     except_body = '\n    '.join([generate_statement() for _ in range(random.randint(0, 2))])
 
     return f"try:\n    {try_body}\nexcept {except_type}:\n    {except_body}"
 
  
 
 def generate_statement():
 
     statements = [
 
         f"{generate_variable_name()} = {generate_number()}",
 
         f"{generate_variable_name()} = {generate_variable_name()} + {generate_number()}",
 
         f"{generate_variable_name()} = '{generate_string()[1:-1]}'",  # 去掉引号
 
         f"print({generate_variable_name() if random.random() > 0.5 else generate_string()})",
 
         f"if {generate_variable_name()} > {generate_number()}: print('true')",
 
         f"for i in range({generate_number()}): print(i)",
 
         generate_function_definition(),
 
         generate_class_definition(),
 
         generate_try_except_block()
 
     ]
 
     if random.random() > 0.7:  # 有一定概率生成import语句（较低概率以避免过多import）
 
         statements.append(generate_import_statement())
 
     return random.choice(statements)
 
  
 
 def generate_code(num_statements):
 
     code = []
 
     for _ in range(num_statements):
 
         code.append(generate_statement())
 
     # 去除连续的import语句（可选，为了代码整洁）
 
     cleaned_code = []
 
     for stmt in code:
 
         if not cleaned_code or not stmt.startswith("import ") or not cleaned_code[-1].startswith("import "):
 
             cleaned_code.append(stmt)
 
     return '\n\n'.join(cleaned_code)
 
  
 
 if __name__ == "__main__":
 
     num_statements = random.randint(10, 30)  # 生成10到30个语句
 
     random_code = generate_code(num_statements)
 
     print("Generated Python Code:")
 
     print(random_code)
 
  
 
     # 将生成的代码写入一个文件
 
     with open("complex_random_code.py", "w") as file:
 
         file.write(random_code)
 
     print("Code has been written to complex_random_code.py")