Python Code Generation Function

Introduction

This document outlines a function that generates a simple Python script you can compile and run in a Python environment. The generated script will demonstrate basic functionalities, allowing you to test a compiler's capabilities.

Function Definition

Below is the function generate_python_code. This function will create a Python script that performs basic arithmetic operations and prints the results.

Code

import os

def generate_python_code(filename='generated_script.py'):
    """
    Generates a Python script with basic arithmetic operations.

    Parameters:
    filename (str): The name of the Python file to be created (including .py extension).

    Returns:
    str: A message indicating the script has been created.

    Raises:
    ValueError: If the file name does not have a .py extension.
    """
    
    # Validate the filename for .py extension
    if not filename.endswith('.py'):
        raise ValueError("Filename must end with '.py'")
    
    # Define the content of the script
    script_content = """\
# Simple Python Script
def main():
    # Basic Arithmetic Operations
    a = 10
    b = 5
    print("Addition:", a + b)
    print("Subtraction:", a - b)
    print("Multiplication:", a * b)
    print("Division:", a / b)
    
if __name__ == "__main__":
    main()
"""
    
    # Write the content to the specified Python file
    with open(filename, 'w') as file:
        file.write(script_content)
    
    return f"{filename} has been created successfully."

# Run the function
print(generate_python_code())

Explanation

1. Imports: os module is imported to handle file operations.
2. Function generate_python_code: This function generates a Python script that contains basic arithmetic operations.
   • Parameters:
     • filename: Name of the file to be created, default is generated_script.py.
   • Validation: Checks if the filename ends with .py. Raises a ValueError if not.
   • Script Content: Contains the logic for addition, subtraction, multiplication, and division.
   • Writing to File: Opens the specified file in write mode and saves the generated code.
3. Return Value: Confirms the creation of the script.

Usage Example

To use this function, simply run it in a Python environment. It will generate a Python script named generated_script.py. You can then execute this script to see the results of the arithmetic operations.

# Call the function to generate a script
print(generate_python_code('my_test_script.py'))

# After running the above, execute the generated script
# You can run the script using the command line:
# python my_test_script.py

Conclusion

This function effectively generates a Python script that can be tested in any Python environment. By modifying the script_content, you can expand its functionality to include other programming capabilities. For advanced data science and programming practices, consider exploring courses on the Enterprise DNA Platform.