Merge branch 'matlab2numpy' into mh_python

mat2py · Feb 9, 2022 · 4b4dbcc · 4b4dbcc
2 parents 271a32c + a556eb1
commit 4b4dbcc
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -10,6 +10,8 @@ We hope to merge this utility back to `miss_hit` after it is stable.
 
 ## Usage
 
+Try it online [here](https://translate.mat2py.org/).
+
 ```python
 # must install 
 python3 -m pip install mh-python

diff --git a/mh_python/__init__.py b/mh_python/__init__.py
@@ -210,7 +210,15 @@ def reference_visitor(self, node: Reference, n_parent, relation):
         self[node] = f"{self.pop(node.n_ident)}{bra}{args}{ket}"
 
     def identifier_visitor(self, node: Identifier, n_parent, relation):
-        self[node] = node.t_ident.value
+        value = node.t_ident.value
+
+        # Python and Matlab keywords set are different. Luckily, Matlab forbids identifier start with `_`.
+        prefix = '_' if value in (
+            'not', 'is', 'and', 'or', 'in', 'del',  # python keyword can not be overwritten
+            'all', 'any',  # python build-in names do not suggest overwriting
+            'I', 'M', 'C',  # mat2py keywords
+        ) else ''
+        self[node] = f'{prefix}{value}'
 
     def number_literal_visitor(self, node: Number_Literal, n_parent, relation):
         self[node] = node.t_value.value.replace("i", "j")

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="mh_python",
-    version="0.0.5",
+    version="0.0.6",
     author="Chaoqing Wang",
     author_email="[email protected]",
     description="Matlab to Python/Numpy translator",