diff --git a/bin/ls.py b/bin/ls.py index e948689..578579b 100644 --- a/bin/ls.py +++ b/bin/ls.py @@ -3,8 +3,17 @@ import os def get_app_class(template): class App(template): def F_main(self): - for file in os.listdir(self.argv[1]): - print(file) + if len(self.argv) == 1: + directory = "." + else: + directory = self.argv[1] + + try: + for file in os.listdir(directory): + print(file) + except OSError as e: + print(str(e)) + self.exit() return App