import commands
from ConfigParser import ConfigParser
import os
import sys

def use_version(version):
    target_str = 'tidb_version = ' + version + '\n'
    new_file = ''
    file = open('inventory.ini', 'r')
    for line in file:
        if 'tidb_version' in line:
            new_file += target_str
        else:
            new_file += line
    file.close()
    os.remove('inventory.ini')
    file = open('inventory.ini', 'w')
    file.write(new_file)
    file.close()

use_version(sys.argv[1])