# (c) 2006 Canonical
# Author: Michael Vogt <[email protected]>
#
# Released under the GPL
#

import os
import subprocess
import tempfile

def find_string_and_replace(findString, setString, file_list,
startswith=True, append=True):
""" find all strings that startswith findString and replace them with
setString
"""
for fname in file_list:
out = tempfile.NamedTemporaryFile(delete=False,
dir=os.path.dirname(fname))
foundString = False
if (os.path.exists(fname) and
os.access(fname, os.R_OK)):
# look for the line
with open(fname) as f:
for line in f:
tmp = line.strip()
if startswith and tmp.startswith(findString):
foundString = True
line = setString
if not startswith and tmp == findString:
foundString = True
line = setString
out.write(line.encode('UTF-8'))
# if we have not found them append them
if not foundString and append:
out.write(setString.encode('UTF-8'))
out.flush()
# rename is atomic
os.rename(out.name, fname)
os.chmod(fname, 0o644)

def language2locale(language):
""" generate locale name for LC_* environment variables
"""
first_elem = language.split(':')[0]
locale = subprocess.check_output(
['/usr/share/language-tools/language2locale', first_elem],
universal_newlines=True)
return locale.rstrip()