summaryrefslogtreecommitdiffstats
path: root/app/utils/special_char_converter.py
blob: fb469bb51d0de0a0a1055e5d9184afd3ee4ac5f6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
###############################################################################
# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
# and others                                                                  #
#                                                                             #
# All rights reserved. This program and the accompanying materials            #
# are made available under the terms of the Apache License, Version 2.0       #
# which accompanies this distribution, and is available at                    #
# http://www.apache.org/licenses/LICENSE-2.0                                  #
###############################################################################
import re


class SpecialCharConverter:

    translated_re = re.compile(r'---[.][.][0-9]+[.][.]---')

    def encode_special_characters(self, s):
        SPECIAL_CHARS = [':', '/']
        for c in SPECIAL_CHARS:
            if c in s:
                s = s.replace(c, '---..' + str(ord(c)) + '..---')
        return s

    def decode_special_characters(self, s):
        replaced = []
        for m in re.finditer(self.translated_re, s):
            match = m.group(0)
            char_code = match[5:len(match)-5]
            if char_code not in replaced:
                replaced.append(char_code)
                s = s.replace(match, chr(int(char_code)))
        return s