Merge branch 'main' into fix/cmgrp-command

ansys · Jun 29, 2023 · 03d960f · 03d960f
2 parents 3757498 + 4e5e387
commit 03d960f
Show file tree

Hide file tree

Showing 7 changed files with 287 additions and 144 deletions.
diff --git a/src/ansys/mapdl/core/common_grpc.py b/src/ansys/mapdl/core/common_grpc.py
@@ -1,4 +1,6 @@
 """Common gRPC functions"""
+from typing import List, Literal, get_args
+
 import numpy as np
 
 from ansys.mapdl.core.errors import MapdlConnectionError
@@ -7,7 +9,6 @@
 DEFAULT_CHUNKSIZE = 256 * 1024  # 256 kB
 DEFAULT_FILE_CHUNK_SIZE = 1024 * 1024  # 1MB
 
-
 ANSYS_VALUE_TYPE = {
     0: None,  # UNKNOWN
     1: np.int32,  # INTEGER
@@ -21,7 +22,18 @@
 }
 
 
-VGET_ENTITY_TYPES = [
+VGET_ENTITY_TYPES: List[str] = [
+    "NODE",
+    "ELEM",
+    "KP",
+    "LINE",
+    "AREA",
+    "VOLU",
+    "CDSY",
+    "RCON",
+    "TLAB",
+]
+VGET_ENTITY_TYPES_TYPING = Literal[
     "NODE",
     "ELEM",
     "KP",
@@ -32,6 +44,9 @@
     "RCON",
     "TLAB",
 ]
+
+VGET_ENTITY_TYPES: List[str] = list(get_args(VGET_ENTITY_TYPES_TYPING))
+
 STRESS_TYPES = ["X", "Y", "Z", "XY", "YZ", "XZ", "1", "2", "3", "INT", "EQV"]
 COMP_TYPE = ["X", "Y", "Z", "SUM"]
 VGET_NODE_ENTITY_TYPES = {
@@ -66,7 +81,7 @@ def __init__(self, msg=""):
         RuntimeError.__init__(self, msg)
 
 
-def check_vget_input(entity, item, itnum):
+def check_vget_input(entity: str, item: str, itnum: str) -> str:
     """Verify that entity and item for VGET are valid.
 
     Raises a ``ValueError`` when invalid.