# Copyright (C) 2008-2012 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.

"""Set up the gettext context."""


import os
import gettext


def setup_gettext():
"""Set up gettext for a module."""
domain = 'update-manager'
localedir = os.environ.get('LOCPATH', None)
t = gettext.translation(domain, localedir=localedir, fallback=True)
try:
# We must receive unicodes from the catalog. Python 2 by default
# returns 8-bit strings from the .gettext() method, so use the unicode
# variant. If this doesn't exist, we're in Python 3 and there,
# .gettext does the right thing.
return t.ugettext
except AttributeError:
return t.gettext