From 06064061d66a2678699454e8a765e9712f50847d Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Thu, 11 Oct 2018 23:53:09 +0300 Subject: [PATCH] dagascii: import asciimatics only when needed Related to #1210 Signed-off-by: Ruslan Kuprieiev --- dvc/dagascii.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dvc/dagascii.py b/dvc/dagascii.py index 4d6a18b68..60eb27e17 100644 --- a/dvc/dagascii.py +++ b/dvc/dagascii.py @@ -2,9 +2,6 @@ import sys import math import select -from asciimatics.screen import Screen -from asciimatics.event import KeyboardEvent - from grandalf.graphs import Vertex, Edge, Graph from grandalf.layouts import SugiyamaLayout from grandalf.routing import route_with_lines, EdgeViewer @@ -22,12 +19,15 @@ class AsciiCanvas(object): def draw(self): if sys.stdout.isatty(): # pragma: no cover + from asciimatics.screen import Screen Screen.wrapper(self._do_draw) else: for line in self.canvas: print(''.join(line)) def _do_draw(self, screen): # pragma: no cover + from asciimatics.event import KeyboardEvent + offset_x = 0 offset_y = 0 smaxrow, smaxcol = screen.dimensions