IQ.Lvbs/pycapnp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Applying black formatting

Browse Source 
- Fixing flake8 configuration to agree with black
- Adding black validation check to github actions
This commit is contained in:
Jacob Alexander
2021-10-01 11:00:22 -07:00
parent 5dade41aeb
commit 6e7fffd7de
51 changed files with 2536 additions and 1837 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
capnp/_gen.py
View File

@@ -8,59 +8,73 @@ import schema_capnp
def find_type(code, id):
for node in code['nodes']:
if node['id'] == id:
for node in code["nodes"]:
if node["id"] == id:
return node
return None
def main():
env = Environment(loader=PackageLoader('capnp', 'templates'))
env.filters['format_name'] = lambda name: name[name.find(':') + 1:]
env = Environment(loader=PackageLoader("capnp", "templates"))
env.filters["format_name"] = lambda name: name[name.find(":") + 1 :]
code = schema_capnp.CodeGeneratorRequest.read(sys.stdin)
code = code.to_dict()
code['nodes'] = [node for node in code['nodes'] if 'struct' in node and node['scopeId'] != 0]
for node in code['nodes']:
displayName = node['displayName']
parent, path = displayName.split(':')
node['module_path'] = parent.replace('.', '_') + '.' + '.'.join([x[0].upper() + x[1:] for x in path.split('.')])
node['module_name'] = path.replace('.', '_')
node['c_module_path'] = '::'.join([x[0].upper() + x[1:] for x in path.split('.')])
node['schema'] = '_{}_Schema'.format(node['module_name'])
code["nodes"] = [
node for node in code["nodes"] if "struct" in node and node["scopeId"] != 0
]
for node in code["nodes"]:
displayName = node["displayName"]
parent, path = displayName.split(":")
node["module_path"] = (
parent.replace(".", "_")
+ "."
+ ".".join([x[0].upper() + x[1:] for x in path.split(".")])
)
node["module_name"] = path.replace(".", "_")
node["c_module_path"] = "::".join(
[x[0].upper() + x[1:] for x in path.split(".")]
)
node["schema"] = "_{}_Schema".format(node["module_name"])
is_union = False
for field in node['struct']['fields']:
if field['discriminantValue'] != 65535:
for field in node["struct"]["fields"]:
if field["discriminantValue"] != 65535:
is_union = True
field['c_name'] = field['name'][0].upper() + field['name'][1:]
if 'slot' in field:
field['type'] = list(field['slot']['type'].keys())[0]
if not isinstance(field['slot']['type'][field['type']], dict):
field["c_name"] = field["name"][0].upper() + field["name"][1:]
if "slot" in field:
field["type"] = list(field["slot"]["type"].keys())[0]
if not isinstance(field["slot"]["type"][field["type"]], dict):
continue
sub_type = field['slot']['type'][field['type']].get('typeId', None)
sub_type = field["slot"]["type"][field["type"]].get("typeId", None)
if sub_type:
field['sub_type'] = find_type(code, sub_type)
sub_type = field['slot']['type'][field['type']].get('elementType', None)
field["sub_type"] = find_type(code, sub_type)
sub_type = field["slot"]["type"][field["type"]].get("elementType", None)
if sub_type:
field['sub_type'] = sub_type
field["sub_type"] = sub_type
else:
field['type'] = find_type(code, field['group']['typeId'])
node['is_union'] = is_union
field["type"] = find_type(code, field["group"]["typeId"])
node["is_union"] = is_union
include_dir = os.path.abspath(os.path.join(os.path.dirname(capnp.__file__), '..'))
module = env.get_template('module.pyx')
include_dir = os.path.abspath(os.path.join(os.path.dirname(capnp.__file__), ".."))
module = env.get_template("module.pyx")
for f in code['requestedFiles']:
filename = f['filename'].replace('.', '_') + '_cython.pyx'
for f in code["requestedFiles"]:
filename = f["filename"].replace(".", "_") + "_cython.pyx"
file_code = dict(code)
file_code['nodes'] = [node for node in file_code['nodes'] if node['displayName'].startswith(f['filename'])]
with open(filename, 'w') as out:
file_code["nodes"] = [
node
for node in file_code["nodes"]
if node["displayName"].startswith(f["filename"])
]
with open(filename, "w") as out:
out.write(module.render(code=file_code, file=f, include_dir=include_dir))
setup = env.get_template('setup.py.tmpl')
with open('setup_capnp.py', 'w') as out:
setup = env.get_template("setup.py.tmpl")
with open("setup_capnp.py", "w") as out:
out.write(setup.render(code=code))
print('You now need to build the cython module by running `python setup_capnp.py build_ext --inplace`.')
print(
"You now need to build the cython module by running `python setup_capnp.py build_ext --inplace`."
)
print()