work on Opposite[], DetectPose[], ComparePose[]

Author: capjamesg

visionscript/error_handling.py

@@ -0,0 +1,82 @@
+from spellchecker import SpellChecker
+
+from visionscript.usage import (
+    language_grammar_reference,
+    lowercase_language_grammar_reference,
+)
+
+spell = SpellChecker()
+
+
+def handle_unexpected_characters(e, code, interactive=False):
+    # if line doesn't end with ], add it
+    if not code.strip().endswith("]"):
+        code += "]"
+
+        return
+
+    # if space between statement and [, remove it
+    # get position of [
+    position = code.find("[")
+
+    if code[position - 1] == " ":
+        code = code[: position - 1] + code[position:]
+
+        return
+
+    # replace all “ with "
+    code = code.replace("“", '"')
+    code = code.replace("”", '"')
+
+    # raise error if character not in grammar
+    if e.char not in ["[", "]", "'", '"', ",", " ", '"', '"', "\n", "\t", "\r"]:
+        print(f"Syntax error on line {e.line}, column {e.column}.")
+        print(f"Unexpected character: {e.char!r}")
+        exit(1)
+
+    # raise error if class doesn't exist
+    line = e.line
+    column = e.column
+
+    # check if function name in grammar
+    function_name = code.strip().split("\n")[line - 1].split("[")[0].strip()
+
+    language_grammar_reference_keys = language_grammar_reference.keys()
+
+    if function_name in language_grammar_reference_keys:
+        print(f"Syntax error on line {line}, column {column}.")
+        print(f"Unexpected character: {e.char!r}")
+        exit(1)
+
+    spell.known(lowercase_language_grammar_reference)
+    spell.word_frequency.load_words(lowercase_language_grammar_reference)
+
+    alternatives = spell.candidates(function_name)
+
+    if len(alternatives) == 0:
+        print(f"Function {function_name} does not exist.")
+        exit(1)
+
+    print(f"Function '{function_name}' does not exist. Did you mean one of these?")
+    print("-" * 10)
+
+    for item in list(alternatives):
+        if item.lower() in lowercase_language_grammar_reference:
+            print(
+                list(language_grammar_reference.keys())[
+                    lowercase_language_grammar_reference.index(item.lower())
+                ]
+            )
+
+    if interactive is False:
+        exit(1)
+
+
+def handle_unexpected_token(e, interactive=False):
+    line = e.line
+    column = e.column
+
+    print(f"Syntax error on line {line}, column {column}.")
+    print(f"Unexpected token: {e.token!r}")
+    if interactive is False:
+        exit(1)

visionscript/grammar.py

@@ -1,7 +1,7 @@
 grammar = """
 start: (expr | EOL)*
 
-expr: (set | var | make | in | if | train | label | detect | countinregion | help | get | exit | read | compare | count | cutout | show | size | caption | say | save | load | use | replace | var | classify | segment | comment | contains | if | else | end | run | isita | find | describe | import | rotate | getcolours | getcolors | get_text | greyscale | select | paste | pasterandom | resize | blur | literal | setbrightness | search | similarity | readqr | reset | negate | BOOL | INT | equality | not_equality | input | deploy | getedges | setconfidence | setregion | filterbyclass | crop | shuffle | grid | run | camera | showtext | getfps | gt | lt | expr | increment | decrement | track | getdistinctscenes | getuniqueappearances | usecamera | breakpoint | profile | math | first | last | is | merge | remove | web | wait | associative_array | list | STRING | EOL)
+expr: (set | var | make | in | if | train | label | detect | countinregion | help | get | exit | read | compare | count | cutout | show | size | caption | say | save | load | use | replace | var | classify | segment | comment | contains | if | else | end | run | isita | find | describe | import | rotate | getcolours | getcolors | get_text | greyscale | select | paste | pasterandom | resize | blur | literal | setbrightness | search | similarity | readqr | reset | negate | BOOL | INT | equality | not_equality | input | deploy | getedges | setconfidence | setregion | filterbyclass | crop | shuffle | grid | run | camera | showtext | getfps | gt | lt | expr | increment | decrement | track | getdistinctscenes | getuniqueappearances | usecamera | breakpoint | profile | math | first | last | is | merge | remove | web | wait | opposite | detectpose | comparepose | associative_array | list | STRING | EOL)
 classify: "Classify" "[" STRING ("," STRING)* "]"
 merge: "Merge" "[" (variable | list | associative_array) ("," (variable | list | associative_array))* "]"
 var: variable "=" (expr | STRING | INT)
@@ -17,6 +17,9 @@
 get_text: "GetText[]"
 camera: "Camera[]"
 greyscale: "Greyscale" "[]"
+opposite: "Opposite" "[" (BOOL | expr) "]"
+detectpose: "DetectPose[]"
+comparepose: "ComparePose" "[" (expr) "," (expr) "]" | "ComparePose[]"
 showtext: "ShowText[]" | "ShowText" "[" (expr | STRING) "]"
 search: "Search" "[" STRING "]"
 deploy: "Deploy" "[" STRING "]"