File size: 648 Bytes
b19c92c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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()
|