black + perf optimizations (#21)

acryldata · Jun 12, 2024 · e104997 · e104997
1 parent 932288c
commit e104997
Show file tree

Hide file tree

Showing 10 changed files with 758 additions and 418 deletions.
diff --git a/avrogen/__init__.py b/avrogen/__init__.py
@@ -2,6 +2,10 @@
 from .schema import write_schema_files
 from .protocol import generate_protocol
 from .protocol import write_protocol_files
-__all__ = ['generate_schema', 'generate_protocol', 'write_schema_files', 'write_protocol_files']
-
 
+__all__ = [
+    "generate_schema",
+    "generate_protocol",
+    "write_schema_files",
+    "write_protocol_files",
+]
diff --git a/avrogen/__main__.py b/avrogen/__main__.py
@@ -10,11 +10,11 @@ def main():
     if "-o" in argv:
         avdl, _, output = argv[1:]
     else:
-        avdl, output = argv[1], './'
-    write_protocol_files(open(avdl, 'r').read(), output)
+        avdl, output = argv[1], "./"
+    write_protocol_files(open(avdl, "r").read(), output)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     try:
         exit(main())
     except Exception:

diff --git a/avrogen/avrojson.py b/avrogen/avrojson.py