From 916b7a56cdebca4036a31ef1284190e7b2fc1ee8 Mon Sep 17 00:00:00 2001
From: Marc Beninca <git@marc.beninca.link>
Date: Wed, 5 Jun 2024 11:41:47 +0200
Subject: [PATCH] py/step

---
 cd.sh          | 13 +++++--------
 cd/__init__.py |  7 +++++++
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/cd.sh b/cd.sh
index b64ce20..b88b6fd 100644
--- a/cd.sh
+++ b/cd.sh
@@ -573,9 +573,13 @@ ${CD_PYTHON_PACKAGES}
 cd_write_python_module () {
     cd_step "Write Python module"
     cd_write "${CD_PYTHON_PACKAGES}/env.py" "\
+STEP = ${CD_STEP}
+
 $(cd_echo CD_OS_ID CD_OS_VERSION)
 
-STEP = ${CD_STEP}
+DOWN = \"${CD_DOWN}\"
+UP = \"${CD_UP}\"
+VERTICAL = \"${CD_BOX_VERTICAL}\"
 
 
 def open(*arguments):
@@ -590,13 +594,6 @@ def shut(*arguments):
 
 def split():
     print('${CD_SPLIT}', flush=True)
-
-
-def step(*arguments):
-    STEP += 1
-    print('${CD_DOWN}')
-    print('${CD_BOX_VERTICAL}', STEP, *arguments)
-    print('${CD_UP}', flush=True)
 "
 }
 
diff --git a/cd/__init__.py b/cd/__init__.py
index 279a18f..d1c1653 100644
--- a/cd/__init__.py
+++ b/cd/__init__.py
@@ -107,3 +107,10 @@ def set_ssh(*arguments):
     if ssh_hosts:
         fs.write(known, ssh_hosts)
         os.chmod(known, 0o400)
+
+
+def step(*arguments):
+    env.STEP += 1
+    print(env.DOWN)
+    print(env.VERTICAL, env.STEP, *arguments)
+    print(env.UP, flush=True)