def split(f, delimiter, pseudo_type): if delimiter.type == 'string' and (len(delimiter.value) == 1 or delimiter.value == '\\n'): return method_call(f, 'Split', [Node('char', value=delimiter.value, pseudo_type='Char')], ['List', 'String']) else: return method_call(f, 'Split', [Node('array', elements=[delimiter]), attr(typename('StringSplitOptions', 'Library'), 'None', 'CSharpNone')], pseudo_type=['List', 'String'])
def x(l, f, *args): pseudo_type, args = args[-1], list(args[:-1]) if args and swap: f, args[0] = args[0], f cs = method_call(l, name, [f] + args, pseudo_type) if z: cs.pseudo_type = 'CSharpEnumerable' return method_call(cs, 'ToList', [], pseudo_type) else: return cs
def expand_slice(receiver, from_, to, pseudo_type=None): return method_call( method_call( receiver, 'Take', [Node('binary_op', op='-', left=attr(receiver, 'Length', 'Int'), right=to_node(-to.value), pseudo_type='Int') if to.type == 'int' and to.value < 0 else to], pseudo_type=pseudo_type), 'Drop', [Node('binary_op', op='-', left=attr(receiver, 'Length', 'Int'), right=to_node(-from_.value), pseudo_type='Int') if from_.type == 'int' and from_.value < 0 else from_], pseudo_type=pseudo_type)
def as_expression(self): return [Node( '_py_with', call=call('open', [self.args[0], to_node('r')], 'File'), context='_f', block=[method_call(local('_f', 'File'), 'read', [], 'String')], pseudo_type='Void')], None
def as_expression(self): return [], Node( '_py_with', call=call('open', [self.args[0], to_node('w')], 'File'), context='_f', block=[method_call(local('_f', 'File'), 'write', [self.args[1]], 'Void')], pseudo_type='Void')
def object_len(l, _): return attr( method_call( typename('Object', 'Type'), 'keys', [l], ['List', l.pseudo_type[1]]), 'length', 'Int')
def as_expression(self): return [ Node( '_py_with', call=call('open', [self.args[0], to_node('r')], 'File'), context='_f', block=[method_call(local('_f', 'File'), 'read', [], 'String')], pseudo_type='Void') ], None
def as_expression(self): return [], Node('_py_with', call=call('open', [self.args[0], to_node('w')], 'File'), context='_f', block=[ method_call(local('_f', 'File'), 'write', [self.args[1]], 'Void') ], pseudo_type='Void')
def _expand_api(self, api, receiver, args, pseudo_type, equivalent): ''' the heart of api translation dsl function or <z>(<arg>, ..) can be expanded, <z> can be just a name for a global function, or #name for method, <arg> can be %{self} for self or %{n} for nth arg ''' if callable(api): if receiver: return api(receiver, *(args + [pseudo_type])) else: return api(*(args + [pseudo_type])) elif isinstance(api, str): if '(' in api: call_api, arg_code = api[:-1].split('(') new_args = [ self._parse_part(a.strip(), receiver, args, equivalent) for a in arg_code.split(',') ] else: call_api, arg_code = api, '' new_args = args if '#' in call_api: a, b = call_api.split('#') method_receiver = self._parse_part( a, receiver, args, equivalent) if a else receiver return method_call(method_receiver, b, new_args, pseudo_type=pseudo_type) elif '.' in call_api: a, b = call_api.split('.') static_receiver = self._parse_part( a, receiver, args, equivalent) if a else receiver if b[-1] != '!': return Node('static_call', receiver=static_receiver, message=b, args=new_args, pseudo_type=pseudo_type) else: return Node('attr', object=static_receiver, attr=b[:-1], pseudo_type=pseudo_type) else: if receiver: return call(call_api, [receiver] + new_args, pseudo_type=pseudo_type) else: return call(call_api, new_args, pseudo_type=pseudo_type) else: raise PseudoDSLError('%s not supported by api dsl' % str(api))
def pad(f, count, fill, pseudo_type): return method_call( method_call( f, 'PadLeft', [Node('binary_op', op='+', left=Node('binary_op', op='/', left=Node('binary_op', op='-', left=count, right=attr(f, 'Length', 'Int'), pseudo_type='Int'), right=to_node(2), pseudo_type='Int'), right=attr(f, 'Length', 'Int'), pseudo_type='Int'), fill], pseudo_type='String'), 'PadRight', [count, fill], 'String')
def as_expression(self): # pseudo_type=self.args[0].pseudo_type begin = method_call(self.args[0], 'begin', [], 'CppIterator') end = method_call(self.args[0], 'end', [], 'CppIterator') if self.name == 'slice_to': from_, to = to_node(0), self.args[1] else: from_, to = self.args[1], self.args[2] if from_.type == 'int' and from_.value == 0: start = begin else: start = Node('binary_op', op='+', left=begin, right=from_, pseudo_type='CppIterator') if self.name == 'slice_from': finish = end elif to.type == 'int' and to.value < 0: finish = Node('binary_op', op='-', left=end, right=to_node(-to.value)) else: finish = Node('binary_op', op='+', left=begin, right=to, pseudo_type='CppIterator') return [ Node('_cpp_declaration', name='_sliced', args=[start, finish], decl_type=self.args[0].pseudo_type, pseudo_type='Void')], None
def _expand_api(self, api, receiver, args, pseudo_type, equivalent): ''' the heart of api translation dsl function or <z>(<arg>, ..) can be expanded, <z> can be just a name for a global function, or #name for method, <arg> can be %{self} for self or %{n} for nth arg ''' if callable(api): if receiver: return api(receiver, *(args + [pseudo_type])) else: return api(*(args + [pseudo_type])) elif isinstance(api, str): if '(' in api: call_api, arg_code = api[:-1].split('(') new_args = [self._parse_part( a.strip(), receiver, args, equivalent) for a in arg_code.split(',')] else: call_api, arg_code = api, '' new_args = args if '#' in call_api: a, b = call_api.split('#') method_receiver = self._parse_part( a, receiver, args, equivalent) if a else receiver return method_call(method_receiver, b, new_args, pseudo_type=pseudo_type) elif '.' in call_api: a, b = call_api.split('.') static_receiver = self._parse_part( a, receiver, args, equivalent) if a else receiver if b[-1] != '!': return Node('static_call', receiver=static_receiver, message=b, args=new_args, pseudo_type=pseudo_type) else: return Node('attr', object=static_receiver, attr=b[:-1], pseudo_type=pseudo_type) else: if receiver: return call(call_api, [receiver] + new_args, pseudo_type=pseudo_type) else: return call(call_api, new_args, pseudo_type=pseudo_type) else: raise PseudoDSLError('%s not supported by api dsl' % str(api))
def transform__cpp_cout(self, node, in_block=False, assignment=None): for j, ar in enumerate(node.args): if ar.pseudo_type == 'Exception' or ar.pseudo_type.endswith('Error'): node.args[j] = method_call(ar, 'what', [], 'String') return node
InstanceVariable = [Node('instance_variable', name='egg')] Attr = [Node('attr', object=local('e'), attr='egg')] Assignment = [ Node('assignment', target=local('egg', pseudo_type='Int'), value=local('ham', pseudo_type='Int')), Node('assignment', target=Node('instance_variable', name='egg', pseudo_type='Int'), value=local('ham', pseudo_type='Int')), Node('assignment', target=Node('attr', object=Node('typename', name='T'), attr='egg', pseudo_type='String'), value=local('ham', pseudo_type='String')), Node('assignment', target=Node('index', sequence=local('x', pseudo_type=['List', 'String']), index=to_node(4), pseudo_type='String'), value=to_node('String')) ] Call = [call('map', [local('x')])] MethodCall = [method_call(local('e', 'E'), 'filter', [to_node(42)])] StandardCall = [ Node('standard_call', namespace='io', function='display', args=[to_node(42)], pseudo_type='Void'), Node('standard_call', namespace='io', function='read', args=[], pseudo_type='String'), Node('standard_call', namespace='math', function='ln', args=[Node('local', name='ham', pseudo_type='Int')], pseudo_type='Float'), assignment( local('source', pseudo_type='String'), Node('standard_call', namespace='io', function='read_file', args=[to_node('f.py')], pseudo_type='String')) ] IoDisplay = standard_call('io', 'display', [to_node(2), to_node('z')], 'Void') IoRead = assignment(local('source', 'String'), standard_call('io', 'read', [], 'String')) IoReadFile = assignment(local('source', 'String'), standard_call('io', 'read_file', [to_node('z.py')], 'String')) IoWriteFile = standard_call('io', 'write_file', [to_node('z.py'), local('source', 'String')], 'Void') INT_EXAMPLE = local('z', 'Int')
class CSharpTranslator(ApiTranslator): ''' CSharp api translator The DSL is explained in the ApiTranslator docstring ''' methods = { 'List': { '@equivalent': 'List', 'push': '#Insert', 'pop': lambda l, _: method_call(l, 'RemoveAt', [ Node('binary_op', op='-', left=attr(l, 'Length', 'Int'), right=to_node(1), pseudo_type='Int') ], pseudo_type='Void'), 'insert': '#Insert', 'remove_at': '#RemoveAt', 'remove': '#Remove', 'length': '.Length!', 'slice': expand_slice, 'slice_from': lambda l, from_, pseudo_type: method_call( l, 'Drop', [ Node('binary_op', op='-', left=attr(l, 'Length', 'Int'), right=to_node(-from_.value), pseudo_type='Int') if from_.type == 'int' and from_.value < 0 else from_ ], pseudo_type=pseudo_type), 'slice_to': lambda l, to, pseudo_type: method_call( l, 'Take', [ Node('binary_op', op='-', left=attr(l, 'Length', 'Int'), right=to_node(-to.value), pseudo_type='Int') if to.type == 'int' and to.value < 0 else to ], pseudo_type=pseudo_type), 'slice_': '#Slice', 'join': lambda l, delimiter, _: method_call(delimiter, 'Join', [receiver], 'String'), 'map': linq('Select'), 'filter': linq('Where'), 'reduce': linq('Aggregate', False, swap=True), 'all?': linq('All'), 'any?': linq('Any'), 'concat': '#AddRange', 'present?': '#Any', 'empty?': empty, 'find': '#IndexOf', 'contains?': '#Contains', 'sort': '#Sort' }, 'Dictionary': { '@equivalent': 'Dictionary', 'length': '.Length!', 'keys': '#Keys', 'values': '#Values', 'contains?': '#Contains', 'present?': '#Any', 'empty?': empty }, 'String': { '@equivalent': 'String', 'length': '.Length!', 'substr': lambda f, from_, to, pseudo_type: method_call( f, 'Substring', [from_, normalize(f, from_, to)], pseudo_type), 'substr_from': '#Substring', 'substr_to': lambda f, to, pseudo_type: method_call(f, 'Substring', [ to_node(0), normalize(f, to_node(0), to) ], pseudo_type), 'find_from': '#IndexOf', 'find': '#IndexOf', 'count': lambda f, element, pseudo_type: method_call( f, 'Count', [ Node('anonymous_function', params=[local('sub', 'String')], block=[ Node('comparison', op='==', left=local('sub', 'String'), right=local('s', 'String'), pseudo_type='Boolean') ], return_type='Boolean', pseudo_type=['Function', 'String', 'Boolean']) ], pseudo_type='Int'), 'concat': to_op('+'), 'split': split, 'trim': '#Trim', 'center': pad, 'present?': lambda f, _: Node('comparison', op='!=', left=attr(f, 'Length', 'Int'), right=to_node(0), pseudo_type='Boolean'), 'empty?': lambda f, _: Node('comparison', op='==', left=attr(f, 'Length', 'Int'), right=to_node(0), pseudo_type='Boolean'), 'contains?': '#Contains', 'to_int': 'Int32.Parse(%{self})', 'pad_left': '#PadLeft', 'pad_right': '#PadRight' }, 'Set': { '@equivalent': 'HashSet', 'length': '.Length!', 'contains?': '#Contains', 'union': '#Union', 'intersection': '#Intersection', 'present?': '#Any', 'empty?': empty }, 'Regexp': { '@equivalent': 'Regexp', 'match': '#match' }, 'RegexpMatch': { '@equivalent': 'RegexpMatch', 'group': lambda receiver, index, _: Node( 'index', sequence=attr( Node('index', sequence=attr(receiver, 'Groups', ['List', 'CSharpRegexGroup']), index=to_node(1 + index.value) if index.type == 'int' else Node('binary_op', op='+', left=to_node(1), right=index, pseudo_type='Int'), pseudo_type='CSharpRegexGroup'), 'Captures', ['List', 'RegexpMatch']), index=to_node(0), pseudo_type='RegexpMatch'), 'has_match': '.Success!' }, 'Array': { '@equivalent': 'Any[]', 'length': lambda receiver, _: to_node(receiver.pseudo_type[2]) }, 'Tuple': { '@equivalent': 'Tuple', 'length': lambda _, pseudo_type: to_node(len(pseudo_type) - 1) } } functions = { 'io': { 'display': Display, 'read': 'Console.ReadLine', 'read_file': 'File.ReadAllText', 'write_file': 'File.WriteAllText' }, 'math': { 'ln': 'Math.Log', 'log': 'Math.Log', 'tan': 'Math.Tan', 'sin': 'Math.Sin', 'cos': 'Math.Cos', 'pow': 'Math.Pow' }, 'http': {}, 'system': { 'args': lambda _: local('args', ['List', 'String']), 'index': lambda index, _: Node('index', sequence=local('args', ['List', 'String']), index=to_node(index.value - 1) if index.type == 'int' else Node('binary_op', op='-', left=index, right=to_node(1), pseudo_type='Int'), pseudo_type='String'), 'arg_count': lambda _: Node('binary_op', op='+', left=attr(local('args', ['List', 'String']), 'Length', 'Int'), right=to_node(1), pseudo_type='Int') }, 'regexp': { 'compile': lambda value, _: Node('new_instance', class_name='Regex', args=[value], pseudo_type='Regexp'), 'escape': 'Regex.Escape' } } dependencies = { 'List': { '@all': 'System.Collections.Generic', 'map': 'System.Linq', 'filter': 'System.Linq', 'any?': 'System.Linq', 'all?': 'System.Linq', 'reduce': 'System.Linq' }, 'String': { '@all': 'System.Text' }, 'Dictionary': { '@all': 'System.Collections.Generic', }, 'io': { 'read_file': 'System.IO', 'write_file': 'System.IO' }, 'regexp': { '@all': 'System.Text.RegularExpressions' } }
def end_placeholder(self, receiver, *args, equivalent): return method_call(receiver, 'end', [], 'CppIterator')
value=local('ham', pseudo_type='Int')), Node('assignment', target=Node('attr', object=Node('typename', name='T'), attr='egg', pseudo_type='String'), value=local('ham', pseudo_type='String')), Node('assignment', target=Node('index', sequence=local('x', pseudo_type=['List', 'String']), index=to_node(4), pseudo_type='String'), value=to_node('String')) ] Call = [call('map', [local('x')])] MethodCall = [method_call(local('e', 'E'), 'filter', [to_node(42)])] StandardCall = [ Node('standard_call', namespace='io', function='display', args=[to_node(42)], pseudo_type='Void'), Node('standard_call', namespace='io', function='read', args=[], pseudo_type='String'), Node('standard_call', namespace='math', function='ln', args=[Node('local', name='ham', pseudo_type='Int')],
def object_len(l, _): return attr( method_call(typename('Object', 'Type'), 'keys', [l], ['List', l.pseudo_type[1]]), 'length', 'Int')
class RubyTranslator(ApiTranslator): ''' Ruby api translator The DSL is explained in the ApiTranslator docstring ''' methods = { 'List': { '@equivalent': 'Array', 'push': '#push', 'pop': '#pop', 'length': '#length', 'insert': '#insert', 'remove_at': '#delete_at', 'remove': '#delete', 'slice': expand_slice, 'slice_from': expand_slice, 'slice_to': lambda receiver, to, pseudo_type: expand_slice( receiver, to_node(0), to, pseudo_type), 'join': '#join', 'map': to_method_rb_block('map'), 'filter': to_method_rb_block('select'), 'reduce': to_method_rb_block('reduce'), 'any?': to_method_rb_block('any?'), 'all?': to_method_rb_block('all?'), 'sort': '#sort!', 'present?': lambda receiver, _: Node('unary_op', op='not', value=method_call(receiver, 'empty?', [], 'Boolean'), pseudo_type='Boolean'), 'empty?': '#empty?', 'find': '#index', 'contains?': '#include?' }, 'Dictionary': { '@equivalent': 'Hash', 'length': '#length', 'keys': '#keys', 'values': '#values', 'contains?': '#include?', 'present?': lambda receiver, _: Node('unary_op', op='not', value=method_call(receiver, 'empty?', [], 'Boolean'), pseudo_type='Boolean'), 'empty?': '#empty?' }, 'Set': { '@equivalent': 'Set', 'length': '#length', 'contains?': '#include?', 'union': to_op('|'), 'intersection': '#intersection', 'present?': lambda receiver, _: Node('unary_op', op='not', value=method_call(receiver, 'empty?', [], 'Boolean'), pseudo_type='Boolean'), 'empty?': '#empty?' }, 'Tuple': { '@equivalent': 'Array', 'length': '#length' }, 'Array': { '@equivalent': 'Array', 'length': '#length' }, 'String': { '@equivalent': 'String', 'substr': expand_slice, 'substr_from': expand_slice, 'substr_to': lambda receiver, to, _: expand_slice(receiver, None, to, 'String'), 'length': '#length', 'concat': to_op('+'), 'find': '#index', 'find_from': '#index', 'count': '#count', 'partition': '#partition', 'split': '#split', 'trim': '#trim', 'reversed': '#reverse', 'center': '#center', 'present?': lambda receiver, _: Node('unary_op', op='not', value=method_call(receiver, 'empty?', [], 'Boolean'), pseudo_type='Boolean'), 'empty?': '#empty?', 'contains?': '#include?', 'to_int': '#to_i', 'pad_left': '#ljust', 'pad_right': '#rjust' }, 'Regexp': { '@equivalent': 'Regexp', 'match': '%{0}#scan(%{self})' }, 'Int': { 'to_float': '#to_f' }, 'Float': { 'to_int': '#to_i' }, 'RegexpMatch': { '@equivalent': 'Regexp_', 'group': lambda receiver, index, _: Node('index', sequence=Node('index', sequence=receiver, index=index, pseudo_type= ['List', 'String']), index=to_node(0), pseudo_type='String'), 'has_match': lambda receiver, _: Node('unary_op', op='not', value=method_call(receiver, 'empty?', [], 'Boolean'), pseudo_type='Boolean') } } functions = { 'global': { 'wat': lambda _: Node('block', block=[]), 'exit': lambda status, _: call('exit', [status]) }, 'io': { 'display': display, 'read': 'gets', 'read_file': 'File.read', 'write_file': 'File.write' }, 'math': { 'ln': 'Math.log', 'log': 'Math.log', 'tan': 'Math.tan', 'sin': 'Math.sin', 'cos': 'Math.cos', 'pow': lambda left, right, pseudo_type: Node('binary_op', op='**', left=left, right=right, pseudo_type=pseudo_type) }, 'regexp': { 'compile': lambda value, _: Node( '_rb_regex_interpolation', value=value, pseudo_type='Regexp'), 'escape': 'Regexp.escape' }, 'system': { 'args': lambda _: typename('ARGV', ['List', 'String']), 'arg_count': lambda _: Node('binary_op', op='+', left=method_call(local('ARGV', ['List', 'String']), 'length', [], 'Int'), right=to_node(1), pseudo_type='Int'), 'index': lambda value, _: Node('index', sequence=local('ARGV', ['List', 'String']), index=to_node(value.value - 1) if value.type == 'int' else Node('binary_op', op='-', left=value, right=to_node(1), pseudo_type='Int'), pseudo_type='String') # in ruby args counting starts from 0, not 1 } } dependencies = {'http': {'@all': 'Requests'}, 'Set': {'@all': 'set'}}
def expand_slice(receiver, from_, to, pseudo_type=None): return method_call(method_call(receiver, 'Take', [to], pseudo_type), 'Drop', [from_], pseudo_type)
class GolangTranslator(ApiTranslator): ''' Go api translator The DSL is explained in the ApiTranslator docstring ''' methods = { 'List': { '@equivalent': 'slice', 'push': lambda receiver, element, _: assignment( receiver, call('append', [receiver, element], receiver.pseudo_type)), 'pop': lambda receiver, _: assignment( receiver, Node('_go_slice_to', sequence=receiver, to=Node('binary_op', op='-', left=call('len', [receiver], 'Int'), right=to_node(1), pseudo_type='Int'), pseudo_type=receiver.pseudo_type)), 'length': 'len', 'insert': expand_insert, 'slice': expand_slice, 'slice_from': expand_slice, 'slice_to': lambda receiver, to, pseudo_type: expand_slice( receiver, None, to, pseudo_type), 'join': 'strings.Join(%{self}, %{0})', 'map': expand_map, 'filter': expand_filter, 'find': Find, 'reduce': expand_reduce, 'contains?': ListContains, 'present?': present, 'empty?': empty }, 'Dictionary': { '@equivalent': 'map', 'length': 'len', 'contains?': Contains, 'keys': DictKeys, 'values': DictValues, 'present?': present, 'empty?': empty }, 'String': { '@equivalent': 'str', 'substr': expand_slice, 'substr_from': expand_slice, 'length': 'len', 'substr_to': lambda receiver, to, _: expand_slice( receiver, None, to, pseudo_type='String'), 'find': 'strings.Index(%{self}, %{0})', 'count': 'strings.Count(%{self}, %{0})', 'split': 'strings.Split(%{self}, %{0})', 'concat': to_op('+'), 'contains?': 'strings.Contains(%{self}, %{0})', 'present?': present, 'empty?': empty, 'find_from': lambda f, value, index, _: Node( 'binary_op', op='+', pseudo_type='Int', left=index, right=Node('static_call', receiver=local('strings', 'Library'), message='Index', args=[ Node('_go_slice_from', sequence=f, from_=index, pseudo_type='String'), value ], pseudo_type='Int')), 'to_int': Int }, 'Regexp': { '@equivalent': 'Regexp', 'match': lambda receiver, word, _: method_call( receiver, 'FindAllSubmatch', [ Node('_go_bytes', value=word, pseudo_type='Bytes'), to_node(-1) ], pseudo_type='RegexpMatch') }, 'RegexpMatch': { '@equivalent': 'R', 'group': lambda receiver, index, _: Node('index', sequence=receiver, index=Node('binary_op', op='+', left=index, right=to_node(1), pseudo_type='Int'), pseudo_type='String'), 'has_match': lambda receiver, _: Node('comparison', op='!=', left=receiver, right=Node('null', pseudo_type='None'), pseudo_type='Boolean') }, 'Set': { '@equivalent': 'map[bool]struct{}', 'length': 'len', 'contains?': Contains, 'present?': present, 'empty?': empty }, 'Tuple': { '@equivalent': 'L', 'length': lambda receiver, _: to_node(len(receiver.pseudo_type) - 1) }, 'Array': { '@equivalent': 'int[]', 'length': lambda receiver, _: to_node(receiver.pseudo_type[2]) } } functions = { 'regexp': { 'compile': 'regexp.MustCompile', 'escape': 'regexp.QuoteMeta' }, 'io': { 'display': 'fmt.Println', 'read': Read, 'read_file': ReadFile, 'write_file': 'ioutil.WriteFile' }, 'math': { 'ln': 'math.Log', 'log': 'math.Log', 'tan': 'math.Tan', 'sin': 'math.Sin', 'cos': 'math.Cos', 'pow': 'math.Pow' }, 'system': { 'args': 'os.Args!', 'arg_count': lambda _: call('len', [ attr(local('os', 'Library'), 'Args', ['List', 'String']) ], 'Int'), 'index': lambda value, _: Node('index', sequence=attr(local('os', 'Library'), 'Args', ['List', 'String']), index=value, pseudo_type='String') } } dependencies = { 'regexp': { '@all': 'regexp' }, 'io': { 'display': 'fmt', 'read': ['bufio', 'os'], 'read_file': ['io/ioutil'], 'write_file': ['io/ioutil'] }, 'math': { '@all': 'math' }, 'List': { 'join': 'strings' }, 'String': { 'find': 'strings', 'count': 'strings', 'split': 'strings', 'contains?': 'strings', 'find_from': 'strings', 'to_int': 'strconv' }, 'system': { '@all': 'os' } } errors = {}
def empty(l, _): return Node('unary_op', op='not', pseudo_type='Boolean', value=method_call(l, 'Any', [], 'Boolean'))
class JSTranslator(ApiTranslator): ''' JS api translator The DSL is explained in the ApiTranslator docstring ''' methods = { 'List': { '@equivalent': 'Array', 'push': '#push', 'pop': '#pop', 'length': '.length!', 'insert': '#splice(%{self}, 0, %{0})', 'remove_at': lambda receiver, index, _: method_call( receiver, 'splice', [ index, to_node(index.value + 1) if index.type == 'int' else Node('binary_op', op='+', left=index, right=to_node(1), pseudo_type='Int') ], pseudo_type='Void'), 'remove': '_.pull(%{self}, %{0})', 'slice': '#slice', 'slice_from': '#slice', 'slice_to': '#slice(0, %{0})', 'map': '_.map(%{self}, %{0})', 'filter': '_.filter(%{self}, %{0})', 'reduce': '_.reduce(%{self}, %{0}, %{1})', 'any?': '_.any(%{self}, %{0})', 'all?': '_.all(%{self}, %{0})', 'sort': '#sort', 'empty?': empty, 'present?': present, 'find': '#indexOf', 'contains?': '_.contains(%{self}, %{0})' }, 'Dictionary': { '@equivalent': 'Object', 'length': object_len, 'keys': 'Object.keys(%{self})', 'values': 'Object.values(%{self})', 'contains?': '#hasOwnProperty', 'present?': present, 'empty?': empty }, 'String': { '@equivalent': 'String', 'substr': '#slice', 'substr_from': '#slice', 'length': '.length!', 'substr_to': '#slice(0, %{0})', 'find': '#search', 'find_from': lambda f, value, index, _: Node( 'binary_op', op='+', pseudo_type='Int', left=index, right=method_call(method_call(f, 'slice', [index], 'String'), 'search', [value], 'Int')), 'count': lambda f, count, _: attr( method_call(LODASH, 'where', [count], ['List', 'String']), 'length', 'Int'), 'concat': to_op('+'), 'partition': lambda f, delimiter, _: Node( 'index', sequence=method_call(LODASH, 'partition', [delimiter], 'String' ), index=to_node(1), pseudo_type=['Tuple', 'String', 'String', 'String']), 'split': '#split', 'trim': '#trim', 'reversed': lambda f, _: method_call( method_call( method_call(f, 'split', [to_node('')], ['List', 'String']), 'reverse', [], ['List', 'String']), 'join', [to_node('')], 'String'), 'center': '_.pad(%{self}, %{0}, %{1})', 'present?': lambda f, _: f, 'empty?': lambda f, _: Node( 'unary_op', op='not', value=f, pseudo_type='Boolean'), 'contains?': '_.contains(%{self}, %{0})', 'to_int': 'parseInt', 'pad_left': '_.padLeft(%{self}, %{0}, %{1})', 'pad_right': '_.padRight(%{self}, %{0}, %{1})' }, 'Tuple': { '@equivalent': 'Array', 'length': '.length!' }, 'Array': { '@equivalent': 'Array', 'length': '.length!' }, 'Set': { '@equivalent': 'Object', 'length': object_len, 'contains?': '_.contains(%{self}, %{0})', 'intersection': '_.intersection(%{self}, %{0})', 'union': '_.union(%{self}, %{0})', 'present?': present, 'empty?': empty }, 'RegexpMatch': { '@equivalent': 'Array', 'group': lambda receiver, index, _: Node('index', sequence=receiver, index=to_node(1 + index.value) if index.type == 'int' else Node( 'binary_op', op='+', left=to_node(1), right=index, pseudo_type='Int'), pseudo_type='String'), 'has_match': lambda receiver, _: receiver }, 'Regexp': { '@equivalent': 'Regexp', 'match': '#exec' } } functions = { 'global': { 'wat': lambda _: Node('block', block=[]), 'exit': lambda status, _: call('exit', [status]) }, 'system': { 'args': 'process.argv!', 'arg_count': lambda _: Node('binary_op', op='-', left=attr( attr(local('process', 'Library'), 'argv', ['List', 'String']), 'length', 'Int'), right=to_node(1), pseudo_type='Int'), 'index': lambda value, _: Node('index', sequence=attr(local('process', 'Library'), 'argv', ['List', 'String']), index=to_node(value.value + 1) if value.type == 'int' else Node('binary_op', op='+', left=value, right=value, pseudo_type='Int'), pseudo_type='String') }, 'io': { 'display': 'console.log', 'read_file': "fs.readFileSync(%{0}, 'utf8')", 'write_file': "fs.writeFileSync(%{0}, %{1}, 'utf8')", }, 'math': { 'ln': 'Math.log', 'log': 'Math.log', 'tan': 'Math.tan', 'sin': 'Math.sin', 'cos': 'Math.cos', 'pow': 'Math.pow' }, 'regexp': { 'compile': lambda value, _: Node('new_instance', class_name='RegExp', args=[value], pseudo_type='Regexp'), 'escape': '_.escapeRegExp' } } js_dependencies = {'_': 'lodash'} dependencies = { 'io': { 'read_file': 'fs', 'write_file': 'fs' }, 'regexp': { 'escape': 'lodash' } }
class PythonTranslator(ApiTranslator): ''' Python api translator The DSL is explained in the ApiTranslator docstring ''' methods = { 'List': { '@equivalent': 'list', 'push': '#append', 'pop': '#pop', 'length': 'len', 'insert': '#insert', 'remove_at': lambda receiver, index, _: Node( '_py_del', value=Node('index', sequence=receiver, index=index), pseudo_type='Void'), 'remove': '#remove', 'slice': expand_slice, 'slice_from': expand_slice, 'slice_to': lambda receiver, to, pseudo_type: expand_slice( receiver, None, to, pseudo_type), 'repeat': to_op('*'), 'set_slice': expand_set_slice, 'set_slice_from': expand_set_slice, 'set_slice_to': lambda receiver, to, value, pseudo_type: expand_set_slice( receiver, None, to, value, pseudo_type), 'find': '#index', 'join': lambda receiver, delimiter, _: method_call( delimiter, 'join', [receiver], pseudo_type='String'), # map and filter have implicit or explicit return because otherwise they wouldnt type check 'map': lambda receiver, f, pseudo_type: Node( '_py_listcomp', sequences=Node('for_sequence', sequence=receiver), iterators=Node('for_iterator', iterator=local(f.params[0].name, f.pseudo_type[ 1])), block=f.block[0].value, test=None, pseudo_type=['List', f.pseudo_type[2]]), 'filter': lambda receiver, test, pseudo_type: Node( '_py_listcomp', sequences=Node('for_sequence', sequence=receiver), iterators=Node('for_iterator', iterator=local(test.params[0].name, test. pseudo_type[1])), block=local(test.params[0].name, test.pseudo_type[1]), test=test.block[0].value, pseudo_type=['List', test.pseudo_type[1]]), 'reduce': lambda receiver, aggregator, initial, pseudo_type: Node( 'static_call', receiver=local('functools', 'Library'), message='reduce', args=[aggregator, receiver, initial], pseudo_type=pseudo_type), 'any?': to_py_generatorcomp('any'), 'all?': to_py_generatorcomp('all'), 'sort': '#sort', 'present?': lambda receiver, _: receiver, 'empty?': lambda receiver, _: Node( 'unary_op', op='not', value=receiver, pseudo_type='Boolean'), 'contains?': contains }, 'Dictionary': { '@equivalent': 'dict', 'length': 'len', 'keys': '#keys', 'values': '#values', 'contains?': contains, 'keys': lambda receiver, pseudo_type: call(local( 'list', ['Function', 'Any', 'List']), [ method_call(receiver, 'keys', [], ['dict_keys', pseudo_type[1]]) ], pseudo_type=pseudo_type), 'values': lambda receiver, pseudo_type: call(local( 'list', ['Function', 'Any', 'List']), [ method_call(receiver, 'values', [], ['dict_values', pseudo_type[1]]) ], pseudo_type=pseudo_type), 'present?': lambda receiver, _: receiver, 'empty?': lambda receiver, _: Node( 'unary_op', op='not', value=receiver, pseudo_type='Boolean') }, 'String': { '@equivalent': 'str', 'substr': expand_slice, 'substr_from': expand_slice, 'length': 'len', 'substr_to': lambda receiver, to, _: expand_slice(receiver, None, to, 'String'), 'find': '#index', 'find_from': '#index', 'count': '#count', 'partition': '#partition', 'split': '#split', 'trim': '#strip', 'format': '#format', 'concat': to_op('+'), 'c_format': to_op('%'), 'center': '#center', 'reversed': 'reversed', 'empty?': lambda receiver, _: Node( 'unary_op', op='not', value=receiver, pseudo_type='Boolean'), 'present?': lambda receiver, _: receiver, 'contains?': lambda receiver, element, _: Node('_py_in', sequence=receiver, value=element, pseudo_type='Boolean'), 'to_int': 'int', 'pad_left': '#ljust', 'pad_right': '#rjust' }, 'Int': { 'to_float': 'float(%{self})' }, 'Float': { 'to_int': 'int(%{self})' }, 'Set': { '@equivalent': 'set', 'length': 'len', 'contains?': contains, 'union': to_op('|'), 'intersection': to_op('-'), 'present?': lambda receiver, _: receiver, 'empty?': lambda receiver, _: Node( 'unary_op', op='not', value=receiver, pseudo_type='Boolean') }, 'Tuple': { '@equivalent': 'tuple', 'length': 'len' }, 'Array': { '@equivalent': 'tuple', 'length': 'len' }, 'Regexp': { '@equivalent': '_sre.SRE_Pattern', 'match': '#match' }, 'RegexpMatch': { '@equivalent': '_sre.SRE_Match', 'group': '#group', 'has_match': lambda receiver, _: receiver } } functions = { 'global': { 'wat': lambda _: Node('block', block=[]), 'exit': lambda status, _: call('exit', [status]) }, 'io': { 'display': 'print', 'read': 'input', 'write_file': WriteFile, 'read_file': ReadFile }, 'system': { 'args': 'sys.argv!', 'arg_count': lambda _: call('len', [ attr(local('sys', 'Library'), 'argv', ['List', 'String']) ], 'Int'), 'index': lambda value, _: Node('index', sequence=attr(local('sys', 'Library'), 'argv', ['List', 'String']), index=value, pseudo_type='String') }, 'http': { 'get': 'requests.get', 'post': 'requests.post', }, 'math': { 'ln': 'math.log', 'log': 'math.log', 'tan': 'math.tan', 'sin': 'math.sin', 'cos': 'math.cos', 'pow': lambda left, right, pseudo_type: Node('binary_op', op='**', left=left, right=right, pseudo_type=pseudo_type) }, 'regexp': { 'compile': 're.compile', 'escape': 're.escape' } } dependencies = { 'Regexp': { '@all': 're' }, 'List': { 'reduce': 'functools' }, 'http': { '@all': 'requests' }, 'math': { '@all': 'math' }, 'regexp': { '@all': 're' }, 'system': { '@all': 'sys' } }