StaSH - is it possible to get color output from a script to the StaSH console?
I'm trying to see if there is a way to get colorized output from a script I'm running.
In regular Python, I get color, as I'm using escape codes to the terminal. In StaSH, I obviously don't get that (and thankfully, I don't see the escape codes either, so that's good).
Is there a way to do this programmatically from Python, if I know I'm working with StaSH?
Any help or pointers would be appreciated.
That seems to work now. At least I can get the _stash variable and it's defined.
Now, on to actually getting it to do what I'd like!
Thanks for the help!
I'd say the difference is mainly a "bug" in bin/python.py, in that it does not populate the namespace. You can compare the code by looking at system/shruntime.py, and bin/python.py.
So, another question on this -- it seems that there is something not quite right.
Looking at the documentation, it states:
When setting this attribute, you can pass a string (CSS color name or hex, e.g. 'red' or '#ff0000'), a tuple (e.g. (1.0, 0.0, 0.0, 0.5) for half-transparent red), or a number (e.g. 0.5 for 50% gray).
See the code below:
print "text_color test" print _stash.text_color("red text - using color 'red'", 'red') print _stash.text_color("red text - using #ff0000", "#ff0000") print _stash.text_color("red text - using (1.0, 0.0, 0.0, 1.0)") print _stash.text_color("grey text - using 0.75") print "\n\n" ''' output is: text_color test red text - using color 'red' -- (this is actually red in output!) red text - using #ff0000 red text - using (1.0, 0.0, 0.0, 1.0) grey text - using 0.75 '''
As noted with a comment, the only text that actually prints out red is when I'm using the 'red' color to text_color. Also, the last line (just a number -- 0.75) doesn't do anything except output standard white text.
Is this expected? Should I be able to pass in the other ways to print red of a shade?
Interestingly, the editor will correctly interpret the color values correctly by bringing up a 'popup' of the correct color. So, needless to say, I'm a little confused.
As an aside, does anyone know what color values are actually valid?
I have looked at the code, but haven't been able to find where the list of colors are (if there is such a thing). I've tried a bunch but only found these to actually give me colored output:
Thanks for any help or direction on this!
You might look at how latte does colored text in StaSH. Perhaps you need to turn the color off before applying a new color.
I think you might be confusing pythonista docs with stash docs?
Anyway, in stash, stash.text_color calls stash.style_text, which looks up colors via
graphics, which is an imported alias of shcommon.Graphics. If you open up shcommon.py, you will see the list of colors
30: "black", 31: "red", 32: "green", 33: "brown", 34: "blue", 35: "magenta", 36: "cyan", 37: "white", 39: "default", # white. 50: "gray", 51: "yellow", 52: "smoke",
These are the standard ANSI color code definitions -- stash supports a reasonable approximation of an ANSI terminal (via pyte).
The stash terminal actually has another text_color, which takes a tuple... but you would need to go a little lower level. For basic work, stick with these.
Thanks ccc! I'll give that a look. I was using the terminal escape codes for a bash script, but they don't work in StaSh. A different way sounds fun!
Thanks JonB! I certainly got those from the Pythonista docs. Was under the impression that was how StaSh was doing it. I didn't go down the code chain as far as I should have.
Thanks for the color references. That will work for now. I really didn't relish working through the couple hundred different color names I found on the web for CSS colors. You saved me some work! Thanks!
Good to know that there is another way to set text with a tuple. I might give that a look. With what you gave me though, probably won't need to do anything with that for awhile though!
Thanks again for the help!
Or you could just use base escape codes. That's what I use since they're cleaner and more dynamic:
class stashansi: fore_red = "\x9b31m" fore_blue = "\x9b34m" fore_end = "\x9b39m" back_red = "\x9b41m" back_blue = "\x9b44m" back_end = "\x9b49m" bold = "\x9b1m" underline = "\x9b4m" all_end = "\9x0m" print(stashansi.back_blue + stashansi.fore_red + "Hello world" + back_end + "This is just red with no blue background" + fore_end + "Now it's all just normal text...")
Be aware that whenever you print something with ANSI escape codes, you MUST use the ender escape codes to change the colors back to the original state, or else if your program ends, the colors will continue to be the same even outside of your program!
Thanks AtomBombed! That is very similar to what I was using with standard sh/bash escape codes.
I might update my code to use this, but it's working well right now, so ...
It's good to have the escape codes in one place though!
I forgot and didn't mention -- I went the other way! I had this working on Mac terminal with colors and wanted it to work in StaSh. Now, it selects things dynamically and should work on either without any changes!
You mentioned you went the other way - just wondering what this means, can you share an example that works interchangeably with StaSh and macOS terminal?
I am trying to accomplish a similar goal and any pointers would be greatly appreciated :-)
I made the classes (shown below) to handle the color. I had the TerminalColors class already written, so didn't want to change the interface to handle both cases, so I created the getTerminalColorClass() function to key off of the existence (or not) of the _stash variable that can give access to the text_xxx functions.
There is certainly some refactoring that could be done -- just haven't gotten to it yet. If I used the ANSI codes, as suggested, then it's likely all I would need to change are the actual definitions of the colors. I'd probably use some sort of enum that is defined differently like the getTerminalColorClass() function.
def getTerminalColorClass(): if _stash != None: return StashTerminalColors() return TerminalColors() class TerminalColors(object): """ here is a good url for the different colors and things: http://blog.taylormcgann.com/tag/prompt-color/ """ HEADER = '\033[95m' BLUE = '\033[94m' DARK_BLUE = '\033[34m' GREEN = '\033[92m' DARK_GREEN = '\033[32m' YELLOW = '\033[93m' DARK_YELLOW = '\033[33m' RED = '\033[91m' DARK_RED = '\033[31m' BOLD = '\033[1m' UNDERLINE = '\033[4m' CYAN = '\033[96m' DARK_CYAN = '\033[36m' WHITE = '\033[97m' PURPLE = '\033[95m' DARK_PURPLE = '\033[35m' END = '\033[0m' def __init__(self): pass def simple_print(self, color, str_msg): print( "%s%s" % (self.get_color_string(color, str_msg), self.END)) def complex_print(self, color, str_format, data): str_msg = str_format % data self.simple_print( color, str_msg ) def multi_color_print(self, color_str_tuple_list): str_msg = "" for clr, str in color_str_tuple_list: str_msg += self.get_color_string(clr, str) self.println(str_msg) def get_color_string(self, color, str_msg): return "%s%s" % (color, str_msg) def println(self, str_msg): print( "%s%s" % (str_msg, self.END) ) class StashTerminalColors (object): # from shcommon.py in stash code HEADER = 'UNDERLINE' BLACK = 'black' RED = 'red' GREEN = 'green' BROWN = 'brown' BLUE = 'blue' MAGENTA = 'magenta' PURPLE = 'magenta' CYAN = 'cyan' WHITE = 'white' GRAY = 'gray' YELLOW = 'yellow' SMOKE = 'smoke' DEFAULT = 'white' STRIKE = 'STRIKE' BOLD = 'BOLD' UNDERLINE = 'UNDERLINE' BOLD_ITALIC = 'BOLD_ITALIC' ITALIC = 'ITALIC' END = 'END' def __init__(self): self._stash = _stash def simple_print(self, color, str_msg): print("%s" % self.get_color_string(color, str_msg)) def get_color_string(self, color, str_msg): if color == self.BOLD: prn_str = _stash.text_bold(str_msg) elif color == self.UNDERLINE: prn_str = _stash.text_underline(str_msg) elif color == self.BOLD_ITALIC: prn_str = _stash.text_bold_italic(str_msg) elif color == self.ITALIC: prn_str = _stash.text_italic(str_msg) elif color == self.STRIKE: prn_str = _stash.text_strikethrough(str_msg) else: prn_str = _stash.text_color(str_msg, color) return prn_str def multi_color_print(self, color_str_tuple_list): str_msg = "" for clr, str in color_str_tuple_list: str_msg += self.get_color_string(clr, str) print(str_msg)
The class is pretty simple to use (written explicitly for my use case, so ... YMMV). I'll give examples of usage for both below.
Here are some examples of using simple_print():
tc = getTerminalColorClass() tc.simple_print(tc.BLUE, "this should be BLUE") print("hopefully, this is NOT blue") tc.simple_print(tc.DEFAULT, "this is the DEFAULT color (white)") tc.simple_print(tc.BROWN, "this is BROWN text") tc.simple_print(tc.BOLD, "this should be BOLD") tc.simple_print(tc.UNDERLINE, "this should be UNDERLINE") tc.simple_print(tc.STRIKE, "this should be STRIKETHROUGH") tc.simple_print(tc.ITALIC, "this should be ITALIC") tc.simple_print(tc.BOLD_ITALIC, "this should be BOLD ITALIC") # will combine multiple effects msg = '\n ID Status Date(-t) Owner(-u) Description (-d)\n' tc.simple_print(self.tc.BOLD + self.tc.UNDERLINE, msg)
Here is some example for multi_color_print():
def print_status_footer(): tc = getTerminalColorClass() # statusLine = "Status: [+]add [@]block [-]reject [*]accept [#]workon [.]finish" prn_list = [ (tc.WHITE, "Status: "), (get_color_for_status('+'), "[+]add "), (get_color_for_status('@'), "[@]block "), (get_color_for_status('-'), "[-]reject "), (get_color_for_status('*'), "[*]accept "), (get_color_for_status('#'), "[#]workon "), (get_color_for_status('.'), "[.]finish ") ] tc.multi_color_print(prn_list) tc.simple_print(tc.WHITE, " ST=status PR=priority")
The commented out line shows what it was before I added color to it. The get_color_for_status() just returns one of the colors defined in the terminal color class.
I hope that this helps!
Was finally able to test on my script on bash, so a small addition (actually just replace the getTerminalColorClass() function with the following:
def getTerminalColorClass(): if '_stash' in globals() or '_stash' in locals(): if _stash != None: return StashTerminalColors() return TerminalColors()
How I was checking before was never going to work! ;-)
I appreciate you taking the time to post your code, this is indeed very helpful.
You're welcome! Glad to help.