Export Router (#306)

* Merge branch 'master' into feature/abi (#284) * Move to pyteal as pt in ABI tests with concise prefix (#286) * ABI Strings (#278) * Move to pyteal as pt in #278 (#287) * Merge absolute imports into feature/abi (#288) * Remove temporary I252 ignore on pyteal.ast.abi (#290) * Fix abi import (#303) * Fix abi import * ignore flake8 * move router to ast Co-authored-by: Michael Diamant <[email protected]> Co-authored-by: Jason Paulos <[email protected]>
algorand · Apr 28, 2022 · a18c47d · a18c47d
1 parent 67831ff
commit a18c47d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/pyteal/__init__.pyi b/pyteal/__init__.pyi
@@ -140,6 +140,7 @@ __all__ = [
     "RETURN_METHOD_SELECTOR",
     "Reject",
     "Return",
+    "Router",
     "ScratchIndex",
     "ScratchLoad",
     "ScratchSlot",

diff --git a/pyteal/ast/__init__.py b/pyteal/ast/__init__.py
@@ -136,6 +136,8 @@
 from pyteal.ast.maybe import MaybeValue
 from pyteal.ast.multi import MultiValue
 
+from pyteal.ast.router import Router
+
 # abi
 import pyteal.ast.abi as abi  # noqa: I250
 
@@ -273,5 +275,6 @@
     "For",
     "Break",
     "Continue",
+    "Router",
     "abi",
 ]
diff --git a/pyteal/ast/abi/router.py → pyteal/ast/router.py b/pyteal/ast/abi/router.py → pyteal/ast/router.py
@@ -4,12 +4,12 @@
 from pyteal.config import METHOD_ARG_NUM_LIMIT
 from pyteal.errors import TealInputError
 from pyteal.types import TealType
+from pyteal.ast.subroutine import SubroutineFnWrapper
 from pyteal.ast.cond import Cond
 from pyteal.ast.expr import Expr
 from pyteal.ast.app import OnComplete, EnumInt
 from pyteal.ast.int import Int
 from pyteal.ast.seq import Seq
-from pyteal.ast.subroutine import SubroutineFnWrapper
 from pyteal.ast.methodsig import MethodSignature
 from pyteal.ast.naryexpr import And, Or
 from pyteal.ast.txn import Txn

diff --git a/pyteal/ast/abi/router_test.py → pyteal/ast/router_test.py b/pyteal/ast/abi/router_test.py → pyteal/ast/router_test.py