diff --git a/examples/dict_to_schema/run.py b/examples/dict_to_schema/run.py index d8c633b..838779d 100644 --- a/examples/dict_to_schema/run.py +++ b/examples/dict_to_schema/run.py @@ -1,4 +1,5 @@ import codegen +from codegen.sdk.enums import ProgrammingLanguage from codegen import Codebase @@ -96,7 +97,7 @@ def run(codebase: Codebase): if __name__ == "__main__": print("Initializing codebase...") - codebase = Codebase.from_repo("modal-labs/modal-client", commit="81941c24897889a2ff2f627c693fa734967e693c") + codebase = Codebase.from_repo("modal-labs/modal-client", commit="81941c24897889a2ff2f627c693fa734967e693c", programming_language=ProgrammingLanguage.PYTHON) print("Running codemod...") run(codebase) diff --git a/examples/flask_to_fastapi_migration/run.py b/examples/flask_to_fastapi_migration/run.py index d3e7c03..90db1d3 100644 --- a/examples/flask_to_fastapi_migration/run.py +++ b/examples/flask_to_fastapi_migration/run.py @@ -1,7 +1,7 @@ +import codebase from codegen import Codebase # Initialize codebase -codebase = Codebase("./") # Define the target directory TARGET_DIR = "repo-before" @@ -98,7 +98,8 @@ def update_jinja2_syntax(file): func_call.set_kwarg("request", "request") -def main(): +@codebase.function("flask_to_fastapi_migration") +def run(): """Main function to run the Flask to FastAPI migration""" print("🚀 Starting Flask to FastAPI migration...\n") @@ -128,4 +129,6 @@ def main(): if __name__ == "__main__": - main() + codebase = Codebase("./") + + run() diff --git a/examples/freezegun_to_timemachine_migration/run.py b/examples/freezegun_to_timemachine_migration/run.py index b5e6bd1..543795d 100644 --- a/examples/freezegun_to_timemachine_migration/run.py +++ b/examples/freezegun_to_timemachine_migration/run.py @@ -1,4 +1,5 @@ import codegen +from codegen.sdk.enums import ProgrammingLanguage from codegen import Codebase @@ -58,5 +59,5 @@ def run(codebase: Codebase): if __name__ == "__main__": - codebase = Codebase.from_repo("getmoto/moto", commit="786a8ada7ed0c7f9d8b04d49f24596865e4b7901") + codebase = Codebase.from_repo("getmoto/moto", commit="786a8ada7ed0c7f9d8b04d49f24596865e4b7901", programming_language=ProgrammingLanguage.PYTHON) run(codebase) diff --git a/examples/generate_training_data/run.py b/examples/generate_training_data/run.py index 6c1d8d2..17fd116 100644 --- a/examples/generate_training_data/run.py +++ b/examples/generate_training_data/run.py @@ -2,6 +2,7 @@ import codegen from codegen import Codebase +from codegen.sdk.enums import ProgrammingLanguage from codegen.sdk.core.external_module import ExternalModule from codegen.sdk.core.import_resolution import Import from codegen.sdk.core.symbol import Symbol @@ -94,7 +95,7 @@ def run(codebase: Codebase): if __name__ == "__main__": print("Initializing codebase...") - codebase = Codebase.from_repo("fastapi/fastapi") + codebase = Codebase.from_repo("fastapi/fastapi", commit="887270ff8a54bb58c406b0651678a27589793d2f", programming_language=ProgrammingLanguage.PYTHON) print("Generating training data...") training_data = run(codebase) diff --git a/examples/usesuspensequery_to_usesuspensequeries/run.py b/examples/usesuspensequery_to_usesuspensequeries/run.py index 09a7e69..c68174c 100644 --- a/examples/usesuspensequery_to_usesuspensequeries/run.py +++ b/examples/usesuspensequery_to_usesuspensequeries/run.py @@ -1,5 +1,6 @@ import codegen from codegen import Codebase +from codegen.sdk.enums import ProgrammingLanguage from codegen.sdk.core.detached_symbols.function_call import FunctionCall @@ -76,11 +77,11 @@ def run(codebase: Codebase): print("\nModification complete:") print(f"Files modified: {files_modified}") print(f"Functions modified: {functions_modified}") + codebase.commit() if __name__ == "__main__": print("Initializing codebase...") - codebase = Codebase.from_repo("deepfence/ThreatMapper") - + codebase = Codebase.from_repo("deepfence/ThreatMapper", programming_language=ProgrammingLanguage.TYPESCRIPT) print("Running codemod...") run(codebase)