X-Git-Url: https://git.tld-linux.org/?p=tld-builder.git;a=blobdiff_plain;f=TLD_Builder%2Futil.py;h=1d485a170c20378fab39a6ad092f1fd04a5dc1ee;hp=05cf076753ee6ec6ce8ccc4804bbf4f6b01c256c;hb=9ea122b00e3a99b16246e1e38576916a2e25c0aa;hpb=b06ac0a74084c2ed513ecead310ea47208344d14 diff --git a/TLD_Builder/util.py b/TLD_Builder/util.py index 05cf076..1d485a1 100644 --- a/TLD_Builder/util.py +++ b/TLD_Builder/util.py @@ -5,6 +5,7 @@ import sys import os import log import string +import codecs def uuid_python(): return str(uuid_random()) @@ -95,3 +96,19 @@ def cmp_to_key(mycmp): def __ne__(self, other): return mycmp(self.obj, other.obj) != 0 return K + +def to_bytes(s): + if type(s) is bytes: + return s + elif type(s) is str or (sys.version_info[0] < 3 and type(s) is unicode): + return codecs.encode(s, 'utf-8') + else: + raise TypeError("Expected bytes or string, but got %s." % type(s)) + +def to_str(s): + if type(s) is bytes: + return codecs.decode(s, 'utf-8') + elif type(s) is str or (sys.version_info[0] < 3 and type(s) is unicode): + return s + else: + raise TypeError("Expected bytes or string, but got %s." % type(s))