def __render_datetime_field(buffer: CodeChunk):
buffer.write('')
buffer.write('from datetime import datetime')
buffer.write('from marshmallow import fields as marshmallow_fields')
buffer.write(
"DATETIME_FIELD = field(metadata={'dataclasses_json': {'encoder': datetime.isoformat, 'decoder': datetime.fromisoformat, 'mm_field': marshmallow_fields.DateTime(format='iso')}})"
)
buffer.write('')
def __render_enum(buffer: CodeChunk, enum: ParsedEnum):
with buffer.write_block(f'class {enum.name}(Enum):'):
for value_name, value in enum.values.items():
if isinstance(value, str):
value = f"'{value}'"
buffer.write(f'{value_name} = {value}')
buffer.write('')
def test_codegen_write_block(module_compiler):
gen = CodeChunk()
with gen.write_block('def {name}(a, b):', name='sum'):
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_write_block(module_compiler):
gen = CodeChunk()
with gen.write_block('def {name}(a, b):', name='sum'):
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def __render_enum(buffer: CodeChunk, enum: ParsedEnum):
with buffer.write_block(f'class {enum.name}(Enum):'):
for value_name, value in enum.values.items():
if isinstance(value, str):
value = f"'{value}'"
buffer.write(f'{value_name} = {value}')
buffer.write('')
def test_codegen_write_simple_strings(module_compiler):
gen = CodeChunk()
gen.write('def sum(a, b):')
gen.indent()
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_block(module_compiler):
gen = CodeChunk()
gen.write('def sum(a, b):')
with gen.block():
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_write_simple_strings(module_compiler):
gen = CodeChunk()
gen.write('def sum(a, b):')
gen.indent()
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_write_template_strings_kwargs(module_compiler):
gen = CodeChunk()
gen.write('def {method}(a, b):', method='sum')
gen.indent()
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_block(module_compiler):
gen = CodeChunk()
gen.write('def sum(a, b):')
with gen.block():
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_write_template_strings_kwargs(module_compiler):
gen = CodeChunk()
gen.write('def {method}(a, b):', method='sum')
gen.indent()
gen.write('return a + b')
code = str(gen)
m = module_compiler(code)
assert m.sum(2, 3) == 5
def test_codegen_write_lines(module_compiler):
lines = ['@staticmethod', 'def sum(a, b):' ' return a + b']
gen = CodeChunk()
gen.write('class Math:')
gen.indent()
gen.write_lines(lines)
code = str(gen)
m = module_compiler(code)
assert m.Math.sum(2, 3) == 5
def test_codegen_write_lines(module_compiler):
lines = [
'@staticmethod',
'def sum(a, b):'
' return a + b'
]
gen = CodeChunk()
gen.write('class Math:')
gen.indent()
gen.write_lines(lines)
code = str(gen)
m = module_compiler(code)
assert m.Math.sum(2, 3) == 5
def __render_field(parsed_query: ParsedQuery, buffer: CodeChunk, field: ParsedField):
enum_names = [e.name for e in parsed_query.enums]
is_enum = field.type in enum_names
suffix = ''
field_type = field.type
if is_enum:
suffix = f'= enum_field({field.type})'
if field.type == 'DateTime':
suffix = '= DATETIME_FIELD'
field_type = 'datetime'
if field.nullable:
suffix = f'= {field.default_value}'
buffer.write(f'{field.name}: {field_type} {suffix}')
def __render_field(parsed_query: ParsedQuery, buffer: CodeChunk,
field: ParsedField):
enum_names = [e.name for e in parsed_query.enums]
is_enum = field.type in enum_names
suffix = ''
field_type = field.type
if is_enum:
suffix = f'= enum_field({field.type})'
if field.type == 'DateTime':
suffix = '= DATETIME_FIELD'
field_type = 'datetime'
if field.nullable:
suffix = f'= {field.default_value}'
buffer.write(f'{field.name}: {field_type} {suffix}')
def __render_field(parsed_query: ParsedQuery, buffer: CodeChunk,
field: ParsedField):
enum_names = [e.name for e in parsed_query.enums]
is_enum = field.type in enum_names
suffix = ''
field_type = field.type
if is_enum:
suffix = f'= enum_field({field.type})'
if field.type == 'DateTime':
suffix = '= DATETIME_FIELD'
if (field.nullable):
suffix = "= field(default={default}, metadata={{'dataclasses_json': {{'encoder': datetime.isoformat, 'decoder': datetime.fromisoformat, 'mm_field': marshmallow_fields.DateTime(format='iso')}}}})".format(
default=field.default_value)
field_type = 'datetime'
elif field.nullable:
suffix = f'= {field.default_value}'
buffer.write(f'{field.name}: {field_type} {suffix}')
def __render_object(self, parsed_query: ParsedQuery, buffer: CodeChunk, obj: ParsedObject):
class_parents = '' if not obj.parents else f'({", ".join(obj.parents)})'
buffer.write('@dataclass_json')
buffer.write('@dataclass')
with buffer.write_block(f'class {obj.name}{class_parents}:'):
# render child objects
for child_object in obj.children:
self.__render_object(parsed_query, buffer, child_object)
# render fields
sorted_fields = sorted(obj.fields, key=lambda f: 1 if f.nullable else 0)
for field in sorted_fields:
self.__render_field(parsed_query, buffer, field)
# pass if not children or fields
if not (obj.children or obj.fields):
buffer.write('pass')
buffer.write('')
def __render_object(self, parsed_query: ParsedQuery, buffer: CodeChunk,
obj: ParsedObject):
class_parents = '' if not obj.parents else f'({", ".join(obj.parents)})'
buffer.write('@dataclass_json')
buffer.write('@dataclass')
with buffer.write_block(f'class {obj.name}{class_parents}:'):
# render child objects
for child_object in obj.children:
self.__render_object(parsed_query, buffer, child_object)
# render fields
sorted_fields = sorted(obj.fields,
key=lambda f: 1 if f.nullable else 0)
for field in sorted_fields:
self.__render_field(parsed_query, buffer, field)
# pass if not children or fields
if not (obj.children or obj.fields):
buffer.write('pass')
buffer.write('')
def __render_operation(self, parsed_query: ParsedQuery, buffer: CodeChunk,
parsed_op: ParsedOperation):
buffer.write('@dataclass_json')
buffer.write('@dataclass')
with buffer.write_block(f'class {parsed_op.name}:'):
buffer.write('__QUERY__ = """')
buffer.write(parsed_query.query)
buffer.write('"""')
buffer.write('')
# Render children
for child_object in parsed_op.children:
self.__render_object(parsed_query, buffer, child_object)
# operation fields
buffer.write('')
buffer.write(f'data: {parsed_op.name}Data = None')
buffer.write('errors: Any = None')
buffer.write('')
# Execution functions
if parsed_op.variables:
vars_args = ', '.join([
self.__render_variable_definition(var)
for var in parsed_op.variables
]) + ','
variables_dict = '{' + ', '.join(
f'"{var.name}": {var.name}'
for var in parsed_op.variables) + '}'
else:
vars_args = ''
variables_dict = 'None'
buffer.write('@classmethod')
with buffer.write_block(
f'def execute(cls, {vars_args} on_before_callback: Callable[[Mapping[str, str], Mapping[str, str]], None] = None):'
):
buffer.write(f'client = Client(\'{self.config.endpoint}\')')
buffer.write(f'variables = {variables_dict}')
buffer.write(
'response_text = client.call(cls.__QUERY__, variables=variables, on_before_callback=on_before_callback)'
)
buffer.write('return cls.from_json(response_text)')
buffer.write('')
buffer.write('@classmethod')
with buffer.write_block(
f'async def execute_async(cls, {vars_args} on_before_callback: Callable[[Mapping[str, str], Mapping[str, str]], None] = None):'
):
buffer.write(
f'client = AsyncIOClient(\'{self.config.endpoint}\')')
buffer.write(f'variables = {variables_dict}')
buffer.write(
f'response_text = await client.call(cls.__QUERY__, variables=variables, on_before_callback=on_before_callback)'
)
buffer.write(f'return cls.from_json(response_text)')
buffer.write('')
buffer.write('')
def __render_enum_field(buffer: CodeChunk):
with buffer.write_block('def enum_field(enum_type):'):
with buffer.write_block('def encode_enum(value):'):
buffer.write('return value.value')
buffer.write('')
with buffer.write_block('def decode_enum(t, value):'):
buffer.write('return t(value)')
buffer.write('')
buffer.write(
"return field(metadata={'dataclasses_json': {'encoder': encode_enum, 'decoder': partial(decode_enum, enum_type)}})"
)
buffer.write('')
def render(self, parsed_query: ParsedQuery):
# We sort fragment nodes to be first and operations to be last because of dependecies
buffer = CodeChunk()
buffer.write('# AUTOGENERATED file. Do not Change!')
buffer.write('from functools import partial')
buffer.write('from typing import Any, Callable, Mapping, List')
buffer.write('from enum import Enum')
buffer.write('from dataclasses import dataclass, field')
buffer.write('from dataclasses_json import dataclass_json')
buffer.write('from gql.clients import Client, AsyncIOClient')
buffer.write('')
if self.config.custom_header:
buffer.write_lines(self.config.custom_header.split('\n'))
buffer.write('')
self.__render_datetime_field(buffer)
# Enums
if parsed_query.enums:
self.__render_enum_field(buffer)
for enum in parsed_query.enums:
self.__render_enum(buffer, enum)
sorted_objects = sorted(parsed_query.objects, key=lambda obj: 1 if isinstance(obj, ParsedOperation) else 0)
for obj in sorted_objects:
if isinstance(obj, ParsedObject):
self.__render_object(parsed_query, buffer, obj)
elif isinstance(obj, ParsedOperation):
self.__render_operation(parsed_query, buffer, obj)
return str(buffer)
def render(self, parsed_query: ParsedQuery):
# We sort fragment nodes to be first and operations to be last because of dependecies
buffer = CodeChunk()
buffer.write('# AUTOGENERATED file. Do not Change!')
buffer.write('from functools import partial')
buffer.write('from typing import Any, Callable, Mapping, List')
buffer.write('from enum import Enum')
buffer.write('from dataclasses import dataclass, field')
buffer.write('from dataclasses_json import dataclass_json')
buffer.write('from gql.clients import Client, AsyncIOClient')
buffer.write('')
if self.config.custom_header:
buffer.write_lines(self.config.custom_header.split('\n'))
buffer.write('')
self.__render_datetime_field(buffer)
# Enums
if parsed_query.enums:
self.__render_enum_field(buffer)
for enum in parsed_query.enums:
self.__render_enum(buffer, enum)
sorted_objects = sorted(parsed_query.objects,
key=lambda obj: 1
if isinstance(obj, ParsedOperation) else 0)
for obj in sorted_objects:
if isinstance(obj, ParsedObject):
self.__render_object(parsed_query, buffer, obj)
elif isinstance(obj, ParsedOperation):
self.__render_operation(parsed_query, buffer, obj)
return str(buffer)
def __render_operation(self, parsed_query: ParsedQuery, buffer: CodeChunk, parsed_op: ParsedOperation):
buffer.write('@dataclass_json')
buffer.write('@dataclass')
with buffer.write_block(f'class {parsed_op.name}:'):
buffer.write('__QUERY__ = """')
buffer.write(parsed_query.query)
buffer.write('"""')
buffer.write('')
# Render children
for child_object in parsed_op.children:
self.__render_object(parsed_query, buffer, child_object)
# operation fields
buffer.write('')
buffer.write(f'data: {parsed_op.name}Data = None')
buffer.write('errors: Any = None')
buffer.write('')
# Execution functions
if parsed_op.variables:
vars_args = ', '.join([self.__render_variable_definition(var) for var in parsed_op.variables]) + ','
variables_dict = '{' + ', '.join(f'"{var.name}": {var.name}' for var in parsed_op.variables) + '}'
else:
vars_args = ''
variables_dict = 'None'
buffer.write('@classmethod')
with buffer.write_block(f'def execute(cls, {vars_args} on_before_callback: Callable[[Mapping[str, str], Mapping[str, str]], None] = None):'):
buffer.write(f'client = Client(\'{self.config.endpoint}\')')
buffer.write(f'variables = {variables_dict}')
buffer.write('response_text = client.call(cls.__QUERY__, variables=variables, on_before_callback=on_before_callback)')
buffer.write('return cls.from_json(response_text)')
buffer.write('')
buffer.write('@classmethod')
with buffer.write_block(f'async def execute_async(cls, {vars_args} on_before_callback: Callable[[Mapping[str, str], Mapping[str, str]], None] = None):'):
buffer.write(f'client = AsyncIOClient(\'{self.config.endpoint}\')')
buffer.write(f'variables = {variables_dict}')
buffer.write(f'response_text = await client.call(cls.__QUERY__, variables=variables, on_before_callback=on_before_callback)')
buffer.write(f'return cls.from_json(response_text)')
buffer.write('')
buffer.write('')
def __render_datetime_field(buffer: CodeChunk):
buffer.write('')
buffer.write('from datetime import datetime')
buffer.write('from marshmallow import fields as marshmallow_fields')
buffer.write("DATETIME_FIELD = field(metadata={'dataclasses_json': {'encoder': datetime.isoformat, 'decoder': datetime.fromisoformat, 'mm_field': marshmallow_fields.DateTime(format='iso')}})")
buffer.write('')
def __render_enum_field(buffer: CodeChunk):
with buffer.write_block('def enum_field(enum_type):'):
with buffer.write_block('def encode_enum(value):'):
buffer.write('return value.value')
buffer.write('')
with buffer.write_block('def decode_enum(t, value):'):
buffer.write('return t(value)')
buffer.write('')
buffer.write("return field(metadata={'dataclasses_json': {'encoder': encode_enum, 'decoder': partial(decode_enum, enum_type)}})")
buffer.write('')