From 4328d63e52b88e97108fdea3aa8ff27ff33063f1 Mon Sep 17 00:00:00 2001
From: Victor Smirnov <vityaman.dev@yandex.ru>
Date: Mon, 12 Feb 2024 11:49:42 +0300
Subject: [PATCH] #19 [tafka] transformed from imports to module imports

---
 sleepy/tafka/emit.py |  3 +--
 sleepy/tafka/text.py | 62 ++++++++++++++++----------------------------
 2 files changed, 23 insertions(+), 42 deletions(-)

diff --git a/sleepy/tafka/emit.py b/sleepy/tafka/emit.py
index 19ecf94..487378c 100644
--- a/sleepy/tafka/emit.py
+++ b/sleepy/tafka/emit.py
@@ -3,8 +3,7 @@
 import sleepy.tafka.representation as taf
 from sleepy import program
 from sleepy.core import MetaTable
-from sleepy.program import ProgramVisitor
-from sleepy.program.unit import ProgramUnit
+from sleepy.program import ProgramUnit, ProgramVisitor
 
 
 class TafkaEmitVisitor(ProgramVisitor[None]):
diff --git a/sleepy/tafka/text.py b/sleepy/tafka/text.py
index e466508..5cf5c52 100644
--- a/sleepy/tafka/text.py
+++ b/sleepy/tafka/text.py
@@ -1,26 +1,8 @@
 from io import StringIO
 from typing import override
 
-from .representation import (
-    And,
-    Block,
-    Conditional,
-    Copy,
-    Div,
-    Eq,
-    Goto,
-    Invokation,
-    Load,
-    Lt,
-    Mul,
-    Or,
-    Procedure,
-    Rem,
-    Return,
-    Statement,
-    Sum,
-    Var,
-)
+import sleepy.tafka.representation as taf
+
 from .walker import TafkaWalker
 
 
@@ -36,7 +18,7 @@ def writeln(self, line: str) -> None:
         self.write(f"{'  ' * self.indent}{line}\n")
 
     @override
-    def enter_procedure(self, procedure: Procedure) -> None:
+    def enter_procedure(self, procedure: taf.Procedure) -> None:
         self.writeln(
             f"procedure @{procedure.name}("
             f"{', '.join(map(repr, procedure.parameters))}) "
@@ -44,36 +26,36 @@ def enter_procedure(self, procedure: Procedure) -> None:
         )
 
     @override
-    def exit_procedure(self, procedure: Procedure) -> None:
+    def exit_procedure(self, procedure: taf.Procedure) -> None:
         self.writeln("}")
 
     @override
-    def enter_block(self, block: Block) -> None:
+    def enter_block(self, block: taf.Block) -> None:
         self.writeln(f"{block.label.name}:")
         self.indent += 1
 
     @override
-    def exit_block(self, block: Block) -> None:
+    def exit_block(self, block: taf.Block) -> None:
         self.indent -= 1
 
     @override
-    def enter_statement(self, statement: Statement) -> None:
+    def enter_statement(self, statement: taf.Statement) -> None:
         pass
 
     @override
-    def exit_statement(self, statement: Statement) -> None:
+    def exit_statement(self, statement: taf.Statement) -> None:
         pass
 
     @override
-    def on_return(self, ret: Return) -> None:
+    def on_return(self, ret: taf.Return) -> None:
         self.writeln(f"return {ret.value}")
 
     @override
-    def on_goto(self, goto: Goto) -> None:
+    def on_goto(self, goto: taf.Goto) -> None:
         self.writeln(f"goto {goto.block.label.name}")
 
     @override
-    def on_conditional(self, conditional: Conditional) -> None:
+    def on_conditional(self, conditional: taf.Conditional) -> None:
         self.writeln(
             f"if {conditional.condition!r} "
             f"then {conditional.then_branch.label.name} "
@@ -82,48 +64,48 @@ def on_conditional(self, conditional: Conditional) -> None:
         )
 
     @override
-    def on_invokation(self, target: Var, source: Invokation) -> None:
+    def on_invokation(self, target: taf.Var, source: taf.Invokation) -> None:
         self.writeln(
             f"{target!r} = invoke @{source.closure.name}"
             f" {', '.join(map(repr, source.args))}",
         )
 
     @override
-    def on_load(self, target: Var, source: Load) -> None:
+    def on_load(self, target: taf.Var, source: taf.Load) -> None:
         self.writeln(f"{target!r} = load {source.constant!r}")
 
     @override
-    def on_copy(self, target: Var, source: Copy) -> None:
+    def on_copy(self, target: taf.Var, source: taf.Copy) -> None:
         self.writeln(f"{target!r} = copy {source.argument!r}")
 
     @override
-    def on_sum(self, target: Var, source: Sum) -> None:
+    def on_sum(self, target: taf.Var, source: taf.Sum) -> None:
         self.writeln(f"{target!r} = sum {source.left!r}, {source.right!r}")
 
     @override
-    def on_mul(self, target: Var, source: Mul) -> None:
+    def on_mul(self, target: taf.Var, source: taf.Mul) -> None:
         self.writeln(f"{target!r} = mul {source.left!r}, {source.right!r}")
 
     @override
-    def on_div(self, target: Var, source: Div) -> None:
+    def on_div(self, target: taf.Var, source: taf.Div) -> None:
         self.writeln(f"{target!r} = div {source.left!r}, {source.right!r}")
 
     @override
-    def on_rem(self, target: Var, source: Rem) -> None:
+    def on_rem(self, target: taf.Var, source: taf.Rem) -> None:
         self.writeln(f"{target!r} = rem {source.left!r}, {source.right!r}")
 
     @override
-    def on_eq(self, target: Var, source: Eq) -> None:
+    def on_eq(self, target: taf.Var, source: taf.Eq) -> None:
         self.writeln(f"{target!r} = eq {source.left!r}, {source.right!r}")
 
     @override
-    def on_lt(self, target: Var, source: Lt) -> None:
+    def on_lt(self, target: taf.Var, source: taf.Lt) -> None:
         self.writeln(f"{target!r} = lt {source.left!r}, {source.right!r}")
 
     @override
-    def on_and(self, target: Var, source: And) -> None:
+    def on_and(self, target: taf.Var, source: taf.And) -> None:
         self.writeln(f"{target!r} = and {source.left!r}, {source.right!r}")
 
     @override
-    def on_or(self, target: Var, source: Or) -> None:
+    def on_or(self, target: taf.Var, source: taf.Or) -> None:
         self.writeln(f"{target!r} = or {source.left!r}, {source.right!r}")