| import argparse | |
| from src.universal_refactor.pipelines import InferencePipeline | |
| from src.universal_refactor.utils import read_text | |
| parser=argparse.ArgumentParser() | |
| parser.add_argument('--mode',choices=['refactor','convert'],default='refactor') | |
| parser.add_argument('--file');parser.add_argument('--lang',default='python') | |
| parser.add_argument('--src');parser.add_argument('--tgt') | |
| def main(): | |
| a=parser.parse_args();pipe=InferencePipeline() | |
| code=read_text(a.file) | |
| if a.mode=='refactor': | |
| out=pipe.analyze_and_refactor(code,a.lang);print(out) | |
| else: | |
| print(pipe.convert(code,a.src,a.tgt)) | |
| if __name__=='__main__': main() | |