summaryrefslogtreecommitdiffstats
path: root/Testcases/cfgm_common/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'Testcases/cfgm_common/utils.py')
-rw-r--r--Testcases/cfgm_common/utils.py110
1 files changed, 110 insertions, 0 deletions
diff --git a/Testcases/cfgm_common/utils.py b/Testcases/cfgm_common/utils.py
new file mode 100644
index 0000000..a023912
--- /dev/null
+++ b/Testcases/cfgm_common/utils.py
@@ -0,0 +1,110 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright (c) 2015 Juniper Networks
+# All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+#
+# @author: Numan Siddique, eNovance.
+
+
+import urllib
+from collections import OrderedDict
+import sys
+import cgitb
+import cStringIO
+
+def detailed_traceback():
+ buf = cStringIO.StringIO()
+ cgitb.Hook(format="text", file=buf).handle(sys.exc_info())
+ tb_txt = buf.getvalue()
+ buf.close()
+ return tb_txt
+# end detailed_traceback
+
+def encode_string(enc_str, encoding='utf-8'):
+ """Encode the string using urllib.quote_plus
+
+ Eg. @input:
+ enc_str = 'neté
+ type - 'unicode' or 'str'
+ @retval
+ enc_str = 'net%C3%A9%C3%B9'
+ type - str
+ """
+ try:
+ enc_str.encode()
+ except (UnicodeDecodeError, UnicodeEncodeError):
+ if type(enc_str) is unicode:
+ enc_str = enc_str.encode(encoding)
+ enc_str = urllib.quote_plus(enc_str)
+ except Exception:
+ pass
+ return enc_str
+
+
+def decode_string(dec_str, encoding='utf-8'):
+ """Decode the string previously encoded using urllib.quote_plus.
+
+ Eg. If dec_str = 'net%C3%A9%C3%B9'
+ type - 'unicode' or 'str'
+ @retval
+ ret_dec_str = 'neté
+ type - unicode
+ """
+ ret_dec_str = dec_str
+ try:
+ if type(ret_dec_str) is unicode:
+ ret_dec_str = str(ret_dec_str)
+ ret_dec_str = urllib.unquote_plus(ret_dec_str)
+ return ret_dec_str.decode(encoding)
+ except Exception:
+ return dec_str
+
+
+class CacheContainer(object):
+ def __init__(self, size):
+ self.container_size = size
+ self.dictionary = OrderedDict()
+
+ def __getitem__(self, key, default=None):
+ value = self.dictionary[key]
+ # item accessed - put it in the front
+ del self.dictionary[key]
+ self.dictionary[key] = value
+
+ return value
+
+ def __setitem__(self, key, value):
+ self.dictionary[key] = value
+ if len(self.dictionary.keys()) > self.container_size:
+ # container is full, loose the least used item
+ self.dictionary.popitem(last=False)
+
+ def __contains__(self, key):
+ return key in self.dictionary
+
+ def __repr__(self):
+ return str(self.dictionary)
+
+
+def CamelCase(input):
+ words = input.replace('_', '-').split('-')
+ name = ''
+ for w in words:
+ name += w.capitalize()
+ return name
+#end CamelCase