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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
"""Sourced from https://github.com/nteract/papermill/blob/master/papermill/translators.py"""
import math
import sys
class Translator(object):
@classmethod
def translate_raw_str(cls, val):
"""Reusable by most interpreters"""
return '{}'.format(val)
@classmethod
def translate_escaped_str(cls, str_val):
"""Reusable by most interpreters"""
if isinstance(str_val, str):
str_val = str_val.encode('unicode_escape')
if sys.version_info >= (3, 0):
str_val = str_val.decode('utf-8')
str_val = str_val.replace('"', r'\"')
return '"{}"'.format(str_val)
@classmethod
def translate_str(cls, val):
"""Default behavior for translation"""
return cls.translate_escaped_str(val)
@classmethod
def translate_none(cls, val):
"""Default behavior for translation"""
return cls.translate_raw_str(val)
@classmethod
def translate_int(cls, val):
"""Default behavior for translation"""
return cls.translate_raw_str(val)
@classmethod
def translate_float(cls, val):
"""Default behavior for translation"""
return cls.translate_raw_str(val)
@classmethod
def translate_bool(cls, val):
"""Default behavior for translation"""
return 'true' if val else 'false'
@classmethod
def translate_dict(cls, val):
raise NotImplementedError('dict type translation not implemented for {}'.format(cls))
@classmethod
def translate_list(cls, val):
raise NotImplementedError('list type translation not implemented for {}'.format(cls))
@classmethod
def translate(cls, val):
"""Translate each of the standard json/yaml types to appropriate objects."""
if val is None:
return cls.translate_none(val)
elif isinstance(val, str):
return cls.translate_str(val)
# Needs to be before integer checks
elif isinstance(val, bool):
return cls.translate_bool(val)
elif isinstance(val, int):
return cls.translate_int(val)
elif isinstance(val, float):
return cls.translate_float(val)
elif isinstance(val, dict):
return cls.translate_dict(val)
elif isinstance(val, list):
return cls.translate_list(val)
elif isinstance(val, tuple):
return cls.translate_tuple(val)
elif val.__class__.__name__ == "TestbookObjectReference":
return val.name
# Use this generic translation as a last resort
return cls.translate_escaped_str(val)
@classmethod
def comment(cls, cmt_str):
raise NotImplementedError('comment translation not implemented for {}'.format(cls))
@classmethod
def assign(cls, name, str_val):
return '{} = {}'.format(name, str_val)
class PythonTranslator(Translator):
@classmethod
def translate_float(cls, val):
if math.isfinite(val):
return cls.translate_raw_str(val)
elif math.isnan(val):
return "float('nan')"
elif val < 0:
return "float('-inf')"
else:
return "float('inf')"
@classmethod
def translate_bool(cls, val):
return cls.translate_raw_str(val)
@classmethod
def translate_dict(cls, val):
escaped = ', '.join(
["{}: {}".format(cls.translate_str(k), cls.translate(v)) for k, v in val.items()]
)
return '{{{}}}'.format(escaped)
@classmethod
def translate_list(cls, val):
escaped = ', '.join([cls.translate(v) for v in val])
return '[{}]'.format(escaped)
@classmethod
def translate_tuple(cls, val):
escaped = ', '.join([cls.translate(v) for v in val]) + ', '
return '({})'.format(escaped)
|
