Наше приложение использует два файла для подключения к серверам баз данных:
$ ls -l ../secure total 8 -rw-r----- 1 app_env_1 app 1306 Nov 27 16:12 external.properties -rw------- 1 app_env_1 app 12 Nov 27 15:51 secret.key
Задача – написать скрипт, с помощью которого их можно редактировать, не прибегая к использованию редакторов (т.к. у товарищей из Индии это вызывает… трудности…).
UPD Это была первая версия, вторая – тут: Python: модуль ConfigParser для работы с файлами конфигураций, и скрипт редактирования файлов
У нас нет возможности установки дополнительных модулей типа pysed и argparse, поэтому – использовался обычный sed и if/elif для проверки аргументов.
В результате – получился такой скрипт:
#!/usr/bin/env python
import os
import sys
import subprocess
import getpass
import shutil
import time
_USAGE = ('nUsage: %s <--property> [env] OR <--secret>nnExample:nn%s --property main rpln' % (__file__,__file__))
if len(sys.argv) < 2:
print(_USAGE)
sys.exit(3)
DBS = ['main', 'rpl', 'lima', 'rf', 'tcp', 'gpc']
try:
SEC = os.environ['APP_SECURE_LOCATION_PWD']
PROP = os.environ['APP_EXT_PROP']
except KeyError as e:
print('ERROR: %s' % e)
sys.exit(2)
def backup(file):
''''Create backup copy for given in argument file, with name including timestamp.'''
try:
bak = (file + time.strftime('_%Y-%m-%d-%H_%M_bak'))
print('nCreating backup copy of %s to %s...' % (file, bak))
shutil.copyfile(file, bak)
print('Done.')
except IOError as e:
print('ERROR: %s' % e)
sys.exit(1)
def answer(prompt, choice='Yes or no, please!'):
'''Return OK, if input Ok, and False if input No.
Thus - used in other functions, can run or pass next actios.'''
while True:
result = raw_input(prompt)
if result in ('y', 'Y', 'yes', 'Yes'):
return(True)
elif result in ('n', 'N', 'no', 'No'):
return(False)
else:
print(choice)
def new_secret(file):
'''Creates backup of <SEC> config;
clears file;
accepts new keyword twice and check if it is identical;
finally - writes new line to config.'''
backup(file)
try:
with open(SEC, 'w+') as sec:
newsecret1 = getpass.getpass('nPlease, enter new secret keyword: ')
newsecret2 = getpass.getpass('Repeat: ')
if newsecret1 == newsecret2:
print('nWriting new keyword...')
sec.write(newsecret1)
print('Done.nnPlease, note: you need to regenerate all passwords inside %s config.n' % PROP)
else:
print('New keyword must be entered.n')
except KeyboardInterrupt:
print('nnExit.n')
sys.exit(0)
def edit_entry(oldline, file):
'''Edit <PROP> file, replacing old line with new raw_input() line.
Uses OS 'sed' utility.'''
if answer('Edit? [y/n] '):
newline = raw_input('Please,set new value: ')
if answer('nNew value is: %snIs it correct? [y/n] ' % newline):
subprocess.call(("sed -i -e 's/%s/%s/g' %s" % (oldline.rstrip('n'), newline.rstrip('n'), file)), shell=True)
print('New value witen to file.')
def change_property(file, **argv):
'''Can accept few databases names.
If none given - uses <DBS> kist with all databases.
For each database - selects it's HOST (url iside file), USER and it's Password.
Offers edit each entry with edit_entry() function.'''
list = sys.argv[2:]
if len(list) < 2:
list = DBS
backup(file)
try:
for host in list:
print('nDatabase: %s' % host)
with open(PROP, 'r') as prop:
for line in prop.readlines():
if host in line:
if 'url' in line:
print('nHOST: %s' % line.rstrip('n'))
edit_entry(line, PROP)
elif 'user' in line:
print('nUsername: %s' % line.rstrip('n'))
edit_entry(line, PROP)
elif 'password' in line:
print('nPassword: %s' % line.rstrip('n'))
edit_entry(line, PROP)
except KeyboardInterrupt:
print('nnExit.n')
sys.exit(0)
if sys.argv[1] == '--property':
change_property(PROP)
elif sys.argv[1] == '--secret':
new_secret(SEC)
elif sys.argv[1] == '*':
print(_USAGE)
При запуске – скрипт создаёт резервную копию:
$ ls -l ../secure total 12 -rw-r----- 1 app_env_1 app 1306 Nov 27 16:12 external.properties -rw-r----- 1 app_env_1 app 1306 Nov 29 12:44 external.properties_2014-11-29-12_44_bak -rw------- 1 app_env_1 app 12 Nov 27 15:51 secret.key
Как видно из скрипта – можно выбрать, подключение к какой базе или базам редактировать.
Например – для main и rpl:
$ ./app_secure.py --property main rpl Creating backup copy of /home/app_env_1/secure/external.properties to /home/app_env_1/secure/external.properties_2014-11-29-12_44_bak... Done. Database: main HOST: jdbc.main.url=jdbc:oracle:thin:@domain.com:1845:SID1 Edit? [y/n] n Username: jdbc.main.username=OLD_USER Edit? [y/n] n Password: jdbc.main.password=ENC(iFU//78v***Onk) Edit? [y/n] n
И сам процесс редактирования:
$ ./app_secure.py --property main Creating backup copy of /home/app_env_1/secure/external.properties to /home/app_env_1/secure/external.properties_2014-11-29-12_45_bak... Done. Database: main HOST: jdbc.main.url=jdbc:oracle:thin:@domain.com:1845:SID1 Edit? [y/n] n Username: jdbc.main.username=OLD_USER Edit? [y/n] y Please,set new value: jdbc.main.username=NEW_USER_NAME New value is: jdbc.main.username=NEW_USER_NAME Is it correct? [y/n] y New value witen to file.
