96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
#Copyright 2010 Mika Bostrom
|
|
#This program is free software: you can redistribute it and/or modify
|
|
#it under the terms of the GNU Affero 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 Affero General Public License
|
|
#along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#In the "official" distribution you can find the license in agpl-3.0.txt.
|
|
|
|
# Error logging
|
|
import sys
|
|
|
|
# String manipulation
|
|
import codecs
|
|
|
|
# Settings
|
|
import Configuration
|
|
|
|
encoder_to_utf = codecs.lookup('utf-8')
|
|
encoder_to_sys = codecs.lookup(Configuration.LOCALE_ENCODING)
|
|
coder_hex = codecs.lookup('hex_codec')
|
|
|
|
hex_coding = False #FIXME: Should only be on if db is not UTF8 - test in Database.py?
|
|
# I'm saving a few cycles with this one
|
|
not_needed1, not_needed2, not_needed3 = False, False, False
|
|
if Configuration.LOCALE_ENCODING == 'UTF8':
|
|
not_needed1, not_needed2, not_needed3 = True, True, True
|
|
|
|
def to_utf8(s):
|
|
if not_needed1: return s
|
|
|
|
try:
|
|
#(_out, _len) = encoder_to_utf.encode(s)
|
|
_out = unicode(s, Configuration.LOCALE_ENCODING).encode('utf-8')
|
|
return _out
|
|
except UnicodeDecodeError:
|
|
sys.stderr.write(_('Could not convert: "%s"\n') % s)
|
|
raise
|
|
except UnicodeEncodeError:
|
|
sys.stderr.write(_('Could not encode: "%s"\n') % s)
|
|
raise
|
|
except TypeError: # TypeError is raised when we give unicode() an already encoded string
|
|
return s
|
|
|
|
def to_db_utf8(s):
|
|
if not_needed2: return s
|
|
|
|
try:
|
|
(_out, _len) = encoder_to_utf.encode(unicode(s))
|
|
return _out
|
|
except UnicodeDecodeError:
|
|
sys.stderr.write(_('Could not convert: "%s"\n') % s)
|
|
raise
|
|
except UnicodeEncodeError:
|
|
sys.stderr.write(_('Could not encode: "%s"\n') % s)
|
|
raise
|
|
|
|
def to_gui(s):
|
|
if not_needed3: return s
|
|
|
|
try:
|
|
# we usually don't want to use 'replace' but this is only for displaying
|
|
# in the gui so it doesn't matter if names are missing an accent or two
|
|
(_out, _len) = encoder_to_sys.encode(s, 'replace')
|
|
return _out
|
|
except UnicodeDecodeError:
|
|
sys.stderr.write(_('Could not convert: "%s"\n') % s)
|
|
raise
|
|
except UnicodeEncodeError:
|
|
sys.stderr.write(_('Could not encode: "%s"\n') % s)
|
|
raise
|
|
|
|
def to_hex(s):
|
|
try:
|
|
out = coder_hex.encode(s)[0]
|
|
return out
|
|
except UnicodeDecodeError:
|
|
sys.stderr.write(_('Could not convert: "%s"\n') % s)
|
|
return s
|
|
|
|
def from_hex(s):
|
|
try:
|
|
out = coder_hex.decode(s)[0]
|
|
return out
|
|
except UnicodeDecodeError:
|
|
sys.stderr.write(_('Could not convert: "%s"\n') % s)
|
|
return s
|