zsh support

README.md

@@ -9,7 +9,7 @@ This project is currently a work-in-progress. Only the fish shell is supported a
 ## Requires
  - python3
  - source-highlight
- - less/cat/ls/
+ - less/cat/ls/vim
 
 ## TODO
  - Documentation/License

do_what.py

@@ -3,7 +3,7 @@ import os
 import subprocess
 import configparser
 
-# supported values: fish, bash
+# supported values: fish, bash, zsh
 shell="fish"
 
 print_file="cat"
@@ -14,7 +14,7 @@ list_directory="ls --color=auto"
 def set_runtime_var(name, value, options=""):
 	if ( shell == "fish" ):
 		print("set "+name+" "+options+" \""+value+"\"")
-	elif ( shell == "bash" ):
+	elif ( shell == "bash" || shell == "zsh" ):
 		print("export "+name+"="+"\""+value+"\"")
 
 def is_binary_file(filepathname):