summaryrefslogtreecommitdiff
path: root/py/makemoduledefs.py
diff options
context:
space:
mode:
authormicroDev <70126934+microDev1@users.noreply.github.com>2021-03-15 19:27:36 +0530
committermicroDev <70126934+microDev1@users.noreply.github.com>2021-03-15 19:27:36 +0530
commita52eb88031620a81521b937f2a0651dbac2bb350 (patch)
tree017cbf8f686b252241182ef61ce48e8457aa1577 /py/makemoduledefs.py
parent090b6ba42fcdfbb16844f77892087f91aa6ea201 (diff)
run code formatting script
Diffstat (limited to 'py/makemoduledefs.py')
-rw-r--r--py/makemoduledefs.py48
1 files changed, 25 insertions, 23 deletions
diff --git a/py/makemoduledefs.py b/py/makemoduledefs.py
index 18d327f00..774560713 100644
--- a/py/makemoduledefs.py
+++ b/py/makemoduledefs.py
@@ -12,14 +12,11 @@ import os
import argparse
-pattern = re.compile(
- r"[\n;]\s*MP_REGISTER_MODULE\((.*?),\s*(.*?),\s*(.*?)\);",
- flags=re.DOTALL
-)
+pattern = re.compile(r"[\n;]\s*MP_REGISTER_MODULE\((.*?),\s*(.*?),\s*(.*?)\);", flags=re.DOTALL)
def find_c_file(obj_file, vpath):
- """ Search vpaths for the c file that matches the provided object_file.
+ """Search vpaths for the c file that matches the provided object_file.
:param str obj_file: object file to find the matching c file for
:param List[str] vpath: List of base paths, similar to gcc vpath
@@ -27,7 +24,7 @@ def find_c_file(obj_file, vpath):
"""
c_file = None
relative_c_file = os.path.splitext(obj_file)[0] + ".c"
- relative_c_file = relative_c_file.lstrip('/\\')
+ relative_c_file = relative_c_file.lstrip("/\\")
for p in vpath:
possible_c_file = os.path.join(p, relative_c_file)
if os.path.exists(possible_c_file):
@@ -38,7 +35,7 @@ def find_c_file(obj_file, vpath):
def find_module_registrations(c_file):
- """ Find any MP_REGISTER_MODULE definitions in the provided c file.
+ """Find any MP_REGISTER_MODULE definitions in the provided c file.
:param str c_file: path to c file to check
:return: List[(module_name, obj_module, enabled_define)]
@@ -54,7 +51,7 @@ def find_module_registrations(c_file):
def generate_module_table_header(modules):
- """ Generate header with module table entries for builtin modules.
+ """Generate header with module table entries for builtin modules.
:param List[(module_name, obj_module, enabled_define)] modules: module defs
:return: None
@@ -66,15 +63,20 @@ def generate_module_table_header(modules):
for module_name, obj_module, enabled_define in modules:
mod_def = "MODULE_DEF_{}".format(module_name.upper())
mod_defs.append(mod_def)
- print((
- "#if ({enabled_define})\n"
- " extern const struct _mp_obj_module_t {obj_module};\n"
- " #define {mod_def} {{ MP_ROM_QSTR({module_name}), MP_ROM_PTR(&{obj_module}) }},\n"
- "#else\n"
- " #define {mod_def}\n"
- "#endif\n"
- ).format(module_name=module_name, obj_module=obj_module,
- enabled_define=enabled_define, mod_def=mod_def)
+ print(
+ (
+ "#if ({enabled_define})\n"
+ " extern const struct _mp_obj_module_t {obj_module};\n"
+ " #define {mod_def} {{ MP_ROM_QSTR({module_name}), MP_ROM_PTR(&{obj_module}) }},\n"
+ "#else\n"
+ " #define {mod_def}\n"
+ "#endif\n"
+ ).format(
+ module_name=module_name,
+ obj_module=obj_module,
+ enabled_define=enabled_define,
+ mod_def=mod_def,
+ )
)
print("\n#define MICROPY_REGISTERED_MODULES \\")
@@ -87,13 +89,13 @@ def generate_module_table_header(modules):
def main():
parser = argparse.ArgumentParser()
- parser.add_argument("--vpath", default=".",
- help="comma separated list of folders to search for c files in")
- parser.add_argument("files", nargs="*",
- help="list of c files to search")
+ parser.add_argument(
+ "--vpath", default=".", help="comma separated list of folders to search for c files in"
+ )
+ parser.add_argument("files", nargs="*", help="list of c files to search")
args = parser.parse_args()
- vpath = [p.strip() for p in args.vpath.split(',')]
+ vpath = [p.strip() for p in args.vpath.split(",")]
modules = set()
for obj_file in args.files:
@@ -103,5 +105,5 @@ def main():
generate_module_table_header(sorted(modules))
-if __name__ == '__main__':
+if __name__ == "__main__":
main()