Rename a list of yang files with revision numbers

rename-by-revision.py

#!/usr/bin/env python
import pyang
import argparse
import os

parser = argparse.ArgumentParser(description='rename a list of yang model files by the revision information')
parser.add_argument('yang_files', type=str, nargs='+',
                    help='list of yang files')
args = parser.parse_args()

repos = pyang.FileRepository(".")
for fname in args.yang_files:
    targ_module = os.path.basename(fname).split(".")[0]
    if '@' in targ_module:
        targ_module = targ_module.split('@')[0]
    ctx = pyang.Context(repos)
    fd = open(fname, 'r')
    text = fd.read()
    ctx.add_module(fname, text)
    for ((m,r), v) in ctx.modules.iteritems():
        if m==targ_module:
            targ_dir = os.path.dirname(fname)
            if len(targ_dir)==0:
                os.rename(fname,targ_module+'@'+r+'.yang')
            else:
                os.rename(fname,targ_dir+'/'+targ_module+'@'+r+'.yang')
                
            break